def test_manual_retry_task(self): task_id = manual_retry() result = task.run(task_id) self.assertEqual('fail', result) self.assertTrue(task.exists(task_id)) self.assertFalse(task.is_complete(task_id)) result = task.run(task_id) self.assertNotEqual('fail', result) self.assertTrue(task.exists(task_id)) self.assertTrue(task.is_complete(task_id))
def test_object_retry_deleted_object(self): obj = ObjectWithTasks(42) task_id = obj.retry_value() task.run(task_id) self.assertTrue(task.exists(task_id)) self.assertFalse(task.is_complete(task_id)) del obj result = task.run(task_id) self.assertNotEqual('fail', result) self.assertTrue(task.exists(task_id)) self.assertTrue(task.is_complete(task_id)) self.assertEqual(result, 42)
def test_generator_retry_task(self): task_id = generator_retry() rval = task.run(task_id) result = rval.next() self.assertEqual('fail', result) self.assertTrue(task.exists(task_id)) self.assertFalse(task.is_complete(task_id)) rval = task.run(task_id) result = rval.next() self.assertNotEqual('fail', result) self.assertRaises(StopIteration, rval.next) self.assertTrue(task.exists(task_id)) self.assertTrue(task.is_complete(task_id))
def test_complex_task(self): task_id = complex_task(10) rval = task.run(task_id) self.assertTrue(task.exists(task_id)) self.assertFalse(task.is_complete(task_id)) total = 0 for x in xrange(5): total += rval.next() self.assertTrue(task.exists(task_id)) self.assertFalse(task.is_complete(task_id)) rval = task.run(task_id) for x in xrange(5): total += rval.next() self.assertRaises(StopIteration, rval.next) self.assertTrue(task.exists(task_id)) self.assertTrue(task.is_complete(task_id)) task_id = complex_task(10) rval = task.run(task_id) self.assertEqual(total, sum(list(rval)))
def test_change_task_name(self): task_id = one_name() task.run(task_id) self.assertTrue(task.exists(task_id)) self.assertTrue(task.is_complete(task_id)) self.assertEqual(task.get(task_id)['task_name'], 'another_name')
def test_finish_task(self): task_id = finish() task.run(task_id) self.assertTrue(task.exists(task_id)) self.assertTrue(task.is_complete(task_id))