[logstream] test polling should use a different context

knative · Jun 21, 2024 · f006890 · f006890
1 parent 915c009
commit f006890
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/test/logstream/v2/stream_test.go b/test/logstream/v2/stream_test.go
@@ -304,8 +304,9 @@ func TestNamespaceStream(t *testing.T) {
 
 	// We can't assume that the cancel signal doesn't race the pod creation signal, so
 	// we retry a few times to give some leeway.
-	if err := wait.PollUntilContextTimeout(ctx, 10*time.Millisecond, time.Second, true, func(ctx context.Context) (bool, error) {
-		if _, err := podClient.Create(context.Background(), knativePod, metav1.CreateOptions{}); err != nil {
+	pollCtx := context.Background()
+	if err := wait.PollUntilContextTimeout(pollCtx, 10*time.Millisecond, time.Second, true, func(ctx context.Context) (bool, error) {
+		if _, err := podClient.Create(pollCtx, knativePod, metav1.CreateOptions{}); err != nil {
 			return false, err
 		}
 
@@ -314,7 +315,7 @@ func TestNamespaceStream(t *testing.T) {
 			return true, nil
 		case <-logFuncInvoked:
 			t.Log("Log was still produced, trying again...")
-			if err := podClient.Delete(context.Background(), knativePod.Name, metav1.DeleteOptions{}); err != nil {
+			if err := podClient.Delete(pollCtx, knativePod.Name, metav1.DeleteOptions{}); err != nil {
 				return false, err
 			}
 			return false, nil