def test_run_failed(self):
def start(t):
t.started = True
def error(t, err):
t.err_msg = str(err)
def success(t, result):
t.success_result = result
def fail_task(inital, incr_by=1):
raise IOError('Math is hard.')
task = Task(on_start=start, on_success=success, on_error=error)
# Should fail.
task.to_call(fail_task, 2)
try:
task.run()
except IOError:
pass
self.assertTrue(task.started)
self.assertEqual(task.err_msg, 'Math is hard.')
def test_execute_success(self):
task = Task(retries=3, async=True)
task.to_call(so_computationally_expensive, 2, 7)
res = self.gator.execute(task)
self.assertEqual(res, 9)
self.assertEqual(task.retries, 3)
self.assertEqual(task.status, SUCCESS)
def test_cancel(self):
self.assertEqual(self.gator.backend.len(ALL), 0)
task_1 = Task(async=True)
task_2 = Task(task_id='hello', async=True)
self.gator.push(task_1, so_computationally_expensive, 1, 1)
self.gator.push(task_2, so_computationally_expensive, 3, 5)
self.assertEqual(self.gator.backend.len(ALL), 2)
task = self.gator.cancel(task_2.task_id)
self.assertEqual(task.status, CANCELED)
self.assertEqual(self.gator.backend.len(ALL), 1)
def test_default_init(self):
task = Task()
self.assertNotEqual(task.task_id, None)
self.assertEqual(task.retries, 0)
self.assertEqual(task. async, True)
self.assertEqual(task.on_start, None)
self.assertEqual(task.on_success, None)
self.assertEqual(task.on_error, None)
self.assertEqual(task.depends_on, None)
self.assertEqual(task.status, WAITING)
self.assertEqual(task.func, None)
self.assertEqual(task.func_args, [])
self.assertEqual(task.func_kwargs, {})
def test_get(self):
self.assertEqual(self.gator.backend.len(ALL), 0)
task_1 = Task(async=True)
task_2 = Task(task_id='hello', async=True)
self.gator.push(task_1, so_computationally_expensive, 1, 1)
self.gator.push(task_2, so_computationally_expensive, 3, 5)
self.assertEqual(self.gator.backend.len(ALL), 2)
res = self.gator.get(task_2.task_id)
self.assertEqual(res, 8)
res = self.gator.get(task_1.task_id)
self.assertEqual(res, 2)
def test_execute_failed(self):
task = Task(retries=3, async=True)
task.to_call(fail_task, 2, 7)
self.assertEqual(task.status, WAITING)
try:
self.gator.execute(task)
self.assertEqual(task.status, RETRYING)
self.gator.execute(task)
self.gator.execute(task)
self.gator.execute(task)
self.fail()
except IOError:
self.assertEqual(task.retries, 0)
self.assertEqual(task.status, FAILED)
def test_execute_retries(self):
task = Task(retries=3, async=True)
task.to_call(eventual_success(), 2, 7)
try:
self.gator.execute(task)
except IOError:
pass
try:
self.gator.execute(task)
except IOError:
pass
res = self.gator.execute(task)
self.assertEqual(res, 9)
self.assertEqual(task.retries, 1)
def test_custom_init(self):
task = Task(task_id='hello',
retries=3,
async=False,
on_start=start,
on_success=success,
on_error=error,
depends_on=['id1', 'id3', 'id4'])
self.assertNotEqual(task.task_id, None)
self.assertEqual(task.retries, 3)
self.assertEqual(task. async, False)
self.assertEqual(task.on_start, start)
self.assertEqual(task.on_success, success)
self.assertEqual(task.on_error, error)
self.assertEqual(task.depends_on, ['id1', 'id3', 'id4'])
self.assertEqual(task.status, WAITING)
self.assertEqual(task.func, None)
self.assertEqual(task.func_args, [])
self.assertEqual(task.func_kwargs, {})
def test_run_success(self):
def start(t):
t.started = True
def error(t, err):
t.err_msg = str(err)
def success(t, result):
t.success_result = result
def success_task(initial, incr_by=1):
return initial + incr_by
task = Task(on_start=start, on_success=success, on_error=error)
# Should succeed.
task.to_call(success_task, 12, 3)
result = task.run()
self.assertEqual(result, 15)
self.assertTrue(task.started)
self.assertEqual(task.success_result, 15)
def setUp(self):
super(TaskTestCase, self).setUp()
self.task = Task()
