mirror of
https://github.com/gofiber/storage.git
synced 2025-10-04 08:16:36 +08:00
167 lines
3.2 KiB
Go
167 lines
3.2 KiB
Go
package leveldb
|
|
|
|
import (
|
|
"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)
|
|
}
|