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