fix datastore Put and added tests (#402)

This commit is contained in:
Pedro Nasser
2016-12-07 14:59:54 -02:00
committed by Travis Reeder
parent 0baf03841f
commit 9382f0b133
6 changed files with 132 additions and 37 deletions

View File

@@ -1,27 +1,13 @@
package datastore
import (
"bytes"
"context"
"log"
"os"
"testing"
"github.com/Sirupsen/logrus"
"github.com/gin-gonic/gin"
"github.com/iron-io/functions/api/models"
)
func setLogBuffer() *bytes.Buffer {
var buf bytes.Buffer
buf.WriteByte('\n')
logrus.SetOutput(&buf)
gin.DefaultErrorWriter = &buf
gin.DefaultWriter = &buf
log.SetOutput(&buf)
return &buf
}
const tmpBolt = "/tmp/func_test_bolt.db"
func TestBolt(t *testing.T) {
@@ -35,16 +21,6 @@ func TestBolt(t *testing.T) {
t.Fatalf("Error when creating datastore: %v", err)
}
testApp := &models.App{
Name: "Test",
}
testRoute := &models.Route{
AppName: testApp.Name,
Path: "/test",
Image: "iron/hello",
}
// Testing insert app
_, err = ds.InsertApp(ctx, nil)
if err != models.ErrDatastoreEmptyApp {
@@ -272,4 +248,43 @@ func TestBolt(t *testing.T) {
t.Log(buf.String())
t.Fatalf("Test RemoveApp: failed to remove the route")
}
// Testing Put/Get
err = ds.Put(ctx, nil, nil)
if err != models.ErrDatastoreEmptyKey {
t.Log(buf.String())
t.Fatalf("Test Put(nil,nil): expected error `%v`, but it was `%v`", models.ErrDatastoreEmptyKey, err)
}
err = ds.Put(ctx, []byte("test"), []byte("success"))
if err != nil {
t.Log(buf.String())
t.Fatalf("Test Put: unexpected error: %v", err)
}
val, err := ds.Get(ctx, []byte("test"))
if err != nil {
t.Log(buf.String())
t.Fatalf("Test Put: unexpected error: %v", err)
}
if string(val) != "success" {
t.Log(buf.String())
t.Fatalf("Test Get: expected value to be `%v`, but it was `%v`", "success", string(val))
}
err = ds.Put(ctx, []byte("test"), nil)
if err != nil {
t.Log(buf.String())
t.Fatalf("Test Put: unexpected error: %v", err)
}
val, err = ds.Get(ctx, []byte("test"))
if err != nil {
t.Log(buf.String())
t.Fatalf("Test Put: unexpected error: %v", err)
}
if string(val) != "" {
t.Log(buf.String())
t.Fatalf("Test Get: expected value to be `%v`, but it was `%v`", "", string(val))
}
}