def flask_celery_app(flask_celery):
test_app = flask_celery.get_celery_app()
test_app.loader.import_task_module("celery.contrib.testing.tasks")
with setup_default_app(test_app):
test_app.set_default()
test_app.set_current()
yield test_app
def setUp(self):
celery_app = TestApp(set_as_current=False,
enable_logging=True,
config=self['celery_config'],
**self['celery_parameters'])
self['celery_app_fixture'] = setup_default_app(celery_app)
self['celery_app_fixture'].__enter__()
for module in self['celery_includes']:
celery_app.loader.import_task_module(module)
self['celery_worker_fixture'] = start_worker(
celery_app, pool='prefork', **self['celery_worker_parameters'])
self['celery_worker_fixture'].__enter__()
def _create_app(enable_logging=False,
use_trap=False,
parameters={},
**config):
# type: (Any, **Any) -> Celery
"""Utility context used to setup Celery app for pytest fixtures."""
test_app = TestApp(
set_as_current=False,
enable_logging=enable_logging,
config=config,
**parameters
)
with setup_default_app(test_app, use_trap=use_trap):
yield test_app
def swh_scheduler_celery_app():
"""Set up a Celery app as swh.scheduler and swh worker tests would expect it"""
test_app = TestApp(
set_as_current=True,
enable_logging=True,
task_cls="swh.scheduler.task:SWHTask",
config={
"accept_content": ["application/x-msgpack", "application/json"],
"task_serializer": "msgpack",
"result_serializer": "json",
},
)
with setup_default_app(test_app, use_trap=False):
from swh.scheduler.celery_backend import config
config.app = test_app
test_app.set_default()
test_app.set_current()
yield test_app
def dagster_celery_app():
app = make_app()
with setup_default_app(app, use_trap=False):
yield app
def dagster_celery_app(rabbitmq): # pylint: disable=redefined-outer-name, unused-argument
app = make_app()
execute_plan = create_task(app) # pylint: disable=unused-variable
with setup_default_app(app, use_trap=False):
yield app
