mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
fix datastore Put and added tests (#402)
This commit is contained in:
committed by
Travis Reeder
parent
0baf03841f
commit
9382f0b133
@@ -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))
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user