Merge branch 'vsukhin/feature/stream-service-logs' into sandbox/twrc

kubeshop · Nov 29, 2024 · 17b1c82 · 17b1c82
2 parents c1c77dc + 3e2c45f
commit 17b1c82
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/pkg/agent/testworkflows.go b/pkg/agent/testworkflows.go
@@ -243,9 +243,6 @@ func (ag *Agent) executeWorkflowNotificationsRequest(ctx context.Context, req *c
 }
 
 func (ag *Agent) executeWorkflowServiceNotificationsRequest(ctx context.Context, req *cloud.TestWorkflowServiceNotificationsRequest) error {
-	timeoutCtx, cancel := context.WithTimeout(ctx, serviceWaitTimeout)
-	defer cancel()
-
 	notificationsCh, err := retry.DoWithData(
 		func() (<-chan testkube.TestWorkflowExecutionNotification, error) {
 			// We have a race condition here
@@ -256,11 +253,10 @@ func (ag *Agent) executeWorkflowServiceNotificationsRequest(ctx context.Context,
 		},
 		retry.DelayType(retry.FixedDelay),
 		retry.Delay(logRetryDelay),
-		retry.Context(timeoutCtx),
 		retry.RetryIf(func(err error) bool {
 			return errors.Is(err, registry.ErrResourceNotFound)
 		}),
-		retry.Attempts(0),
+		retry.UntilSucceeded(),
 	)
 
 	if err != nil {