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 __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)
