diff --git a/swh/scheduler/tests/celery_testing.py b/swh/scheduler/tests/celery_testing.py --- a/swh/scheduler/tests/celery_testing.py +++ b/swh/scheduler/tests/celery_testing.py @@ -5,6 +5,10 @@ CELERY_BROKER_PROTOCOLS = ['amqp', 'redis', 'sqs'] +CELERY_RESULT_BACKENDS = { + 'amqp': lambda broker_url: 'rpc://', + 'redis': lambda broker_url: broker_url, + } def setup_celery(): @@ -22,6 +26,10 @@ break if broker_url: current_app.conf.broker_url = broker_url + scheme = urllib.parse.splittype(broker_url)[0] + backend = CELERY_RESULT_BACKENDS.get(scheme) + if backend: + current_app.conf.result_backend = backend(broker_url) class CeleryTestFixture: