63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
|
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)
|
||
|
|
}
|