update awsbatch array unit test

Signed-off-by: Paul Dittamo <[email protected]>
flyteorg · Dec 13, 2023 · ccd31ce · ccd31ce
1 parent fadec71
commit ccd31ce
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flyteplugins/go/tasks/plugins/array/awsbatch/monitor_test.go b/flyteplugins/go/tasks/plugins/array/awsbatch/monitor_test.go
@@ -248,7 +248,7 @@ func TestCheckSubTasksState(t *testing.T) {
 
 		newState, err := CheckSubTasksState(ctx, tCtx, jobStore, &config.Config{}, &State{
 			State: &arrayCore.State{
-				CurrentPhase:         arrayCore.PhaseWriteToDiscoveryThenFail,
+				CurrentPhase:         arrayCore.PhaseCheckingSubTaskExecutions,
 				ExecutionArraySize:   2,
 				OriginalArraySize:    2,
 				OriginalMinSuccesses: 2,
@@ -264,6 +264,6 @@ func TestCheckSubTasksState(t *testing.T) {
 
 		assert.NoError(t, err)
 		p, _ := newState.GetPhase()
-		assert.Equal(t, arrayCore.PhaseWriteToDiscoveryThenFail, p)
+		assert.Equal(t, arrayCore.PhaseAbortSubTasks, p)
 	})
 }