db_test.go

package rosedb

import (
	"errors"
	"io"
	"math/rand"
	"os"
	"sync"
	"testing"
	"time"

	"github.com/rosedblabs/rosedb/v2/utils"
	"github.com/stretchr/testify/assert"
)

func TestDB_Put_Normal(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	for i := 0; i < 100; i++ {
		err := db.Put(utils.GetTestKey(rand.Int()), utils.RandomValue(128))
		assert.Nil(t, err)
		err = db.Put(utils.GetTestKey(rand.Int()), utils.RandomValue(KB))
		assert.Nil(t, err)
		err = db.Put(utils.GetTestKey(rand.Int()), utils.RandomValue(5*KB))
		assert.Nil(t, err)
	}

	// reopen
	err = db.Close()
	assert.Nil(t, err)
	db2, err := Open(options)
	assert.Nil(t, err)
	defer func() {
		_ = db2.Close()
	}()
	stat := db2.Stat()
	assert.Equal(t, 300, stat.KeysNum)
}

func TestDB_Get_Normal(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	// not exist
	val1, err := db.Get([]byte("not-exist"))
	assert.Nil(t, val1)
	assert.Equal(t, ErrKeyNotFound, err)

	generateData(t, db, 1, 100, 128)
	for i := 1; i < 100; i++ {
		val, err := db.Get(utils.GetTestKey(i))
		assert.Nil(t, err)
		assert.Equal(t, len(val), len(utils.RandomValue(128)))
	}
	generateData(t, db, 200, 300, KB)
	for i := 200; i < 300; i++ {
		val, err := db.Get(utils.GetTestKey(i))
		assert.Nil(t, err)
		assert.Equal(t, len(val), len(utils.RandomValue(KB)))
	}
	generateData(t, db, 400, 500, 4*KB)
	for i := 400; i < 500; i++ {
		val, err := db.Get(utils.GetTestKey(i))
		assert.Nil(t, err)
		assert.Equal(t, len(val), len(utils.RandomValue(4*KB)))
	}
}

func TestDB_Close_Sync(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	err = db.Sync()
	assert.Nil(t, err)
}

func TestDB_Concurrent_Put(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	wg := sync.WaitGroup{}
	m := sync.Map{}
	wg.Add(10)
	for i := 0; i < 10; i++ {
		go func() {
			defer wg.Done()
			for i := 0; i < 10000; i++ {
				key := utils.GetTestKey(rand.Int())
				m.Store(string(key), struct{}{})
				e := db.Put(key, utils.RandomValue(128))
				assert.Nil(t, e)
			}
		}()
	}
	wg.Wait()

	var count int
	m.Range(func(key, value any) bool {
		count++
		return true
	})
	assert.Equal(t, count, db.index.Size())
}

func TestDB_Concurrent_Get(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	for i := 0; i < 10000; i++ {
		err = db.Put(utils.GetTestKey(i), utils.RandomValue(128))
		assert.Nil(t, err)
	}
	for i := 10000; i < 20000; i++ {
		err = db.Put(utils.GetTestKey(i), utils.RandomValue(4096))
		assert.Nil(t, err)
	}

	var wg sync.WaitGroup
	wg.Add(50)
	for i := 0; i < 50; i++ {
		go func() {
			defer wg.Done()
			db.Ascend(func(key []byte, value []byte) (bool, error) {
				assert.NotNil(t, key)
				assert.NotNil(t, value)
				return true, nil
			})
		}()
	}
	wg.Wait()
}

func TestDB_Ascend(t *testing.T) {
	// Create a test database instance
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	// Insert some test data
	data := []struct {
		key   []byte
		value []byte
	}{
		{[]byte("key1"), []byte("value1")},
		{[]byte("key2"), []byte("value2")},
		{[]byte("key3"), []byte("value3")},
	}

	for _, d := range data {
		if err := db.Put(d.key, d.value); err != nil {
			t.Fatalf("Failed to put data: %v", err)
		}
	}

	// Test Ascend function
	var result []string
	db.Ascend(func(k []byte, v []byte) (bool, error) {
		result = append(result, string(k))
		return true, nil // No error here
	})

	if err != nil {
		t.Errorf("Ascend returned an error: %v", err)
	}

	expected := []string{"key1", "key2", "key3"}
	if len(result) != len(expected) {
		t.Errorf("Unexpected number of results. Expected: %v, Got: %v", expected, result)
	} else {
		for i, val := range expected {
			if result[i] != val {
				t.Errorf("Unexpected result at index %d. Expected: %v, Got: %v", i, val, result[i])
			}
		}
	}
}

func TestDB_Descend(t *testing.T) {
	// Create a test database instance
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	// Insert some test data
	data := []struct {
		key   []byte
		value []byte
	}{
		{[]byte("key1"), []byte("value1")},
		{[]byte("key2"), []byte("value2")},
		{[]byte("key3"), []byte("value3")},
	}

	for _, d := range data {
		if err := db.Put(d.key, d.value); err != nil {
			t.Fatalf("Failed to put data: %v", err)
		}
	}

	// Test Descend function
	var result []string
	db.Descend(func(k []byte, v []byte) (bool, error) {
		result = append(result, string(k))
		return true, nil
	})

	if err != nil {
		t.Errorf("Descend returned an error: %v", err)
	}

	expected := []string{"key3", "key2", "key1"}
	if len(result) != len(expected) {
		t.Errorf("Unexpected number of results. Expected: %v, Got: %v", expected, result)
	} else {
		for i, val := range expected {
			if result[i] != val {
				t.Errorf("Unexpected result at index %d. Expected: %v, Got: %v", i, val, result[i])
			}
		}
	}
}

func TestDB_AscendRange(t *testing.T) {
	// Create a test database instance
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	// Insert some test data
	data := []struct {
		key   []byte
		value []byte
	}{
		{[]byte("apple"), []byte("value1")},
		{[]byte("banana"), []byte("value2")},
		{[]byte("cherry"), []byte("value3")},
		{[]byte("date"), []byte("value4")},
		{[]byte("grape"), []byte("value5")},
		{[]byte("kiwi"), []byte("value6")},
	}

	for _, d := range data {
		if err := db.Put(d.key, d.value); err != nil {
			t.Fatalf("Failed to put data: %v", err)
		}
	}

	// Test AscendRange
	var resultAscendRange []string
	db.AscendRange([]byte("banana"), []byte("grape"), func(k []byte, v []byte) (bool, error) {
		resultAscendRange = append(resultAscendRange, string(k))
		return true, nil
	})
	assert.Equal(t, []string{"banana", "cherry", "date"}, resultAscendRange)
}

func TestDB_DescendRange(t *testing.T) {
	// Create a test database instance
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	// Insert some test data
	data := []struct {
		key   []byte
		value []byte
	}{
		{[]byte("apple"), []byte("value1")},
		{[]byte("banana"), []byte("value2")},
		{[]byte("cherry"), []byte("value3")},
		{[]byte("date"), []byte("value4")},
		{[]byte("grape"), []byte("value5")},
		{[]byte("kiwi"), []byte("value6")},
	}

	for _, d := range data {
		if err := db.Put(d.key, d.value); err != nil {
			t.Fatalf("Failed to put data: %v", err)
		}
	}

	// Test DescendRange
	var resultDescendRange []string
	db.DescendRange([]byte("grape"), []byte("cherry"), func(k []byte, v []byte) (bool, error) {
		resultDescendRange = append(resultDescendRange, string(k))
		return true, nil
	})
	assert.Equal(t, []string{"grape", "date"}, resultDescendRange)
}

func TestDB_AscendGreaterOrEqual(t *testing.T) {
	// Create a test database instance
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	// Insert some test data
	data := []struct {
		key   []byte
		value []byte
	}{
		{[]byte("apple"), []byte("value1")},
		{[]byte("banana"), []byte("value2")},
		{[]byte("cherry"), []byte("value3")},
		{[]byte("date"), []byte("value4")},
		{[]byte("grape"), []byte("value5")},
		{[]byte("kiwi"), []byte("value6")},
	}

	for _, d := range data {
		if err := db.Put(d.key, d.value); err != nil {
			t.Fatalf("Failed to put data: %v", err)
		}
	}

	// Test AscendGreaterOrEqual
	var resultAscendGreaterOrEqual []string
	db.AscendGreaterOrEqual([]byte("date"), func(k []byte, v []byte) (bool, error) {
		resultAscendGreaterOrEqual = append(resultAscendGreaterOrEqual, string(k))
		return true, nil
	})
	assert.Equal(t, []string{"date", "grape", "kiwi"}, resultAscendGreaterOrEqual)
}

func TestDB_DescendLessOrEqual(t *testing.T) {
	// Create a test database instance
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	// Insert some test data
	data := []struct {
		key   []byte
		value []byte
	}{
		{[]byte("apple"), []byte("value1")},
		{[]byte("banana"), []byte("value2")},
		{[]byte("cherry"), []byte("value3")},
		{[]byte("date"), []byte("value4")},
		{[]byte("grape"), []byte("value5")},
		{[]byte("kiwi"), []byte("value6")},
	}

	for _, d := range data {
		if err := db.Put(d.key, d.value); err != nil {
			t.Fatalf("Failed to put data: %v", err)
		}
	}

	// Test DescendLessOrEqual
	var resultDescendLessOrEqual []string
	db.DescendLessOrEqual([]byte("grape"), func(k []byte, v []byte) (bool, error) {
		resultDescendLessOrEqual = append(resultDescendLessOrEqual, string(k))
		return true, nil
	})
	assert.Equal(t, []string{"grape", "date", "cherry", "banana", "apple"}, resultDescendLessOrEqual)
}

func TestDB_AscendKeys(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	err = db.Put([]byte("aacd"), utils.RandomValue(10))
	assert.Nil(t, err)

	validate := func(target [][]byte, pattern []byte) {
		var keys [][]byte
		db.AscendKeys(pattern, false, func(k []byte) (bool, error) {
			keys = append(keys, k)
			return true, nil
		})
		assert.Equal(t, keys, target)
	}

	validate([][]byte{[]byte("aacd")}, nil)

	err = db.Put([]byte("bbde"), utils.RandomValue(10))
	assert.Nil(t, err)
	err = db.Put([]byte("cdea"), utils.RandomValue(10))
	assert.Nil(t, err)
	err = db.Put([]byte("bcae"), utils.RandomValue(10))
	assert.Nil(t, err)

	validate([][]byte{[]byte("aacd"), []byte("bbde"), []byte("bcae"), []byte("cdea")}, nil)
}

func TestDB_AscendKeysExpired(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	validate := func(target [][]byte, pattern []byte) {
		var keys [][]byte
		db.AscendKeys(pattern, true, func(k []byte) (bool, error) {
			keys = append(keys, k)
			return true, nil
		})
		assert.Equal(t, keys, target)
	}

	err = db.PutWithTTL([]byte("bbde"), utils.RandomValue(10), time.Millisecond*500)
	assert.Nil(t, err)
	err = db.Put([]byte("cdea"), utils.RandomValue(10))
	assert.Nil(t, err)
	err = db.Put([]byte("bcae"), utils.RandomValue(10))
	assert.Nil(t, err)
	time.Sleep(time.Millisecond * 600)

	validate([][]byte{[]byte("bcae"), []byte("cdea")}, nil)
}

func TestDB_AscendKeysRange(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	// add some data
	err = db.Put([]byte("apple"), []byte("value1"))
	assert.Nil(t, err)
	err = db.Put([]byte("banana"), []byte("value2"))
	assert.Nil(t, err)
	err = db.PutWithTTL([]byte("cherry"), []byte("value3"), time.Millisecond*100)
	assert.Nil(t, err)
	err = db.Put([]byte("date"), []byte("value4"))
	assert.Nil(t, err)
	err = db.Put([]byte("grape"), []byte("value5"))
	assert.Nil(t, err)

	// normal iteration
	var result []string
	err = db.AscendKeysRange([]byte("banana"), []byte("grape"), nil, false, func(k []byte) (bool, error) {
		result = append(result, string(k))
		return true, nil
	})
	assert.Nil(t, err)
	assert.Equal(t, []string{"banana", "cherry", "date"}, result)

	// Regular expression iteration
	result = nil
	err = db.AscendKeysRange([]byte("apple"), []byte("grape"), []byte(".*e$"), false, func(k []byte) (bool, error) {
		result = append(result, string(k))
		return true, nil
	})
	assert.Nil(t, err)
	assert.Equal(t, []string{"apple", "date"}, result)

	// Filter expired keys
	time.Sleep(time.Millisecond * 200)
	result = nil
	err = db.AscendKeysRange([]byte("banana"), []byte("grape"), nil, true, func(k []byte) (bool, error) {
		result = append(result, string(k))
		return true, nil
	})
	assert.Nil(t, err)
	assert.Equal(t, []string{"banana", "date"}, result)
}

func TestDB_DescendKeysRange(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	// add some data
	err = db.Put([]byte("apple"), []byte("value1"))
	assert.Nil(t, err)
	err = db.Put([]byte("banana"), []byte("value2"))
	assert.Nil(t, err)
	err = db.PutWithTTL([]byte("cherry"), []byte("value3"), time.Millisecond*100)
	assert.Nil(t, err)
	err = db.Put([]byte("date"), []byte("value4"))
	assert.Nil(t, err)
	err = db.Put([]byte("grape"), []byte("value5"))
	assert.Nil(t, err)

	// normal iteration
	var result []string
	err = db.DescendKeysRange([]byte("grape"), []byte("banana"), nil, false, func(k []byte) (bool, error) {
		result = append(result, string(k))
		return true, nil
	})
	assert.Nil(t, err)
	assert.Equal(t, []string{"grape", "date", "cherry"}, result)

	// Regular expression iteration
	result = nil
	err = db.DescendKeysRange([]byte("grape"), []byte("apple"), []byte(".*e$"), false, func(k []byte) (bool, error) {
		result = append(result, string(k))
		return true, nil
	})
	assert.Nil(t, err)
	assert.Equal(t, []string{"grape", "date"}, result)

	// Filter expired keys
	time.Sleep(time.Millisecond * 200)
	result = nil
	err = db.DescendKeysRange([]byte("grape"), []byte("banana"), nil, true, func(k []byte) (bool, error) {
		result = append(result, string(k))
		return true, nil
	})
	assert.Nil(t, err)
	assert.Equal(t, []string{"grape", "date"}, result)
}

func TestDB_DescendKeys(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	err = db.Put([]byte("aacd"), utils.RandomValue(10))
	assert.Nil(t, err)

	validate := func(target [][]byte, pattern []byte) {
		var keys [][]byte
		db.DescendKeys(pattern, false, func(k []byte) (bool, error) {
			keys = append(keys, k)
			return true, nil
		})
		assert.Equal(t, keys, target)
	}

	validate([][]byte{[]byte("aacd")}, nil)

	err = db.Put([]byte("bbde"), utils.RandomValue(10))
	assert.Nil(t, err)
	err = db.Put([]byte("cdea"), utils.RandomValue(10))
	assert.Nil(t, err)
	err = db.Put([]byte("bcae"), utils.RandomValue(10))
	assert.Nil(t, err)

	validate([][]byte{[]byte("cdea"), []byte("bcae"), []byte("bbde"), []byte("aacd")}, nil)
}

func TestDB_DescendKeysExpired(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	validate := func(target [][]byte, pattern []byte) {
		var keys [][]byte
		db.DescendKeys(pattern, true, func(k []byte) (bool, error) {
			keys = append(keys, k)
			return true, nil
		})
		assert.Equal(t, keys, target)
	}

	err = db.Put([]byte("bbde"), utils.RandomValue(10))
	assert.Nil(t, err)
	err = db.PutWithTTL([]byte("cdea"), utils.RandomValue(10), time.Millisecond*500)
	assert.Nil(t, err)
	err = db.PutWithTTL([]byte("bcae"), utils.RandomValue(10), time.Millisecond*500)
	assert.Nil(t, err)

	time.Sleep(time.Millisecond * 600)

	validate([][]byte{[]byte("bbde")}, nil)
}

func TestDB_PutWithTTL(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	err = db.PutWithTTL(utils.GetTestKey(1), utils.RandomValue(128), time.Millisecond*100)
	assert.Nil(t, err)
	val1, err := db.Get(utils.GetTestKey(1))
	assert.Nil(t, err)
	assert.NotNil(t, val1)
	time.Sleep(time.Millisecond * 200)
	val2, err := db.Get(utils.GetTestKey(1))
	assert.Equal(t, err, ErrKeyNotFound)
	assert.Nil(t, val2)

	err = db.PutWithTTL(utils.GetTestKey(2), utils.RandomValue(128), time.Millisecond*200)
	assert.Nil(t, err)
	// rewrite
	err = db.Put(utils.GetTestKey(2), utils.RandomValue(128))
	assert.Nil(t, err)
	time.Sleep(time.Millisecond * 200)
	val3, err := db.Get(utils.GetTestKey(2))
	assert.Nil(t, err)
	assert.NotNil(t, val3)

	err = db.Close()
	assert.Nil(t, err)

	db2, err := Open(options)
	assert.Nil(t, err)

	val4, err := db2.Get(utils.GetTestKey(1))
	assert.Equal(t, err, ErrKeyNotFound)
	assert.Nil(t, val4)

	val5, err := db2.Get(utils.GetTestKey(2))
	assert.Nil(t, err)
	assert.NotNil(t, val5)

	_ = db2.Close()
}

func TestDB_RePutWithTTL(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	err = db.Put(utils.GetTestKey(10), utils.RandomValue(10))
	assert.Nil(t, err)
	err = db.PutWithTTL(utils.GetTestKey(10), utils.RandomValue(10), time.Millisecond*100)
	assert.Nil(t, err)
	time.Sleep(time.Second * 1) // wait for expired

	val1, err := db.Get(utils.GetTestKey(10))
	assert.Equal(t, err, ErrKeyNotFound)
	assert.Nil(t, val1)

	err = db.Merge(true)
	assert.Nil(t, err)

	val2, err := db.Get(utils.GetTestKey(10))
	assert.Equal(t, err, ErrKeyNotFound)
	assert.Nil(t, val2)
}

func TestDB_PutWithTTL_Merge(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)
	for i := 0; i < 100; i++ {
		err = db.PutWithTTL(utils.GetTestKey(i), utils.RandomValue(10), time.Second*2)
		assert.Nil(t, err)
	}
	for i := 100; i < 150; i++ {
		err = db.PutWithTTL(utils.GetTestKey(i), utils.RandomValue(10), time.Second*20)
		assert.Nil(t, err)
	}
	time.Sleep(time.Second * 3)

	err = db.Merge(true)
	assert.Nil(t, err)

	for i := 0; i < 100; i++ {
		val, err := db.Get(utils.GetTestKey(i))
		assert.Nil(t, val)
		assert.Equal(t, err, ErrKeyNotFound)
	}
	for i := 100; i < 150; i++ {
		val, err := db.Get(utils.GetTestKey(i))
		assert.Nil(t, err)
		assert.NotNil(t, val)
	}
}

func TestDB_Expire(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	err = db.Put(utils.GetTestKey(1), utils.RandomValue(10))
	assert.Nil(t, err)

	err = db.Expire(utils.GetTestKey(1), time.Second*100)
	assert.Nil(t, err)
	tt1, err := db.TTL(utils.GetTestKey(1))
	assert.Nil(t, err)
	assert.True(t, tt1.Seconds() > 90)

	err = db.PutWithTTL(utils.GetTestKey(2), utils.RandomValue(10), time.Second*1)
	assert.Nil(t, err)

	tt2, err := db.TTL(utils.GetTestKey(2))
	assert.Nil(t, err)
	assert.True(t, tt2.Microseconds() > 500)

	err = db.Close()
	assert.Nil(t, err)

	db2, err := Open(options)
	assert.Nil(t, err)
	defer func() {
		_ = db2.Close()
	}()

	tt3, err := db2.TTL(utils.GetTestKey(1))
	assert.Nil(t, err)
	assert.True(t, tt3.Seconds() > 90)

	time.Sleep(time.Second)
	tt4, err := db2.TTL(utils.GetTestKey(2))
	assert.Equal(t, tt4, time.Duration(-1))
	assert.Equal(t, err, ErrKeyNotFound)
}

func TestDB_Expire2(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	// expire an expired key
	_ = db.PutWithTTL(utils.GetTestKey(1), utils.RandomValue(10), time.Second*1)
	_ = db.Put(utils.GetTestKey(2), utils.RandomValue(10))
	err = db.Expire(utils.GetTestKey(2), time.Second*2)
	assert.Nil(t, err)

	time.Sleep(time.Second * 2)
	_ = db.Close()

	db2, err := Open(options)
	assert.Nil(t, err)
	defer func() {
		_ = db2.Close()
	}()
	err = db2.Expire(utils.GetTestKey(1), time.Second)
	assert.Equal(t, err, ErrKeyNotFound)
	err = db2.Expire(utils.GetTestKey(2), time.Second)
	assert.Equal(t, err, ErrKeyNotFound)
}

func TestDB_DeleteExpiredKeys(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	for i := 0; i < 100001; i++ {
		err = db.PutWithTTL(utils.GetTestKey(i), utils.RandomValue(10), time.Second*1)
		assert.Nil(t, err)
	}

	// wait for key to expire
	time.Sleep(time.Second * 2)

	err = db.DeleteExpiredKeys(time.Second * 2)
	assert.Nil(t, err)
	assert.Equal(t, 0, db.Stat().KeysNum)

}

func TestDB_Multi_DeleteExpiredKeys(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	for i := 0; i < 3; i++ {
		for i := 0; i < 10000; i++ {
			err = db.Put(utils.GetTestKey(i), utils.RandomValue(10))
			assert.Nil(t, err)
		}
		for i := 10000; i < 100001; i++ {
			err = db.PutWithTTL(utils.GetTestKey(i), utils.RandomValue(10), time.Second*1)
			assert.Nil(t, err)
		}

		// wait for key to expire
		time.Sleep(time.Second * 2)

		err = db.DeleteExpiredKeys(time.Second * 2)
		assert.Nil(t, err)
		assert.Equal(t, 10000, db.Stat().KeysNum)
	}
}

func TestDB_Persist(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	// not exist
	err = db.Persist(utils.GetTestKey(1))
	assert.Equal(t, err, ErrKeyNotFound)

	err = db.PutWithTTL(utils.GetTestKey(1), utils.RandomValue(10), time.Second*1)
	assert.Nil(t, err)

	// exist
	err = db.Persist(utils.GetTestKey(1))
	assert.Nil(t, err)
	time.Sleep(time.Second * 2)
	// check ttl
	ttl, err := db.TTL(utils.GetTestKey(1))
	assert.Nil(t, err)
	assert.Equal(t, ttl, time.Duration(-1))
	val1, err := db.Get(utils.GetTestKey(1))
	assert.Nil(t, err)
	assert.NotNil(t, val1)

	// restart
	err = db.Close()
	assert.Nil(t, err)

	db2, err := Open(options)
	assert.Nil(t, err)
	defer func() {
		_ = db2.Close()
	}()

	ttl2, err := db2.TTL(utils.GetTestKey(1))
	assert.Nil(t, err)
	assert.Equal(t, ttl2, time.Duration(-1))
	val2, err := db2.Get(utils.GetTestKey(1))
	assert.Nil(t, err)
	assert.NotNil(t, val2)
}

func TestDB_Invalid_Cron_Expression(t *testing.T) {
	options := DefaultOptions
	options.AutoMergeCronExpr = "*/1 * * * * * *"
	_, err := Open(options)
	assert.NotNil(t, err)
}

func TestDB_Valid_Cron_Expression(t *testing.T) {
	options := DefaultOptions
	{
		options.AutoMergeCronExpr = "* */1 * * * *"
		db, err := Open(options)
		assert.Nil(t, err)
		destroyDB(db)
	}

	{
		options.AutoMergeCronExpr = "*/1 * * * *"
		db, err := Open(options)
		assert.Nil(t, err)
		destroyDB(db)
	}

	{
		options.AutoMergeCronExpr = "5 0 * 8 *"
		db, err := Open(options)
		assert.Nil(t, err)
		destroyDB(db)
	}

	{
		options.AutoMergeCronExpr = "*/2 14 1 * *"
		db, err := Open(options)
		assert.Nil(t, err)
		destroyDB(db)
	}

	{
		options.AutoMergeCronExpr = "@hourly"
		db, err := Open(options)
		assert.Nil(t, err)
		destroyDB(db)
	}
}

func TestDB_Auto_Merge(t *testing.T) {
	options := DefaultOptions
	db, err := Open(options)
	assert.Nil(t, err)
	defer destroyDB(db)

	for i := 0; i < 2000; i++ {
		delKey := utils.GetTestKey(rand.Int())
		err := db.Put(delKey, utils.RandomValue(128))
		assert.Nil(t, err)
		err = db.Put(utils.GetTestKey(rand.Int()), utils.RandomValue(2*KB))
		assert.Nil(t, err)
		err = db.Delete(delKey)
		assert.Nil(t, err)
	}

	{
		reader := db.dataFiles.NewReader()
		var keyCnt int
		for {
			if _, _, err := reader.Next(); errors.Is(err, io.EOF) {
				break
			}
			keyCnt++
		}
		// each record has one data wal and commit at end of batch with wal
		// so totally is 2000 * 3 * 2 = 12000
		assert.Equal(t, 12000, keyCnt)
	}

	mergeDirPath := mergeDirPath(options.DirPath)
	if _, err := os.Stat(mergeDirPath); err != nil {
		assert.True(t, os.IsNotExist(err))
	}
	assert.NoError(t, db.Close())

	{
		options.AutoMergeCronExpr = "* * * * * *" // every second
		db2, err := Open(options)
		assert.Nil(t, err)
		{
			<-time.After(time.Second * 2)
			reader := db2.dataFiles.NewReader()
			var keyCnt int
			for {
				if _, _, err := reader.Next(); errors.Is(err, io.EOF) {
					break
				}
				keyCnt++
			}
			// after merge records are only valid data, so totally is 2000
			assert.Equal(t, 2000, keyCnt)
		}
		_ = db2.Close()
	}
}