go/leveldb/leveldb_test.go

package leveldb

import (
	"bytes"
	"fmt"
	"os"
	"sync"
	"testing"
)

var testConfigJson = []byte(`
    {
        "path" : "./testdb",
        "compression":true,
        "block_size" : 32,
        "write_buffer_size" : 2,
        "cache_size" : 20
    }
    `)

var testOnce sync.Once
var testDB *DB

func getTestDB() *DB {
	f := func() {
		var err error
		testDB, err = Open(testConfigJson)
		if err != nil {
			println(err.Error())
			panic(err)
		}
	}

	testOnce.Do(f)
	return testDB
}

func TestSimple(t *testing.T) {
	db := getTestDB()

	key := []byte("key")
	value := []byte("hello world")
	if err := db.Put(key, value); err != nil {
		t.Fatal(err)
	}

	if v, err := db.Get(key); err != nil {
		t.Fatal(err)
	} else if !bytes.Equal(v, value) {
		t.Fatal("not equal")
	}

	if err := db.Delete(key); err != nil {
		t.Fatal(err)
	}

	if v, err := db.Get(key); err != nil {
		t.Fatal(err)
	} else if v != nil {
		t.Fatal("must nil")
	}
}

func TestBatch(t *testing.T) {
	db := getTestDB()

	key1 := []byte("key1")
	key2 := []byte("key2")

	value := []byte("hello world")

	db.Put(key1, value)
	db.Put(key2, value)

	wb := db.NewWriteBatch()
	wb.Delete(key2)
	wb.Put(key1, []byte("hello world2"))

	if err := wb.Commit(); err != nil {
		t.Fatal(err)
	}

	if v, err := db.Get(key2); err != nil {
		t.Fatal(err)
	} else if v != nil {
		t.Fatal("must nil")
	}

	if v, err := db.Get(key1); err != nil {
		t.Fatal(err)
	} else if string(v) != "hello world2" {
		t.Fatal(string(v))
	}

	wb = db.NewWriteBatch()
	wb.Delete(key1)

	wb.Rollback()

	if v, err := db.Get(key1); err != nil {
		t.Fatal(err)
	} else if string(v) != "hello world2" {
		t.Fatal(string(v))
	}

	db.Delete(key1)
}

func TestIterator(t *testing.T) {
	db := getTestDB()
	for it := db.Iterator(nil, nil, 0); it.Valid(); it.Next() {
		db.Delete(it.Key())
	}

	for i := 0; i < 10; i++ {
		key := []byte(fmt.Sprintf("key_%d", i))
		value := []byte(fmt.Sprintf("value_%d", i))
		db.Put(key, value)
	}

	step := 0
	var it *Iterator
	for it = db.Iterator(nil, nil, 0); it.Valid(); it.Next() {
		key := it.Key()
		value := it.Value()

		if string(key) != fmt.Sprintf("key_%d", step) {
			t.Fatal(string(key), step)
		}

		if string(value) != fmt.Sprintf("value_%d", step) {
			t.Fatal(string(value), step)
		}

		step++
	}

	it.Close()

	step = 2
	for it = db.Iterator([]byte("key_2"), nil, 3); it.Valid(); it.Next() {
		key := it.Key()
		value := it.Value()

		if string(key) != fmt.Sprintf("key_%d", step) {
			t.Fatal(string(key), step)
		}

		if string(value) != fmt.Sprintf("value_%d", step) {
			t.Fatal(string(value), step)
		}

		step++
	}
	it.Close()

	if step != 5 {
		t.Fatal("invalid step", step)
	}

	step = 2
	for it = db.Iterator([]byte("key_2"), []byte("key_5"), 0); it.Valid(); it.Next() {
		key := it.Key()
		value := it.Value()

		if string(key) != fmt.Sprintf("key_%d", step) {
			t.Fatal(string(key), step)
		}

		if string(value) != fmt.Sprintf("value_%d", step) {
			t.Fatal(string(value), step)
		}

		step++
	}
	it.Close()

	if step != 5 {
		t.Fatal("invalid step", step)
	}

	step = 2
	for it = db.Iterator([]byte("key_5"), []byte("key_2"), 0); it.Valid(); it.Next() {
		step++
	}
	it.Close()

	if step != 2 {
		t.Fatal("must 0")
	}

	step = 9
	for it = db.ReverseIterator(nil, nil, 0); it.Valid(); it.Next() {
		key := it.Key()
		value := it.Value()

		if string(key) != fmt.Sprintf("key_%d", step) {
			t.Fatal(string(key), step)
		}

		if string(value) != fmt.Sprintf("value_%d", step) {
			t.Fatal(string(value), step)
		}

		step--
	}
	it.Close()

	step = 5
	for it = db.ReverseIterator([]byte("key_5"), nil, 3); it.Valid(); it.Next() {
		key := it.Key()
		value := it.Value()

		if string(key) != fmt.Sprintf("key_%d", step) {
			t.Fatal(string(key), step)
		}

		if string(value) != fmt.Sprintf("value_%d", step) {
			t.Fatal(string(value), step)
		}

		step--
	}
	it.Close()

	if step != 2 {
		t.Fatal("invalid step", step)
	}

	step = 5
	for it = db.ReverseIterator([]byte("key_5"), []byte("key_2"), 0); it.Valid(); it.Next() {
		key := it.Key()
		value := it.Value()

		if string(key) != fmt.Sprintf("key_%d", step) {
			t.Fatal(string(key), step)
		}

		if string(value) != fmt.Sprintf("value_%d", step) {
			t.Fatal(string(value), step)
		}

		step--
	}
	it.Close()

	if step != 2 {
		t.Fatal("invalid step", step)
	}

	step = 5
	for it = db.ReverseIterator([]byte("key_2"), []byte("key_5"), 0); it.Valid(); it.Next() {
		step--
	}
	it.Close()

	if step != 5 {
		t.Fatal("must 5")
	}
}

func TestIterator_2(t *testing.T) {
	db := getTestDB()
	for it := db.Iterator(nil, nil, 0); it.Valid(); it.Next() {
		db.Delete(it.Key())
	}

	db.Put([]byte("key_1"), []byte("value_1"))
	db.Put([]byte("key_7"), []byte("value_9"))
	db.Put([]byte("key_9"), []byte("value_9"))

	it := db.Iterator([]byte("key_0"), []byte("key_8"), 0)
	if !it.Valid() {
		t.Fatal("must valid")
	}

	if string(it.Key()) != "key_1" {
		t.Fatal(string(it.Key()))
	}

	it = db.ReverseIterator([]byte("key_8"), []byte("key_0"), 0)
	if !it.Valid() {
		t.Fatal("must valid")
	}

	if string(it.Key()) != "key_7" {
		t.Fatal(string(it.Key()))
	}

	for it := db.Iterator(nil, nil, 0); it.Valid(); it.Next() {
		db.Delete(it.Key())
	}

	it = db.Iterator([]byte("key_0"), []byte("key_8"), 0)
	if it.Valid() {
		t.Fatal("must not valid")
	}

	it = db.ReverseIterator([]byte("key_8"), []byte("key_0"), 0)
	if it.Valid() {
		t.Fatal("must not valid")
	}

}

func TestSnapshot(t *testing.T) {
	db := getTestDB()

	key := []byte("key")
	value := []byte("hello world")

	db.Put(key, value)

	s := db.NewSnapshot()
	defer s.Close()

	db.Put(key, []byte("hello world2"))

	if v, err := s.Get(key); err != nil {
		t.Fatal(err)
	} else if string(v) != string(value) {
		t.Fatal(string(v))
	}

	found := false
	for it := s.Iterator(nil, nil, 0); it.Valid(); it.Next() {
		if string(it.Key()) == string(key) {
			found = true
			break
		}
	}

	if !found {
		t.Fatal("must found")
	}

	found = false
	for it := s.ReverseIterator(nil, nil, 0); it.Valid(); it.Next() {
		if string(it.Key()) == string(key) {
			found = true
			break
		}
	}

	if !found {
		t.Fatal("must found")
	}

}

func TestDestroy(t *testing.T) {
	db := getTestDB()

	db.Destroy()

	if _, err := os.Stat(db.cfg.Path); !os.IsNotExist(err) {
		t.Fatal("must not exist")
	}
}