def test_execute_fail(self):
tid = gen_unique_id()
tw = TaskWrapper(mytask_raising.name, tid, [4], {"f": "x"})
self.assertIsInstance(tw.execute(), ExceptionInfo)
meta = TaskMeta.objects.get(task_id=tid)
self.assertEqual(meta.status, states.FAILURE)
self.assertIsInstance(meta.result, KeyError)
def test_execute_success_no_kwargs(self):
tid = gen_unique_id()
tw = TaskWrapper(mytask_no_kwargs.name, tid, [4], {})
self.assertEqual(tw.execute(), 256)
meta = TaskMeta.objects.get(task_id=tid)
self.assertEqual(meta.result, 256)
self.assertEqual(meta.status, states.SUCCESS)
def test_execute(self):
tid = gen_unique_id()
tw = TaskWrapper("cu.mytask", tid, mytask, [4], {"f": "x"})
self.assertEquals(tw.execute(), 256)
meta = TaskMeta.objects.get(task_id=tid)
self.assertEquals(meta.result, 256)
self.assertEquals(meta.status, "DONE")
def test_execute(self):
tid = gen_unique_id()
tw = TaskWrapper(mytask.name, tid, [4], {"f": "x"})
self.assertEquals(tw.execute(), 256)
meta = TaskMeta.objects.get(task_id=tid)
self.assertEquals(meta.result, 256)
self.assertEquals(meta.status, "SUCCESS")
def test_execute_success_some_kwargs(self):
tid = gen_unique_id()
tw = TaskWrapper(mytask_some_kwargs.name, tid, [4], {})
self.assertEqual(tw.execute(logfile="foobaz.log"), 256)
meta = TaskMeta.objects.get(task_id=tid)
self.assertEqual(some_kwargs_scratchpad.get("logfile"), "foobaz.log")
self.assertEqual(meta.result, 256)
self.assertEqual(meta.status, states.SUCCESS)
def test_execute_fail(self):
tid = gen_unique_id()
tw = TaskWrapper("cu.mytask-raising", tid, mytask_raising, [4],
{"f": "x"})
self.assertTrue(isinstance(tw.execute(), ExceptionInfo))
meta = TaskMeta.objects.get(task_id=tid)
self.assertEquals(meta.status, "FAILURE")
self.assertTrue(isinstance(meta.result, KeyError))
def test_execute_ack(self):
tid = gen_unique_id()
tw = TaskWrapper(mytask.name, tid, [4], {"f": "x"},
on_ack=on_ack)
self.assertEqual(tw.execute(), 256)
meta = TaskMeta.objects.get(task_id=tid)
self.assertTrue(scratch["ACK"])
self.assertEqual(meta.result, 256)
self.assertEqual(meta.status, states.SUCCESS)
def test_execute_ack(self):
tid = gen_unique_id()
tw = TaskWrapper("cu.mytask", tid, mytask, [4], {"f": "x"},
on_acknowledge=on_ack)
self.assertEquals(tw.execute(), 256)
meta = TaskMeta.objects.get(task_id=tid)
self.assertTrue(scratch["ACK"])
self.assertEquals(meta.result, 256)
self.assertEquals(meta.status, "DONE")
