def test_revoked(self): tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) with assert_signal_called(task_revoked, sender=tw.task, terminated=False, expired=False, signum=None): revoked.add(tw.id) self.assertTrue(tw.revoked()) self.assertTrue(tw._already_revoked) self.assertTrue(tw.acknowledged)
def test_on_accepted_terminates(self): signum = signal.SIGKILL pool = Mock() tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"}) with assert_signal_called(task_revoked, sender=tw.task, terminated=True, expired=False, signum=signum): tw.terminate(pool, signal="KILL") self.assertFalse(pool.terminate_job.call_count) tw.on_accepted(pid=314, time_accepted=time.time()) pool.terminate_job.assert_called_with(314, signum)
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_revoked(self): tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'}) with assert_signal_called(task_revoked, sender=tw.task, terminated=False, expired=False, signum=None): revoked.add(tw.id) self.assertTrue(tw.revoked()) self.assertTrue(tw._already_revoked) self.assertTrue(tw.acknowledged)
def test_revoked_expires_expired(self): tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'}, expires=datetime.utcnow() - timedelta(days=1)) with assert_signal_called(task_revoked, sender=tw.task, terminated=False, expired=True, signum=None): tw.revoked() self.assertIn(tw.id, revoked) self.assertEqual(mytask.backend.get_status(tw.id), states.REVOKED)
def test_revoked_expires_expired(self): tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'}, expires=datetime.utcnow() - timedelta(days=1)) with assert_signal_called(task_revoked, sender=tw.task, terminated=False, expired=True, signum=None): tw.revoked() self.assertIn(tw.id, revoked) self.assertEqual(mytask.backend.get_status(tw.id), states.REVOKED)
def test_on_accepted_terminates(self): signum = signal.SIGKILL pool = Mock() tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'}) with assert_signal_called(task_revoked, sender=tw.task, terminated=True, expired=False, signum=signum): tw.terminate(pool, signal='KILL') self.assertFalse(pool.terminate_job.call_count) tw.on_accepted(pid=314, time_accepted=time.time()) pool.terminate_job.assert_called_with(314, signum)
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)