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_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_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_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_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))
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))
