From c683aa88caedb67d19fba782f3b115b311dc99d0 Mon Sep 17 00:00:00 2001
From: mancini <mancini@astron.nl>
Date: Tue, 17 Sep 2019 14:34:51 +0200
Subject: [PATCH] add test
---
tests/test_slurm_cli.py | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/tests/test_slurm_cli.py b/tests/test_slurm_cli.py
index 025dbd1..fa76f26 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
--
GitLab