def test_on_timeout(self): class MockLogger(object): def __init__(self): self.warnings = [] self.errors = [] def warning(self, msg, *args, **kwargs): self.warnings.append(msg) def error(self, msg, *args, **kwargs): self.errors.append(msg) tw = TaskRequest(mytask.name, gen_unique_id(), [1], {"f": "x"}) tw.logger = MockLogger() tw.on_timeout(soft=True) self.assertIn("Soft time limit exceeded", tw.logger.warnings[0]) tw.on_timeout(soft=False) self.assertIn("Hard time limit exceeded", tw.logger.errors[0])
def test_on_timeout(self): class MockLogger(object): def __init__(self): self.warnings = [] self.errors = [] def warning(self, msg, *args, **kwargs): self.warnings.append(msg % args) def error(self, msg, *args, **kwargs): self.errors.append(msg % args) tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.logger = MockLogger() tw.on_timeout(soft=True, timeout=1337) self.assertIn("Soft time limit (1337s) exceeded", tw.logger.warnings[0]) tw.on_timeout(soft=False, timeout=1337) self.assertIn("Hard time limit (1337s) exceeded", tw.logger.errors[0]) self.assertEqual(mytask.backend.get_status(tw.id), states.FAILURE) mytask.ignore_result = True try: tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.logger = MockLogger() finally: tw.on_timeout(soft=True, timeout=1336) self.assertEqual(mytask.backend.get_status(tw.id), states.PENDING) mytask.ignore_result = False
def test_on_timeout(self): class MockLogger(object): def __init__(self): self.warnings = [] self.errors = [] def warning(self, msg, *args, **kwargs): self.warnings.append(msg % args) def error(self, msg, *args, **kwargs): self.errors.append(msg % args) tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.logger = MockLogger() tw.on_timeout(soft=True, timeout=1337) self.assertIn("Soft time limit (1337s) exceeded", tw.logger.warnings[0]) tw.on_timeout(soft=False, timeout=1337) self.assertIn("Hard time limit (1337s) exceeded", tw.logger.errors[0]) self.assertEqual(mytask.backend.get_status(tw.task_id), states.FAILURE) mytask.ignore_result = True try: tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.logger = MockLogger() finally: tw.on_timeout(soft=True, timeout=1336) self.assertEqual(mytask.backend.get_status(tw.task_id), states.PENDING) mytask.ignore_result = False
def test_on_timeout(self, warn, error): job = TaskRequest(self.mytask.name, uuid(), [1], {"f": "x"}, app=self.app) job.on_timeout(soft=True, timeout=1337) self.assertIn("Soft time limit", warn.call_args[0][0]) job.on_timeout(soft=False, timeout=1337) self.assertIn("Hard time limit", error.call_args[0][0]) self.assertEqual(self.mytask.backend.get_status(job.id), states.FAILURE) self.mytask.ignore_result = True job = TaskRequest(self.mytask.name, uuid(), [1], {"f": "x"}, app=self.app) job.on_timeout(soft=True, timeout=1336) self.assertEqual(self.mytask.backend.get_status(job.id), states.PENDING)
def test_on_timeout(self, warn, error): tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.on_timeout(soft=True, timeout=1337) self.assertIn("Soft time limit", warn.call_args[0][0]) tw.on_timeout(soft=False, timeout=1337) self.assertIn("Hard time limit", error.call_args[0][0]) self.assertEqual(mytask.backend.get_status(tw.id), states.FAILURE) mytask.ignore_result = True try: tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.on_timeout(soft=True, timeout=1336) self.assertEqual(mytask.backend.get_status(tw.id), states.PENDING) finally: mytask.ignore_result = False