update unit test

Signed-off-by: Paul Dittamo <[email protected]>
flyteorg · Dec 14, 2023 · cc4c99b · cc4c99b
1 parent 08056f3
commit cc4c99b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/flyteplugins/go/tasks/plugins/array/k8s/management_test.go b/flyteplugins/go/tasks/plugins/array/k8s/management_test.go
@@ -679,9 +679,11 @@ func TestTerminateSubTasks(t *testing.T) {
 			assert.Nil(t, err)
 			assert.Equal(t, uint32(1), nextState.PhaseVersion)
 			assert.Equal(t, arrayCore.PhaseWriteToDiscoveryThenFail, nextState.CurrentPhase)
-			assert.Len(t, externalResources, subtaskCount)
+			assert.Len(t, externalResources, terminateCounter)
+
 			for _, externalResource := range externalResources {
-				assert.True(t, externalResource.IsAbortedSubtask || core.Phases[externalResource.Phase].IsTerminal() || core.Phases[externalResource.Phase] == core.PhaseUndefined)
+				phase := core.Phases[externalResource.Phase]
+				assert.True(t, phase.IsAborted())
 			}
 		})
 	}