Correct testcase of "retry"

tektoncd · Feb 4, 2020 · 446606e · 446606e
1 parent aee09da
commit 446606e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/retry_test.go b/test/retry_test.go
@@ -99,6 +99,7 @@ func TestTaskRunRetry(t *testing.T) {
 		}
 		podNames[r.PodName] = struct{}{}
 	}
+	podNames[tr.Status.PodName] = struct{}{}
 	if len(tr.Status.RetriesStatus) != numRetries {
 		t.Errorf("TaskRun %q had %d retriesStatuses, want %d", tr.Name, len(tr.Status.RetriesStatus), numRetries)
 	}
@@ -107,9 +108,9 @@ func TestTaskRunRetry(t *testing.T) {
 	pods, err := c.KubeClient.Kube.CoreV1().Pods(namespace).List(metav1.ListOptions{})
 	if err != nil {
 		t.Fatalf("Failed to list Pods: %v", err)
-	} else if len(pods.Items) != numRetries {
+	} else if len(pods.Items) != numRetries+1 {
 		// TODO: Make this an error.
-		t.Logf("BUG: Found %d Pods, want %d", len(pods.Items), numRetries)
+		t.Logf("BUG: Found %d Pods, want %d", len(pods.Items), numRetries+1)
 	}
 	for _, p := range pods.Items {
 		if _, found := podNames[p.Name]; !found {