wukongd/utils/kv_utills_test.go

package utils

import (
	"github.com/cznic/kv"
	"os"
	"testing"
)

func TestOpenOrCreateKv(t *testing.T) {
	db, err := OpenOrCreateKv("test.kv", &kv.Options{})
	Expect(t, "<nil>", err)
	db.Close()

	db, err = OpenOrCreateKv("test.kv", &kv.Options{})
	Expect(t, "<nil>", err)
	err = db.Set([]byte("key1"), []byte("value1"))
	Expect(t, "<nil>", err)

	buffer := make([]byte, 100)
	buffer, err = db.Get(nil, []byte("key1"))
	Expect(t, "<nil>", err)
	Expect(t, "value1", string(buffer))

	walFile := db.WALName()
	db.Close()
	os.Remove(walFile)
	os.Remove("test.kv")
}