fix: fix tmq poll tests

taosdata · Jan 6, 2025 · 4fca8e2 · 4fca8e2
1 parent c027d2f
commit 4fca8e2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/controller/ws/tmq/tmq_test.go b/controller/ws/tmq/tmq_test.go
@@ -3380,7 +3380,7 @@ func TestPollError(t *testing.T) {
 	assert.NoError(t, err)
 	assert.Equal(t, 0, pollResp.Code, string(msg))
 	// sleep
-	time.Sleep(time.Second * 20)
+	time.Sleep(time.Second * 5)
 	// poll
 	b, _ = json.Marshal(TMQPollReq{ReqID: 1, BlockingTime: 500})
 	msg, err = doWebSocket(ws, TMQPoll, b)

diff --git a/db/asynctmq/tmqhandle/handler_test.go b/db/asynctmq/tmqhandle/handler_test.go
@@ -24,8 +24,9 @@ func TestTMQHandlerPool(t *testing.T) {
 
 		caller := handler.Caller
 		pollRes := unsafe.Pointer(&struct{}{})
-		caller.PollCall(pollRes)
-		if <-caller.PollResult != pollRes {
+		caller.PollCall(pollRes, 0, "")
+		res := <-caller.PollResult
+		if res.Res != pollRes {
 			t.Errorf("PollCall failed")
 		}
 
@@ -72,8 +73,9 @@ func TestTMQCaller(t *testing.T) {
 	// Test PollCall
 	a := 1
 	res := unsafe.Pointer(&a)
-	caller.PollCall(res)
-	if <-caller.PollResult != res {
+	caller.PollCall(res, 0, "")
+	pollRes := <-caller.PollResult
+	if pollRes.Res != res {
 		t.Error("PollCall failed")
 	}