From 93a81fd558e6a8e00547e2930b0929ac5e5c502f Mon Sep 17 00:00:00 2001
From: yusing <yusing@6uo.me>
Date: Thu, 29 May 2025 20:42:36 +0800
Subject: [PATCH] fix(task): fix task tests

---
 internal/task/task_test.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/internal/task/task_test.go b/internal/task/task_test.go
index 091ef8e..d99fb5e 100644
--- a/internal/task/task_test.go
+++ b/internal/task/task_test.go
@@ -49,7 +49,7 @@ func TestTaskStuck(t *testing.T) {
 	})
 	done := make(chan struct{})
 	go func() {
-		task.Finish(nil)
+		task.FinishAndWait(nil)
 		close(done)
 	}()
 	time.Sleep(time.Millisecond * 100)
@@ -81,7 +81,7 @@ func TestTaskOnCancelOnFinished(t *testing.T) {
 	})
 
 	expect.False(t, shouldTrueOnFinish)
-	task.Finish(nil)
+	task.FinishAndWait(nil)
 	expect.True(t, shouldTrueOnCancel)
 	expect.True(t, shouldTrueOnFinish)
 }
@@ -97,7 +97,7 @@ func TestCommonFlowWithGracefulShutdown(t *testing.T) {
 	})
 
 	go func() {
-		defer task.Finish(nil)
+		defer task.FinishAndWait(nil)
 		for {
 			select {
 			case <-task.Context().Done():