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)
