use correct args to reschedule the monitor job

zooniverse · Feb 23, 2023 · c55e6c7 · c55e6c7
1 parent a66fff2
commit c55e6c7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/app/sidekiq/training_job_monitor_job.rb b/app/sidekiq/training_job_monitor_job.rb
@@ -36,7 +36,7 @@ def perform(training_job_id, context_id)
       PredictionManifestExportJob.perform_async(context.id)
     else
       # reschedule this job to run again in 1 minute
-      TrainingJobMonitorJob.perform_in(MONITOR_JOB_RESCHEDULE_DELAY.minute, training_job.id)
+      TrainingJobMonitorJob.perform_in(MONITOR_JOB_RESCHEDULE_DELAY.minute, training_job.id, context.id)
     end
   end
 end
diff --git a/spec/sidekiq/training_job_monitor_job_spec.rb b/spec/sidekiq/training_job_monitor_job_spec.rb
@@ -34,7 +34,7 @@
       allow(training_job_monitor_result).to receive(:completed?).and_return(false)
       allow(described_class).to receive(:perform_in)
       job.perform(training_job.id, context.id)
-      expect(described_class).to have_received(:perform_in).with(1.minute, training_job_monitor_result.id)
+      expect(described_class).to have_received(:perform_in).with(1.minute, training_job_monitor_result.id, context.id)
     end
 
     context 'when the monitor job returns a completed job' do