def _backend_setup(self, *args, **kwargs): self.celery = turbinia_celery.TurbiniaCelery() self.celery.setup() self.kombu = turbinia_celery.TurbiniaKombu(config.KOMBU_CHANNEL) self.kombu.setup() self.celery_runner = self.celery.app.task(task_runner, name="task_runner")
def setUp(self): request = getTurbiniaRequest() self.kombu = celery.TurbiniaKombu('fake_topic') result = mock.MagicMock() result.payload = request.to_json() self.kombu.queue = mock.MagicMock() self.kombu.queue.__len__.return_value = 1 self.kombu.queue.get.side_effect = [result, queue.Empty('Empty Queue')]
def _backend_setup(self, *args, **kwargs): self.celery = turbinia_celery.TurbiniaCelery() self.celery.setup() self.kombu = turbinia_celery.TurbiniaKombu(config.KOMBU_CHANNEL) self.kombu.setup()