def test_chunks(self):
x = self.add.chunks(range(100), 10)
assert dict(chunks.from_dict(dict(x), app=self.app)) == dict(x)
assert x.group()
assert len(x.group().tasks) == 10
x.group = Mock()
gr = x.group.return_value = Mock()
x.apply_async()
gr.apply_async.assert_called_with((), {}, route_name=self.add.name)
gr.apply_async.reset_mock()
x()
gr.apply_async.assert_called_with((), {}, route_name=self.add.name)
self.app.conf.task_always_eager = True
chunks.apply_chunks(app=self.app, **x['kwargs'])
def test_chunks(self):
x = self.add.chunks(range(100), 10)
self.assertEqual(
dict(chunks.from_dict(dict(x), app=self.app)), dict(x),
)
self.assertTrue(x.group())
self.assertEqual(len(x.group().tasks), 10)
x.group = Mock()
gr = x.group.return_value = Mock()
x.apply_async()
gr.apply_async.assert_called_with((), {}, route_name=self.add.name)
gr.apply_async.reset_mock()
x()
gr.apply_async.assert_called_with((), {}, route_name=self.add.name)
self.app.conf.CELERY_ALWAYS_EAGER = True
chunks.apply_chunks(app=self.app, **x['kwargs'])
def test_chunks(self):
x = add.chunks(range(100), 10)
self.assertEqual(chunks.from_dict(dict(x)), x)
self.assertTrue(x.group())
self.assertEqual(len(x.group().tasks), 10)
x.group = Mock()
gr = x.group.return_value = Mock()
x.apply_async()
gr.apply_async.assert_called_with((), {})
x()
gr.assert_called_with()
self.app.conf.CELERY_ALWAYS_EAGER = True
try:
chunks.apply_chunks(**x['kwargs'])
finally:
self.app.conf.CELERY_ALWAYS_EAGER = False
