test(lru-worker): update lru worker test

This commit is contained in:
bandl 2021-11-11 23:00:21 +08:00
parent 25be4a1509
commit 400d620aa4
1 changed files with 6 additions and 6 deletions

View File

@ -5,7 +5,7 @@ import (
"testing" "testing"
"time" "time"
"gitee.com/wheat-os/wheatCache/pkg/event" "gitee.com/wheat-os/wheatCache/pkg/event2"
"gitee.com/wheat-os/wheatCache/pkg/logx" "gitee.com/wheat-os/wheatCache/pkg/logx"
"gitee.com/wheat-os/wheatCache/pkg/proto" "gitee.com/wheat-os/wheatCache/pkg/proto"
"gitee.com/wheat-os/wheatCache/pkg/structure/stringx" "gitee.com/wheat-os/wheatCache/pkg/structure/stringx"
@ -15,9 +15,9 @@ import (
func TestWorker(t *testing.T) { func TestWorker(t *testing.T) {
ctx := context.Background() ctx := context.Background()
lru := NewLRUCache() lru := NewLRUCache()
produce := event.NewProduce(lru.GetDriver()) produce := event2.NewProduce(lru.GetDriver())
workEvent := produce.NewEvent(OptionEventName) workEvent := produce.NewEvent(OptionEventName)
workEvent.SetValue(WorkFuncEventKey, event.EventWorkFunc(func() (interface{}, error) { workEvent.SetValue(event2.WorkFuncEventKey, event2.EventWorkFunc(func() (interface{}, error) {
v1 := stringx.NewStringSingle() v1 := stringx.NewStringSingle()
key := proto.BaseKey{ key := proto.BaseKey{
Key: "v1", Key: "v1",
@ -36,11 +36,11 @@ func TestWorker(t *testing.T) {
func TestSingleCache_DelToClearSize(t *testing.T) { func TestSingleCache_DelToClearSize(t *testing.T) {
ctx := context.Background() ctx := context.Background()
lru := NewLRUCache() lru := NewLRUCache()
produce := event.NewProduce(lru.GetDriver()) produce := event2.NewProduce(lru.GetDriver())
for i := int32(20000); i > 0; i-- { for i := int32(20000); i > 0; i-- {
workEvent := produce.NewEvent(OptionEventName) workEvent := produce.NewEvent(OptionEventName)
workEvent.SetValue(WorkFuncEventKey, event.EventWorkFunc(func() (interface{}, error) { workEvent.SetValue(event2.WorkFuncEventKey, event2.EventWorkFunc(func() (interface{}, error) {
v1 := stringx.NewStringSingle() v1 := stringx.NewStringSingle()
key := proto.BaseKey{ key := proto.BaseKey{
Key: string(i), Key: string(i),
@ -53,7 +53,7 @@ func TestSingleCache_DelToClearSize(t *testing.T) {
workEvent.InitWaitEvent() workEvent.InitWaitEvent()
produce.Call(ctx, workEvent) produce.Call(ctx, workEvent)
workEvent.StartWaitEvent(2 * time.Second) workEvent.StartWaitEvent(2 * time.Second)
produce.Recovery(workEvent) workEvent.Recovery()
} }
logx.Info("start size is %d", lru.nowSize) logx.Info("start size is %d", lru.nowSize)