wheat-cache/pkg/lru/woker_test.go

package lru

import (
	"context"
	"gitee.com/timedb/wheatCache/pkg/errorx"
	"gitee.com/timedb/wheatCache/pkg/event"
	"gitee.com/timedb/wheatCache/pkg/proto"
	"gitee.com/timedb/wheatCache/pkg/structure"
	"gitee.com/timedb/wheatCache/pkg/structure/stringx"
	"github.com/stretchr/testify/require"
	"testing"
	"time"
)

func TestWorker(t *testing.T) {
	ctx := context.Background()
	lru := NewLRUCache()
	produce := event.NewProduce(lru.GetDriver())
	workEvent := event.NewEvent(OptionEventName)
	workEvent.SetValue(WorkFuncEventKey, event.EventWorkFunc(func() (interface{}, error) {
		v1 := stringx.NewStringSingle()
		key := "v1"
		res, err := v1.Set(&proto.SetRequest{
			Val: "123",
		})
		if err != nil {
			return nil, err
		}
		lru.Add(key, v1)
		return res.Result, nil
	}))
	workEvent.InitWaitEvent()
	produce.Call(ctx,workEvent)
	res, err := workEvent.StartWaitEvent(2 * time.Second)
	require.NoError(t, err)
	require.Equal(t, res, "123")

	workEvent.InitWaitEvent()
	workEvent.SetValue(WorkFuncEventKey, event.EventWorkFunc(func() (interface{}, error) {
		v2, ok := lru.Get("v1")
		if !ok{
			return nil, errorx.New("no this key")
		}
		switch v2.(type) {
		case structure.StringXInterface:
			res, err := v2.(structure.StringXInterface).Get(&proto.GetRequest{
			})
			if err != nil {
				return nil, err
			}
			return res.Result, nil
		default:
			return nil, errorx.New("no this type")
		}
	}))
	produce.Call(ctx, workEvent)
	res, err = workEvent.StartWaitEvent(2 * time.Second)
	require.NoError(t, err)
	require.Equal(t, res, "123")

	workEvent.InitWaitEvent()
	workEvent.SetValue(WorkFuncEventKey, event.EventWorkFunc(func() (interface{}, error) {
		lru.Del()
		v2, ok := lru.Get("v1")
		if !ok{
			return nil, nil
		}
		switch v2.(type) {
		case structure.StringXInterface:
			res, err := v2.(structure.StringXInterface).Get(&proto.GetRequest{
			})
			if err != nil {
				return nil, err
			}
			return res.Result, nil
		default:
			return nil, errorx.New("no this type")
		}
	}))
	produce.Call(ctx, workEvent)
	res, err = workEvent.StartWaitEvent(2 * time.Second)
	require.Equal(t, err, nil)
	require.Equal(t, res, nil)

}