fill in JobRunIdsToCancelStr on executor api (#3953)

Signed-off-by: Chris Martin <chris@cmartinit.co.uk>
armadaproject · Sep 19, 2024 · bfc4975 · bfc4975
1 parent f9a6a6d
commit bfc4975
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/internal/scheduler/api.go b/internal/scheduler/api.go
@@ -128,6 +128,7 @@ func (srv *ExecutorApi) LeaseJobRuns(stream executorapi.ExecutorApi_LeaseJobRuns
 					JobRunIdsToCancel: slices.Map(runsToCancel, func(x string) *armadaevents.Uuid {
 						return armadaevents.MustProtoUuidFromUuidString(x)
 					}),
+					JobRunIdsToCancelStr: runsToCancel,
 				},
 			},
 		}); err != nil {

diff --git a/internal/scheduler/api_test.go b/internal/scheduler/api_test.go
@@ -203,7 +203,8 @@ func TestExecutorApi_LeaseJobRuns(t *testing.T) {
 			expectedMsgs: []*executorapi.LeaseStreamMessage{
 				{
 					Event: &executorapi.LeaseStreamMessage_CancelRuns{CancelRuns: &executorapi.CancelRuns{
-						JobRunIdsToCancel: []*armadaevents.Uuid{armadaevents.MustProtoUuidFromUuidString(runId2)},
+						JobRunIdsToCancel:    []*armadaevents.Uuid{armadaevents.MustProtoUuidFromUuidString(runId2)},
+						JobRunIdsToCancelStr: []string{runId2},
 					}},
 				},
 				{