wheat-cache/pkg/event/event_test.go

package event

import (
	"context"
	"fmt"
	"gitee.com/timedb/wheatCache/pkg/errorx"
	"testing"
	"time"

	"github.com/stretchr/testify/require"
)

const testEvent = "1001"
const waitTestEvent = "1002"

// 简单 非等待响应模式， 使用 event driver
func TestEvent_DriverEventTest(t *testing.T) {
	ctx := context.Background()
	driver := NewDriver(500)
	produce := NewProduce(driver)
	consumer := NewConsumer(driver)

	go produceEvent(t, ctx, produce)
	consumerEvent(t, ctx, consumer)
}

func produceEvent(t *testing.T, ctx context.Context, v ProduceInterface) {
	for i := 0; i < 100; i++ {
		event := NewEvent(testEvent)
		event.SetValue("test", i)
		v.Call(ctx, event)
	}
}

func consumerEvent(t *testing.T, ctx context.Context, v ConsumerInterface) {
	for i := 0; i < 100; i++ {
		event := v.Receive(ctx)
		res, ok := event.GetValue("test")
		require.True(t, ok)
		fmt.Println(res)
		require.Equal(t, res, i)
	}
}

// 响应等待用法
func TestEvent_SpanWaitEvent(t *testing.T) {
	ctx := context.Background()
	driver := NewDriver(500)
	produce := NewProduce(driver)
	consumer := NewConsumer(driver)

	go waitConsumer(t, ctx, consumer)

	waitProduce(t, ctx, produce)
}

func waitProduce(t *testing.T, ctx context.Context, v ProduceInterface) {
	for i := 0; i < 100; i++ {
		event := NewEvent(waitTestEvent)

		event.InitWaitEvent()
		event.SetValue("test", i)
		v.Call(ctx, event)                                // 推送给 consumer
		res, err := event.StartWaitEvent(2 * time.Second) // 最多等待 consumer 回复 2s
		require.NoError(t, err)
		require.Equal(t, fmt.Sprintf("test:%v", i), res)
	}
}

func waitConsumer(t *testing.T, ctx context.Context, v ConsumerInterface) {
	for i := 0; i < 100; i++ {
		event := v.Receive(ctx) // 接受 produce 的 event
		res, ok := event.GetValue("test")
		require.True(t, ok)
		require.Equal(t, res, i)

		// 发送返回值给 produce
		event.ExecWorkAndSendResult(func() (interface{}, error) {
			return fmt.Sprintf("test:%v", res), nil
		})
	}
}

func TestEvent_SetResultErr(t *testing.T) {
	ctx := context.Background()
	event := NewEvent("dddd")
	driver := NewDriver(100)
	produce := NewProduce(driver)
	consumer := NewConsumer(driver)
	go func() {
		event := consumer.Receive(ctx)
		event.SetResultErr(errorx.New("err"))
	}()
	event.InitWaitEvent()
	produce.Call(ctx, event)
	_, err := event.StartWaitEvent(2 * time.Second)
	fmt.Println(err)
	require.Error(t, err)
}