def __init__(self, job_source): from lp.services.job.celeryjob import CeleryRunJob from lazr.jobrunner.celerytask import list_queued self.job_source = job_source self.queue_contents = list_queued(CeleryRunJob.app, [job_source.task_queue]) self.queued_job_ids = set(task[1][0][0] for task in self.queue_contents)
def assertQueueSize(self, app, queues, expected_len): """Assert the message queue (eventually) reaches the specified size. This can be used to avoid race conditions with RabbitMQ's message delivery. """ from lazr.jobrunner.celerytask import list_queued for x in range(600): actual_len = len(list_queued(app, queues)) if actual_len == expected_len: return sleep(0.1) self.fail('Queue size did not reach %d; still at %d' % (expected_len, actual_len))
def assertQueueSize(self, app, queues, expected_len): """Assert the message queue (eventually) reaches the specified size. This can be used to avoid race conditions with RabbitMQ's message delivery. """ from lazr.jobrunner.celerytask import list_queued for x in range(600): actual_len = len(list_queued(app, queues)) if actual_len == expected_len: return sleep(0.1) self.fail("Queue size did not reach %d; still at %d" % (expected_len, actual_len))
def __init__(self, job_source): from lazr.jobrunner.celerytask import list_queued self.job_source = job_source self.queue_contents = list_queued(celery_app, [job_source.task_queue]) self.queued_job_ids = set(task[1][0][0] for task in self.queue_contents)