def test_on_success_eventer(self): tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'}, app=self.app) tw.time_start = 1 tw.eventer = Mock() tw.send_event = Mock() tw.on_success(42) self.assertTrue(tw.send_event.called)
def test_terminate__task_started(self): pool = Mock() tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.time_start = time.time() tw.worker_pid = 313 tw.terminate(pool, signal="KILL") pool.terminate_job.assert_called_with(tw.worker_pid, "KILL")
def test_on_success_eventer(self): job = TaskRequest(self.mytask.name, uuid(), [1], {"f": "x"}, app=self.app) job.time_start = 1 job.eventer = Mock() job.send_event = Mock() job.on_success(42) self.assertTrue(job.send_event.called)
def test_on_success_acks_early(self): tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.time_start = 1 tw.on_success(42) tw._does_info = False tw.on_success(42) self.assertFalse(tw.acknowledged)
def test_on_success_eventer(self): tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.time_start = 1 tw.eventer = Mock() tw.send_event = Mock() tw.on_success(42) self.assertTrue(tw.send_event.called)
def test_terminate__task_reserved(self): pool = Mock() tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.time_start = None tw.terminate(pool, signal="KILL") self.assertFalse(pool.terminate_job.call_count) self.assertTupleEqual(tw._terminate_on_ack, (True, pool, "KILL")) tw.terminate(pool, signal="KILL")
def test_terminate__task_reserved(self): pool = Mock() tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'}) tw.time_start = None tw.terminate(pool, signal='KILL') self.assertFalse(pool.terminate_job.called) self.assertTupleEqual(tw._terminate_on_ack, (pool, 'KILL')) tw.terminate(pool, signal='KILL')
def test_on_success_acks_late(self): job = TaskRequest( self.mytask.name, uuid(), [1], {'f': 'x'}, app=self.app, ) job.time_start = 1 self.mytask.acks_late = True job.on_success(42) self.assertTrue(job.acknowledged)
def test_terminate__task_reserved(self): pool = Mock() tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'}, app=self.app) tw.time_start = None tw.terminate(pool, signal='KILL') self.assertFalse(pool.terminate_job.called) self.assertTupleEqual(tw._terminate_on_ack, (pool, 'KILL')) tw.terminate(pool, signal='KILL')
def test_terminate__task_reserved(self): pool = Mock() job = TaskRequest(self.mytask.name, uuid(), [1], {"f": "x"}, app=self.app) job.time_start = None job.terminate(pool, signal="KILL") self.assertFalse(pool.terminate_job.called) self.assertTupleEqual(job._terminate_on_ack, (pool, "KILL")) job.terminate(pool, signal="KILL")
def test_on_success_when_failure(self): tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.time_start = 1 tw.on_failure = Mock() try: raise KeyError("foo") except Exception: tw.on_success(ExceptionInfo(sys.exc_info())) self.assertTrue(tw.on_failure.called)
def test_on_success_when_failure(self): tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.time_start = 1 tw.on_failure = Mock() try: raise KeyError("foo") except Exception: tw.on_success(ExceptionInfo()) self.assertTrue(tw.on_failure.called)
def test_on_success_acks_late(self): tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.time_start = 1 mytask.acks_late = True try: tw.on_success(42) self.assertTrue(tw.acknowledged) finally: mytask.acks_late = False
def test_on_success_when_failure(self): job = TaskRequest(self.mytask.name, uuid(), [1], {"f": "x"}, app=self.app) job.time_start = 1 job.on_failure = Mock() try: raise KeyError("foo") except Exception: job.on_success(ExceptionInfo()) self.assertTrue(job.on_failure.called)
def test_terminate__task_started(self): pool = Mock() signum = signal.SIGKILL tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) with assert_signal_called(task_revoked, sender=tw.task, terminated=True, expired=False, signum=signum): tw.time_start = time.time() tw.worker_pid = 313 tw.terminate(pool, signal="KILL") pool.terminate_job.assert_called_with(tw.worker_pid, signum)
def test_on_success_when_failure(self): tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'}, app=self.app) tw.time_start = 1 tw.on_failure = Mock() try: raise KeyError('foo') except Exception: tw.on_success(ExceptionInfo()) self.assertTrue(tw.on_failure.called)
def test_on_success_acks_early(self): job = TaskRequest(self.mytask.name, uuid(), [1], {"f": "x"}, app=self.app) job.time_start = 1 job.on_success(42) prev, module._does_info = module._does_info, False try: job.on_success(42) self.assertFalse(job.acknowledged) finally: module._does_info = prev
def test_on_failure_acks_late(self): job = TaskRequest(self.mytask.name, uuid(), [1], {"f": "x"}, app=self.app) job.time_start = 1 self.mytask.acks_late = True try: raise KeyError("foo") except KeyError: exc_info = ExceptionInfo() job.on_failure(exc_info) self.assertTrue(job.acknowledged)
def test_on_success_acks_early(self): tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'}) tw.time_start = 1 tw.on_success(42) prev, module._does_info = module._does_info, False try: tw.on_success(42) self.assertFalse(tw.acknowledged) finally: module._does_info = prev
def test_on_success_BaseException(self): tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.time_start = 1 with self.assertRaises(SystemExit): try: raise SystemExit() except SystemExit: tw.on_success(ExceptionInfo()) else: assert False
def test_on_success_BaseException(self): tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'}, app=self.app) tw.time_start = 1 with self.assertRaises(SystemExit): try: raise SystemExit() except SystemExit: tw.on_success(ExceptionInfo()) else: assert False
def test_on_success_acks_early(self): tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'}, app=self.app) tw.time_start = 1 tw.on_success(42) prev, module._does_info = module._does_info, False try: tw.on_success(42) self.assertFalse(tw.acknowledged) finally: module._does_info = prev
def test_terminate__task_started(self): pool = Mock() signum = signal.SIGKILL job = TaskRequest(self.mytask.name, uuid(), [1], {"f": "x"}, app=self.app) with assert_signal_called( task_revoked, sender=job.task, request=job, terminated=True, expired=False, signum=signum ): job.time_start = time.time() job.worker_pid = 313 job.terminate(pool, signal="KILL") pool.terminate_job.assert_called_with(job.worker_pid, signum)
def test_on_success_BaseException(self): job = TaskRequest( self.mytask.name, uuid(), [1], {'f': 'x'}, app=self.app, ) job.time_start = 1 with self.assertRaises(SystemExit): try: raise SystemExit() except SystemExit: job.on_success(ExceptionInfo()) else: assert False
def test_terminate__task_started(self): pool = Mock() signum = signal.SIGKILL tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'}) with assert_signal_called(task_revoked, sender=tw.task, terminated=True, expired=False, signum=signum): tw.time_start = time.time() tw.worker_pid = 313 tw.terminate(pool, signal='KILL') pool.terminate_job.assert_called_with(tw.worker_pid, signum)
def test_on_failure_acks_late(self): tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) tw.time_start = 1 mytask.acks_late = True try: try: raise KeyError("foo") except KeyError: exc_info = ExceptionInfo(sys.exc_info()) tw.on_failure(exc_info) self.assertTrue(tw.acknowledged) finally: mytask.acks_late = False
def test_on_failure_acks_late(self): tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'}, app=self.app) tw.time_start = 1 mytask.acks_late = True try: try: raise KeyError('foo') except KeyError: exc_info = ExceptionInfo() tw.on_failure(exc_info) self.assertTrue(tw.acknowledged) finally: mytask.acks_late = False