Check for number of workers before soft failing the task. #104195
@ -351,7 +351,7 @@ func TestJobFailureAfterWorkerTaskFailure(t *testing.T) {
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
mf.config.EXPECT().Get().Return(&conf).AnyTimes()
|
mf.config.EXPECT().Get().Return(&conf).Times(2)
|
||||||
|
|
||||||
mf.persistence.EXPECT().FetchTask(gomock.Any(), taskID).Return(&mockTask, nil)
|
mf.persistence.EXPECT().FetchTask(gomock.Any(), taskID).Return(&mockTask, nil)
|
||||||
|
|
||||||
@ -363,7 +363,7 @@ func TestJobFailureAfterWorkerTaskFailure(t *testing.T) {
|
|||||||
mf.persistence.EXPECT().AddWorkerToTaskFailedList(gomock.Any(), &mockTask, &worker).Return(1, nil)
|
mf.persistence.EXPECT().AddWorkerToTaskFailedList(gomock.Any(), &mockTask, &worker).Return(1, nil)
|
||||||
|
|
||||||
mf.persistence.EXPECT().WorkersLeftToRun(gomock.Any(), &mockJob, "misc").
|
mf.persistence.EXPECT().WorkersLeftToRun(gomock.Any(), &mockJob, "misc").
|
||||||
Return(map[string]bool{"60453eec-5a26-43e9-9da2-d00506d492cc": true}, nil)
|
Return(map[string]bool{"e7632d62-c3b8-4af0-9e78-01752928952c": true}, nil)
|
||||||
mf.persistence.EXPECT().FetchTaskFailureList(gomock.Any(), &mockTask).
|
mf.persistence.EXPECT().FetchTaskFailureList(gomock.Any(), &mockTask).
|
||||||
Return([]*persistence.Worker{ /* It shouldn't matter whether the failing worker is here or not. */ }, nil)
|
Return([]*persistence.Worker{ /* It shouldn't matter whether the failing worker is here or not. */ }, nil)
|
||||||
|
|
||||||
@ -377,6 +377,7 @@ func TestJobFailureAfterWorkerTaskFailure(t *testing.T) {
|
|||||||
mf.stateMachine.EXPECT().
|
mf.stateMachine.EXPECT().
|
||||||
JobStatusChange(gomock.Any(), &mockJob, api.JobStatusFailed, "no more workers left to run tasks of type \"misc\"")
|
JobStatusChange(gomock.Any(), &mockJob, api.JobStatusFailed, "no more workers left to run tasks of type \"misc\"")
|
||||||
|
|
||||||
|
// Do the call
|
||||||
echoCtx := mf.prepareMockedJSONRequest(taskUpdate)
|
echoCtx := mf.prepareMockedJSONRequest(taskUpdate)
|
||||||
requestWorkerStore(echoCtx, &worker)
|
requestWorkerStore(echoCtx, &worker)
|
||||||
err := mf.flamenco.TaskUpdate(echoCtx, taskID)
|
err := mf.flamenco.TaskUpdate(echoCtx, taskID)
|
||||||
|
Loading…
Reference in New Issue
Block a user