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)