storage/leveldb/leveldb_test.go

package leveldb

import (
	"fmt"
	"os"
	"testing"
	"time"

	"github.com/stretchr/testify/require"
)

func removeAllFiles(dir string) error {
	return os.RemoveAll(dir)
}

func Test_New_EmptyConfig(t *testing.T) {
	db := New()
	require.NotNil(t, db)

	_, err := os.Stat("./fiber.leveldb")
	require.Nil(t, err)

	err = removeAllFiles("./fiber.leveldb")
	require.Nil(t, err)
}

func Test_New_WithConfig(t *testing.T) {
	db := New(Config{
		DBPath: "./testdb",
	})
	require.NotNil(t, db)
	_, err := os.Stat("./testdb")
	require.Nil(t, err)

	err = removeAllFiles("./testdb")
	require.Nil(t, err)
}

func Test_Set_Overwrite(t *testing.T) {
	db := New()

	db.Set([]byte("key"), []byte("value"), time.Second*1)
	db.Set([]byte("key"), []byte("value2"), time.Second*1)

	value, err := db.Get([]byte("key"))
	require.Nil(t, err)
	require.Equal(t, []byte("value2"), value)

	err = removeAllFiles("./fiber.leveldb")
	require.Nil(t, err)
}

func Test_Get_For0Second(t *testing.T) {
	db := New()

	db.Set([]byte("key"), []byte("value"), 0)

	_, err := db.Get([]byte("key"))
	require.Nil(t, err)

	err = removeAllFiles("./fiber.leveldb")
	require.Nil(t, err)
}

func Test_Get_ForExpired1Second(t *testing.T) {
	db := New()

	db.Set([]byte("key"), []byte("value"), time.Second*1)

	time.Sleep(time.Second * 2)

	value, err := db.Get([]byte("key"))
	require.Nil(t, err)
	require.Nil(t, value)

	err = removeAllFiles("./fiber.leveldb")
	require.Nil(t, err)
}

func Test_Delete_WhileThereIsData(t *testing.T) {
	db := New()

	db.Set([]byte("key"), []byte("value"), time.Second*1)

	err := db.Delete("key")
	require.Nil(t, err)

	value, err := db.Get([]byte("key"))
	require.Nil(t, err)
	require.Nil(t, value)

	err = removeAllFiles("./fiber.leveldb")
	require.Nil(t, err)

}

func Test_Reset(t *testing.T) {
	db := New()

	db.Set([]byte("key1"), []byte("value1"), time.Second*1)
	db.Set([]byte("key2"), []byte("value2"), time.Second*1)
	db.Set([]byte("key3"), []byte("value3"), time.Second*1)

	db.Reset()

	value, err := db.Get([]byte("key1"))
	require.Nil(t, err)
	require.Nil(t, value)

	value, err = db.Get([]byte("key2"))
	require.Nil(t, err)
	require.Nil(t, value)

	value, err = db.Get([]byte("key3"))
	require.Nil(t, err)
	require.Nil(t, value)

	err = removeAllFiles("./fiber.leveldb")
	require.Nil(t, err)

}

func Test_Close(t *testing.T) {
	db := New()

	db.Close()

	err := db.Conn().Put([]byte("key"), []byte("value"), nil)
	require.Error(t, err)

	err = removeAllFiles("./fiber.leveldb")
	require.Nil(t, err)
}

func Test_GarbageCollection_AfterWorking(t *testing.T) {
	db := New(Config{
		GCInterval: time.Second * 1,
	})

	db.Set([]byte("key"), []byte("value"), time.Second*1)

	time.Sleep(time.Second * 2)

	value, err := db.Conn().Get([]byte("key"), nil)
	require.Error(t, err)
	require.Equal(t, []byte{}, value)

	err = removeAllFiles("./fiber.leveldb")
	require.Nil(t, err)
}

func Test_GarbageCollection_BeforeWorking(t *testing.T) {
	db := New(Config{
		GCInterval: time.Second * 1,
	})

	db.Set([]byte("key"), []byte("value"), time.Second*1)

	//time.Sleep(time.Second * 2)

	value, err := db.Conn().Get([]byte("key"), nil)
	require.Nil(t, err)
	require.NotNil(t, value)

	err = removeAllFiles("./fiber.leveldb")
	require.Nil(t, err)
}

func Benchmark_Set(b *testing.B) {
	db := New()
	defer func() {
		db.Close()
		_ = removeAllFiles("./fiber.leveldb")
	}()

	b.ResetTimer()
	b.RunParallel(func(pb *testing.PB) {
		i := 0
		for pb.Next() {
			key := []byte(fmt.Sprintf("key_%d", i))
			value := []byte(fmt.Sprintf("value_%d", i))
			_ = db.Set(key, value, 0)
			i++
		}
	})
}

func Benchmark_Get(b *testing.B) {
	db := New()
	defer func() {
		db.Close()
		_ = removeAllFiles("./fiber.leveldb")
	}()

	key := []byte("test_key")
	value := []byte("test_value")
	_ = db.Set(key, value, 0)

	b.ResetTimer()
	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			_, _ = db.Get(key)
		}
	})
}

func Benchmark_Delete(b *testing.B) {
	db := New()
	defer func() {
		db.Close()
		_ = removeAllFiles("./fiber.leveldb")
	}()

	b.ResetTimer()
	b.RunParallel(func(pb *testing.PB) {
		i := 0
		for pb.Next() {
			key := fmt.Sprintf("key_%d", i)
			_ = db.Set([]byte(key), []byte("value"), 0)
			_ = db.Delete(key)
			i++
		}
	})
}