def test_submit_job(self):
scheduler = EmopSLURM(self.settings)
expected_cmd = [
"sbatch", "--parsable",
"-p", "stakeholder,stakeholder-4g",
"-J", "emop-controller",
"-o", "/dne/emop-controller-%j.out",
"--mem-per-cpu", "4000",
"--cpus-per-task", "1",
os.path.join(self.settings.emop_home, "batch_scripts/emop.slrm")
]
self.mock_rv.communicate.return_value[0] = "1"
retval = scheduler.submit_job('0001', '1')
args, kwargs = self.mock_popen.call_args
PROC_ID = os.environ.get('PROC_ID')
EMOP_CONFIG_PATH = os.environ.get('EMOP_CONFIG_PATH')
self.assertTrue(self.mock_popen.called)
self.assertEqual(expected_cmd, args[0])
self.assertEqual(PROC_ID, '0001')
self.assertEqual(EMOP_CONFIG_PATH, self.settings.config_path)
self.assertTrue(retval)
def test_submit_job_failed(self):
scheduler = EmopSLURM(self.settings)
self.mock_rv.communicate.return_value[0] = "1"
self.mock_rv.returncode = 1
retval = scheduler.submit_job('0001', '1')
self.assertFalse(retval)