route/vendor/github.com/asdine/storm/node_test.go

63 lines
1.5 KiB
Go
Raw Normal View History

2017-10-06 15:29:20 +00:00
package storm
import (
"testing"
"github.com/asdine/storm/codec/gob"
"github.com/asdine/storm/codec/json"
"github.com/boltdb/bolt"
"github.com/stretchr/testify/require"
)
func TestNode(t *testing.T) {
db, cleanup := createDB(t, Root("a"))
defer cleanup()
n1 := db.From("b", "c")
node1, ok := n1.(*node)
require.True(t, ok)
require.Equal(t, db, node1.s)
require.NotEqual(t, db.root, n1)
require.Equal(t, []string{"a"}, db.root.rootBucket)
require.Equal(t, []string{"b", "c"}, node1.rootBucket)
n2 := n1.From("d", "e")
node2, ok := n2.(*node)
require.True(t, ok)
require.Equal(t, []string{"b", "c", "d", "e"}, node2.rootBucket)
}
func TestNodeWithTransaction(t *testing.T) {
db, cleanup := createDB(t)
defer cleanup()
var user User
db.Bolt.Update(func(tx *bolt.Tx) error {
dbx := db.WithTransaction(tx)
err := dbx.Save(&User{ID: 10, Name: "John"})
require.NoError(t, err)
err = dbx.One("ID", 10, &user)
require.NoError(t, err)
require.Equal(t, "John", user.Name)
return nil
})
err := db.One("ID", 10, &user)
require.NoError(t, err)
}
func TestNodeWithCodec(t *testing.T) {
db, cleanup := createDB(t)
defer cleanup()
n := db.From("a").(*node)
require.Equal(t, json.Codec, n.codec)
n = n.From("b", "c", "d").(*node)
require.Equal(t, json.Codec, n.codec)
n = db.WithCodec(gob.Codec).(*node)
n = n.From("e").(*node)
require.Equal(t, gob.Codec, n.codec)
o := n.From("f").WithCodec(json.Codec).(*node)
require.Equal(t, gob.Codec, n.codec)
require.Equal(t, json.Codec, o.codec)
}