vaedb_test.go

/**
 * User: coder.sdp@gmail.com
 * Date: 2023/8/24
 * Time: 12:00
 */

package vaedb

import (
	"fmt"
	"strconv"
	"testing"
)

//t
var cnt = 100000

const keyPrefix = "test"

var db, _ = NewVaeDB(".")

func TestBigSet(t *testing.T) {
	for i := 0; i < cnt; i++ {
		k := keyPrefix + strconv.Itoa(i)
		fmt.Println(db.Set(k, []byte(k+"tv")))
	}
}

func TestBigGet(t *testing.T) {
	for i := 0; i < cnt; i++ {
		k := keyPrefix + strconv.Itoa(i)
		v := string(db.Get(k))
		if v != k+"tv" {
			t.Errorf("err k:%s v%s", k, v)
		}
	}
}

func TestSetAndGet(t *testing.T) {
	//set
	t.Run("c_set", func(t *testing.T) {
		t.Parallel()
		for i := 0; i < cnt; i++ {
			k := keyPrefix + strconv.Itoa(i)
			db.Set(k, []byte(k+"tv"))
		}
	})
	//get
	t.Run("c_get", func(t *testing.T) {
		t.Parallel()
		for i := 0; i < cnt; i++ {
			k := keyPrefix + strconv.Itoa(i)
			v := string(db.Get(k))
			if v != "" && v != k+"tv" {
				t.Errorf("err k:%s v%s", k, v)
			}
		}
	})
}

//get拦截器测试
func TestSetGetInterceptor(t *testing.T) {
	getInter := func(chain *Chain) {
		key := chain.Key
		chain.Key = "test2"
		t.Logf("获取到原key %s,并更改Key %s", key, chain.Key)
		chain.Next()
		t.Logf("获取到key %s,val %s", chain.Key, chain.Key)
	}
	db, _ = NewVaeDB(".", SetGetInterceptor(getInter))
	db.Get(keyPrefix + "1")
}

//set拦截器测试
func TestSetSetInterceptor(t *testing.T) {
	setInter := func(chain *Chain) {
		key := chain.Key
		chain.Key = "test2"
		t.Logf("获取到原key %s,并更改Key %s", key, chain.Key)
		chain.Next()
	}
	db, _ = NewVaeDB(".", SetSetInterceptor(setInter))
	db.Set("test1", []byte("spe"))
	t.Logf("获取 test1 %s", db.Get("test1"))
	t.Logf("获取 test2 %s", db.Get("test2"))
}