diff --git a/python/services/monitorservice/dmod/test/test_monitor_service.py b/python/services/monitorservice/dmod/test/test_monitor_service.py index dee019f93..5b26db925 100644 --- a/python/services/monitorservice/dmod/test/test_monitor_service.py +++ b/python/services/monitorservice/dmod/test/test_monitor_service.py @@ -131,9 +131,9 @@ def setUp(self) -> None: self._connect_metadata_examples.append('{"purpose": "PROMPT", "additional_metadata": false}') self._connect_metadata_examples.append('{"purpose": "CONNECT", "additional_metadata": true}') - self._jobs[0].status = JobStatus(JobExecPhase.MODEL_EXEC, JobExecStep.AWAITING_ALLOCATION) - self._jobs[1].status = JobStatus(JobExecPhase.MODEL_EXEC, JobExecStep.AWAITING_SCHEDULING) - self._jobs[2].status = JobStatus(JobExecPhase.MODEL_EXEC, JobExecStep.SCHEDULED) + self._jobs[0].set_status(JobStatus(JobExecPhase.MODEL_EXEC, JobExecStep.AWAITING_ALLOCATION)) + self._jobs[1].set_status(JobStatus(JobExecPhase.MODEL_EXEC, JobExecStep.AWAITING_SCHEDULING)) + self._jobs[2].set_status(JobStatus(JobExecPhase.MODEL_EXEC, JobExecStep.SCHEDULED)) self._services.append(MockMonitorService(MockMonitor(monitored_jobs=self._jobs))) @@ -141,9 +141,9 @@ def setUp(self) -> None: for j in self._jobs: self._original_statuses.append(j.status) - self._jobs[0].status = JobStatus(JobExecPhase.MODEL_EXEC, JobExecStep.AWAITING_SCHEDULING) - self._jobs[1].status = JobStatus(JobExecPhase.MODEL_EXEC, JobExecStep.SCHEDULED) - self._jobs[2].status = JobStatus(JobExecPhase.MODEL_EXEC, JobExecStep.RUNNING) + self._jobs[0].set_status(JobStatus(JobExecPhase.MODEL_EXEC, JobExecStep.AWAITING_SCHEDULING)) + self._jobs[1].set_status(JobStatus(JobExecPhase.MODEL_EXEC, JobExecStep.SCHEDULED)) + self._jobs[2].set_status(JobStatus(JobExecPhase.MODEL_EXEC, JobExecStep.RUNNING)) self._job_ids = list() self._jobs_by_id = dict() @@ -947,7 +947,7 @@ def test_run_monitor_check_3_a(self): get_next_one = False for status in self._generate_all_status_values(): if get_next_one: - job.status = status + job.set_status(status) break elif status == original_status: get_next_one = True @@ -978,7 +978,7 @@ def test_run_monitor_check_3_b(self): for status in self._generate_all_status_values(): if get_next_one: - job.status = status + job.set_status(status) break elif status == original_status: get_next_one = True @@ -1004,7 +1004,7 @@ def test_run_monitor_check_3_c(self): get_next_one = False for status in self._generate_all_status_values(): if get_next_one: - job.status = status + job.set_status(status) break elif status == original_status: get_next_one = True