feat: test

Author: abenio

pipeline_deduplication_test.go

@@ -283,8 +283,7 @@ func TestDeduplicationPipelineSyncPerform(t *testing.T) {
 
 // TestDeduplicationPipelineDataChanClosed 测试数据通道关闭后的行为
 func TestDeduplicationPipelineDataChanClosed(t *testing.T) {
-	ctx, cancel := context.WithTimeout(context.Background(), time.Second*2)
-	defer cancel()
+	ctx := context.Background()
 
 	var mux sync.Mutex
 	var processedCount int
@@ -323,6 +322,8 @@ func TestDeduplicationPipelineDataChanClosed(t *testing.T) {
 	}
 	close(dataChan) // 关闭数据通道
 
+	time.Sleep(time.Second) // 等待数据发送完毕
+
 	// 等待处理完成
 	select {
 	case err := <-performDone:

pipeline_standard_test.go

@@ -305,8 +305,7 @@ func TestStandardPipelineSyncPerform(t *testing.T) {
 
 // TestStandardPipelineDataChanClosed 测试数据通道关闭后的行为
 func TestStandardPipelineDataChanClosed(t *testing.T) {
-	ctx, cancel := context.WithTimeout(context.Background(), time.Second*2)
-	defer cancel()
+	ctx := context.Background()
 
 	var mux sync.Mutex
 	var processedCount int
@@ -337,7 +336,8 @@ func TestStandardPipelineDataChanClosed(t *testing.T) {
 	for i := 0; i < 100; i++ {
 		dataChan <- i
 	}
-	close(dataChan) // 关闭数据通道
+	close(dataChan)         // 关闭数据通道
+	time.Sleep(time.Second) // 等待数据发送完毕
 
 	// 等待处理完成
 	select {