Add timeout to shouldend test

0xPolygon · May 16, 2023 · d75e370 · d75e370
1 parent 99a1d8a
commit d75e370
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():