Fixes from rename

armadaproject · Apr 13, 2023 · 4b54e5d · 4b54e5d
1 parent bcf42bc
commit 4b54e5d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/internal/scheduler/scheduler_test.go b/internal/scheduler/scheduler_test.go
@@ -404,7 +404,7 @@ func TestScheduler_TestCycle(t *testing.T) {
 				fmt.Sprintf("%T", &armadaevents.EventSequence_Event_CancelledJob{}):     stringSet(tc.expectedJobCancelled),
 				fmt.Sprintf("%T", &armadaevents.EventSequence_Event_ReprioritisedJob{}): stringSet(tc.expectedJobReprioritised),
 				fmt.Sprintf("%T", &armadaevents.EventSequence_Event_JobSucceeded{}):     stringSet(tc.expectedJobSucceeded),
-				fmt.Sprintf("%T", &armadaevents.EventSequence_Event_RequeueJob{}):       stringSet(tc.expectedRequeued),
+				fmt.Sprintf("%T", &armadaevents.EventSequence_Event_JobRequeued{}):      stringSet(tc.expectedRequeued),
 			}
 			err = subtractEventsFromOutstandingEventsByType(publisher.events, outstandingEventsByType)
 			require.NoError(t, err)

diff --git a/internal/scheduleringester/instructions_test.go b/internal/scheduleringester/instructions_test.go
@@ -170,11 +170,11 @@ func TestConvertSequence(t *testing.T) {
 			expected: []DbOperation{
 				UpdateJobQueuedState{f.JobIdString: &JobQueuedStateUpdate{
 					Queued:             true,
-					QueuedStateVersion: f.JobRequeued.GetRequeueJob().UpdateSequenceNumber,
+					QueuedStateVersion: f.JobRequeued.GetJobRequeued().UpdateSequenceNumber,
 				}},
 				UpdateJobSchedulingInfo{f.JobIdString: &JobSchedulingInfoUpdate{
-					JobSchedulingInfo:        protoutil.MustMarshall(f.JobRequeued.GetRequeueJob().SchedulingInfo),
-					JobSchedulingInfoVersion: int32(f.JobRequeued.GetRequeueJob().SchedulingInfo.Version),
+					JobSchedulingInfo:        protoutil.MustMarshall(f.JobRequeued.GetJobRequeued().SchedulingInfo),
+					JobSchedulingInfoVersion: int32(f.JobRequeued.GetJobRequeued().SchedulingInfo.Version),
 				}},
 			},
 		},