def test_error(self):
ctx = Mock()
ctx.notify_error = Mock(
side_effect=lambda *args, **kwargs: print('Error: %s %s' % (args, kwargs)))
q = TasksQueue()
q.add_task('sleep', ('a',), task_context=ctx)
asyncio.ensure_future(q.run_tasks())
loop = asyncio.get_event_loop()
loop.run_until_complete(q.join(0.1))
self.assertFalse(ctx.notify_success.called)
self.assertTrue(ctx.notify_error.called)
def test_queue(self):
ctx = Mock()
ctx.notify_success = Mock(
side_effect=lambda *args, **kwargs: print('Success: %s %s' % (args, kwargs)))
q = TasksQueue()
for _i in range(10):
q.add_task('date', ('%d-%m-%Y %H:%M %Z',), {'utc': True}, task_context=ctx)
asyncio.ensure_future(q.run_tasks())
loop = asyncio.get_event_loop()
loop.run_until_complete(q.join(0.1))
self.assertEqual(ctx.notify_start.call_count, 10)
self.assertEqual(ctx.notify_success.call_count, 10)
self.assertFalse(ctx.notify_error.called)
def test_multi(self):
ctx = Mock()
ctx.notify_error = Mock(
side_effect=lambda *args, **kwargs: print('Error: %s %s' % (args, kwargs)))
loop = asyncio.get_event_loop()
run = loop.run_until_complete
q = TasksQueue()
task_id=q.add_task('multi', (['date', 'sleep', 'date'], [(['%d-%m-%Y %H:%M %Z'], {'utc':True}), ([1], {}),
(['%d-%m-%Y %H:%M %Z'], {'utc':False}),]),
task_context=ctx)
asyncio.ensure_future(q.run_tasks())
loop.run_until_complete(q.join(1.1))
ctx.notify_error.assert_not_called()
self.assertEqual(ctx.notify_success.call_count, 1)
self.assertEqual(ctx.notify_progress.call_count, 3)
for expected, received in zip([0.333, 0.666, 1.0],
map(lambda c: c[0][1],ctx.notify_progress.call_args_list)):
self.assertAlmostEqual(received, expected, 2)
args = ctx.notify_success.call_args[0]
self.assertEqual(args[0], task_id)
results = args[1]['results']
self.assertEqual(len(results), 3)
self.assertEqual(len(results[0]), 20)
self.assertEqual(results[0][-3:], 'UTC')
self.assertEqual(len(results[2]), 20)
self.assertTrue(results[1] is None)
duration = args[2]
self.assertTrue(duration > 1)
def test_queue_priority(self):
ctx = Mock()
done = []
ctx.notify_success = Mock(
side_effect=lambda task_id, res, dur: done.append(task_id))
q = TasksQueue(concurrent=4)
for _i in range(6):
q.add_task('sleep', (1,), task_context=ctx)
speedy_id=q.add_task('sleep', (0.8,), task_priority=MAX_PRIORITY, task_context=ctx)
asyncio.ensure_future(q.run_tasks())
loop = asyncio.get_event_loop()
loop.run_until_complete(q.join(0.1))
self.assertEqual(ctx.notify_start.call_count, 7)
self.assertEqual(ctx.notify_success.call_count, 7)
self.assertFalse(ctx.notify_error.called)
self.assertEqual(len(done), 7)
self.assertEqual(done[0], speedy_id)
def test_queue_concurrency(self):
ctx = Mock()
ctx.notify_success = Mock(
side_effect=lambda *args, **kwargs: print('Success: %s %s' % (args, kwargs)))
q = TasksQueue(concurrent=4)
for _i in range(12):
q.add_task('sleep', (1,), task_context=ctx)
asyncio.ensure_future(q.run_tasks())
loop = asyncio.get_event_loop()
start = time.time()
loop.run_until_complete(q.join(0.1))
duration = time.time() - start
self.assertTrue(
abs(duration - 3.1) < 0.1, 'Should take app 3.1 secs, but took %f' % duration)
self.assertEqual(ctx.notify_start.call_count, 12)
self.assertEqual(ctx.notify_success.call_count, 12)
def test_multi_empty(self):
ctx = Mock()
ctx.notify_error = Mock(
side_effect=lambda *args, **kwargs: print('Error: %s %s' % (args, kwargs)))
loop = asyncio.get_event_loop()
run = loop.run_until_complete
q = TasksQueue()
# test empty multi
task_id=q.add_task('multi', ([], []), task_context=ctx)
asyncio.ensure_future(q.run_tasks())
run(q.join(0.1))
self.assertEqual(ctx.notify_success.call_count, 1)