wukongd/utils/kv_utills_test.go

package utils

import (
	//"github.com/cznic/kv"
	"github.com/huichen/wukong/storage"
	"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")
// }

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

	db, err = OpenOrCreateKv("test.kv", &storage.Options{})
	Expect(t, "<nil>", err)
	db.Buc
	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")
}