Add timeout to shouldend test

0xPolygon · May 19, 2023 · 80ba145 · 80ba145
1 parent 10b3199
commit 80ba145
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/helper/common/common_test.go b/helper/common/common_test.go
@@ -117,7 +117,7 @@ func TestRetryForever_AlwaysReturnError_ShouldNeverEnd(t *testing.T) {
 }
 
 func TestRetryForever_ReturnNilAfterFirstRun_ShouldEnd(t *testing.T) {
-	ctx, cancel := context.WithCancel(context.Background())
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second)
 	RetryForever(ctx, time.Millisecond*100, func(ctx context.Context) error {
 		select {
 		case <-ctx.Done():