Warning

This document is for an old release of Galaxy. You can alternatively view this page in the latest release if it exists or view the top of the latest release's documentation.

Source code for galaxy.app_unittest_utils.celery_helper

from functools import wraps


def rebind_container_to_task(app):
    import galaxy.app
    galaxy.app.app = app
    from galaxy.celery import (
        CELERY_TASKS,
        tasks,
    )

    def magic_bind_dynamic(func):
        return wraps(func)(app.magic_partial(func, shared=None))

    for task in CELERY_TASKS:
        task_fn = getattr(tasks, task, None)
        if task_fn:
            task_fn = getattr(task_fn, '__wrapped__', task_fn)
            container_bound_task = magic_bind_dynamic(task_fn)
            setattr(tasks, task, container_bound_task)