diff --git a/tests/test_slurm_cli.py b/tests/test_slurm_cli.py index 025dbd19300c74dc27bc2f5f0d2edc70bbeaaa09..fa76f2643d4febc4e1aca3ff8e83d63db96d6f90 100644 --- a/tests/test_slurm_cli.py +++ b/tests/test_slurm_cli.py @@ -44,6 +44,13 @@ class TestSlurmControl(unittest.TestCase): with self.assertRaises(slurm_control.SlurmCallError): _ = slurm_control.get_jobs_status() + @patch('slurm_cli.slurm_control.run_process') + def test_get_job_statuses_job_id_list_missing(self, call_mock): + output_string = '1;test_job;CD;COMPLETED;None\n2;test_job;F;FAILED;error 1\n' + call_mock.return_value = get_mocked_call_output(0, output_string, 'stuff') + _ = slurm_control.get_jobs_status(['1', '2']) + self.assertIn('1,2', call_mock.call_args[0][0]) + if __name__ == '__main__': logging.basicConfig(level=logging.DEBUG, format='%(asctime)s-%(name)s-%(levelname)s %(message)s') \ No newline at end of file