def test_expand_route_not_found(self):
expand = E(current_app.conf.CELERY_QUEUES)
route = routes.MapRoute({"a": {"queue": "x"}})
with self.assertRaises(QueueNotFound):
expand(route.route_for_task("a"))
def test_route_for_task(self):
expand = E(current_app.conf.CELERY_QUEUES)
route = routes.MapRoute({mytask.name: b_queue})
self.assertDictContainsSubset(
b_queue, expand(route.route_for_task(mytask.name)))
self.assertIsNone(route.route_for_task("celery.awesome"))
def test_expand_route_not_found(self):
expand = E(conf.QUEUES)
route = routes.MapRoute({"a": {"queue": "x"}})
self.assertRaises(QueueNotFound, expand, route.route_for_task("a"))
def test_route_for_task_expanded_route(self):
expand = E(current_app.conf.CELERY_QUEUES)
route = routes.MapRoute({"celery.ping": {"queue": "foo"}})
self.assertDictContainsSubset(
a_queue, expand(route.route_for_task("celery.ping")))
self.assertIsNone(route.route_for_task("celery.awesome"))
def test_route_for_task(self):
expand = E(conf.QUEUES)
route = routes.MapRoute({"celery.ping": b_queue})
self.assertDictContainsSubset(b_queue,
expand(route.route_for_task("celery.ping")))
self.assertIsNone(route.route_for_task("celery.awesome"))
