kv_test.go

package storm

import (
	"net/mail"
	"testing"
	"time"

	"github.com/asdine/storm/codec/gob"
	"github.com/asdine/storm/codec/json"
	bolt "go.etcd.io/bbolt"
	"github.com/stretchr/testify/require"
)

func TestGet(t *testing.T) {
	db, cleanup := createDB(t)
	defer cleanup()

	err := db.Set("trash", 10, 100)
	require.NoError(t, err)

	var nb int
	err = db.Get("trash", 10, &nb)
	require.NoError(t, err)
	require.Equal(t, 100, nb)

	tm := time.Now()
	err = db.Set("logs", tm, "I'm hungry")
	require.NoError(t, err)

	var message string
	err = db.Get("logs", tm, &message)
	require.NoError(t, err)
	require.Equal(t, "I'm hungry", message)

	var hand int
	err = db.Get("wallet", "100 bucks", &hand)
	require.Equal(t, ErrNotFound, err)

	err = db.Set("wallet", "10 bucks", 10)
	require.NoError(t, err)

	err = db.Get("wallet", "100 bucks", &hand)
	require.Equal(t, ErrNotFound, err)

	err = db.Get("logs", tm, nil)
	require.Equal(t, ErrPtrNeeded, err)

	err = db.Get("", nil, nil)
	require.Equal(t, ErrPtrNeeded, err)

	err = db.Get("", "100 bucks", &hand)
	require.Equal(t, ErrNotFound, err)
}

func TestGetBytes(t *testing.T) {
	db, cleanup := createDB(t)
	defer cleanup()

	err := db.SetBytes("trash", "a", []byte("hi"))
	require.NoError(t, err)

	val, err := db.GetBytes("trash", "a")
	require.NoError(t, err)
	require.Equal(t, []byte("hi"), val)

	_, err = db.GetBytes("trash", "b")
	require.Equal(t, ErrNotFound, err)
}

func TestSet(t *testing.T) {
	db, cleanup := createDB(t)
	defer cleanup()

	err := db.Set("b1", 10, 10)
	require.NoError(t, err)
	err = db.Set("b1", "best friend's mail", &mail.Address{Name: "Gandalf", Address: "gandalf@lorien.ma"})
	require.NoError(t, err)
	err = db.Set("b2", []byte("i'm already a slice of bytes"), "a value")
	require.NoError(t, err)
	err = db.Set("b2", []byte("i'm already a slice of bytes"), nil)
	require.NoError(t, err)
	err = db.Set("b1", 0, 100)
	require.NoError(t, err)
	err = db.Set("b1", nil, 100)
	require.Error(t, err)

	db.Bolt.View(func(tx *bolt.Tx) error {
		b1 := tx.Bucket([]byte("b1"))
		require.NotNil(t, b1)
		b2 := tx.Bucket([]byte("b2"))
		require.NotNil(t, b2)

		k1, err := toBytes(10, json.Codec)
		require.NoError(t, err)
		val := b1.Get(k1)
		require.NotNil(t, val)

		k2 := []byte("best friend's mail")
		val = b1.Get(k2)
		require.NotNil(t, val)

		k3, err := toBytes(0, json.Codec)
		require.NoError(t, err)
		val = b1.Get(k3)
		require.NotNil(t, val)

		return nil
	})

	err = db.Set("", 0, 100)
	require.Error(t, err)

	err = db.Set("b", nil, 100)
	require.Error(t, err)

	err = db.Set("b", 10, nil)
	require.NoError(t, err)

	err = db.Set("b", nil, nil)
	require.Error(t, err)
}

func TestSetMetadata(t *testing.T) {
	db, cleanup := createDB(t, Batch())
	defer cleanup()

	w := User{ID: 10, Name: "John"}
	err := db.Set("User", 10, &w)
	require.NoError(t, err)
	n := db.WithCodec(gob.Codec)
	err = n.Set("User", 10, &w)
	require.Equal(t, ErrDifferentCodec, err)
}

func TestDelete(t *testing.T) {
	db, cleanup := createDB(t)
	defer cleanup()

	err := db.Set("files", "myfile.csv", "a,b,c,d")
	require.NoError(t, err)
	err = db.Delete("files", "myfile.csv")
	require.NoError(t, err)
	err = db.Delete("files", "myfile.csv")
	require.NoError(t, err)
	err = db.Delete("i don't exist", "myfile.csv")
	require.Equal(t, ErrNotFound, err)
	err = db.Delete("", nil)
	require.Equal(t, ErrNotFound, err)
}

func TestKeyExists(t *testing.T) {
	db, cleanup := createDB(t)
	defer cleanup()

	err := db.Set("files", "myfile.csv", "a,b,c,d")
	require.NoError(t, err)

	exists, err := db.KeyExists("files", "myfile.csv")
	require.NoError(t, err)
	require.True(t, exists)

	err = db.Delete("files", "myfile.csv")
	require.NoError(t, err)

	exists, err = db.KeyExists("files", "myfile.csv")
	require.NoError(t, err)
	require.False(t, exists)

	exists, err = db.KeyExists("i don't exist", "myfile.csv")
	require.Equal(t, ErrNotFound, err)

	exists, err = db.KeyExists("", nil)
	require.Equal(t, ErrNotFound, err)
}