def setUp(self):
from slurmy import JobHandler, test_mode
test_mode(True)
self.test_dir = os.path.join(options.Main.workdir, 'slurmy_unittest/jobconfig')
self.jh = JobHandler(work_dir = self.test_dir, verbosity = 0, name = 'test_jobconfig', do_snapshot = False)
self.run_script = 'echo "test"'
self.run_script_trigger = '@SLURMY.FINISHED; @SLURMY.SUCCESS;'
def test_local_listener(self):
from slurmy import JobHandler, Status, Type, test_mode
test_mode(True)
jh = JobHandler(work_dir=self.test_dir,
verbosity=0,
name='test_local_listener',
local_max=1)
jh.add_job(run_script=self.run_script_fail,
name='test',
job_type=Type.LOCAL)
jh.run_jobs()
status_fail = jh.jobs.test.status
jh.jobs.test.config.backend.run_script = self.run_script_success
jh.run_jobs(retry=True)
status_success = jh.jobs.test.status
test_mode(False)
self.assertIs(status_fail, Status.FAILED)
self.assertIs(status_success, Status.SUCCESS)
def tearDown(self):
from slurmy import test_mode
test_mode(False)
