def move_context(self, **kwargs):
with patch("celery.contrib.migrate.start_filter") as start:
with patch("celery.contrib.migrate.republish") as republish:
pred = Mock(name="predicate")
move(pred, app=self.app, connection=self.app.connection(), **kwargs)
start.assert_called()
callback = start.call_args[0][2]
yield callback, pred, republish
def move_context(self, **kwargs):
with patch('celery.contrib.migrate.start_filter') as start:
with patch('celery.contrib.migrate.republish') as republish:
pred = Mock(name='predicate')
move(pred, app=self.app,
connection=self.app.connection(), **kwargs)
self.assertTrue(start.called)
callback = start.call_args[0][2]
yield callback, pred, republish
def move_context(self, **kwargs):
with patch('celery.contrib.migrate.start_filter') as start:
with patch('celery.contrib.migrate.republish') as republish:
pred = Mock(name='predicate')
move(pred, app=self.app,
connection=self.app.connection(), **kwargs)
start.assert_called()
callback = start.call_args[0][2]
yield callback, pred, republish