fix tests

Signed-off-by: Kevin Su <pingsutw@apache.org>
flyteorg · Jun 14, 2022 · 6615886 · 6615886
1 parent bbccd70
commit 6615886
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/go/tasks/plugins/k8s/pod/container_test.go b/go/tasks/plugins/k8s/pod/container_test.go
@@ -222,9 +222,12 @@ func TestContainerTaskExecutor_GetTaskStatus(t *testing.T) {
 	})
 
 	t.Run("raw container failed", func(t *testing.T) {
-		j.Annotations = map[string]string{RawContainerName: "raw-container"}
-		j.Status.ContainerStatuses = []v1.ContainerStatus{{Name: j.Annotations[RawContainerName], State: v1.ContainerState{Terminated: &v1.ContainerStateTerminated{ExitCode: 1}}}}
-		phaseInfo, err := DefaultPodPlugin.GetTaskPhase(ctx, nil, j)
+		pod := &v1.Pod{
+			Status: v1.PodStatus{},
+		}
+		pod.Annotations = map[string]string{RawContainerName: "raw-container"}
+		pod.Status.ContainerStatuses = []v1.ContainerStatus{{Name: pod.Annotations[RawContainerName], State: v1.ContainerState{Terminated: &v1.ContainerStateTerminated{ExitCode: 1}}}}
+		phaseInfo, err := DefaultPodPlugin.GetTaskPhase(ctx, nil, pod)
 		assert.NoError(t, err)
 		assert.NotNil(t, phaseInfo)
 		assert.Equal(t, pluginsCore.PhaseRetryableFailure, phaseInfo.Phase())