From d6d8d6484ee4dab61db2e35542b506593e86a6b7 Mon Sep 17 00:00:00 2001 From: malcolm m Date: Fri, 10 Apr 2015 11:42:29 -0500 Subject: [PATCH] allow user-specified backend for result testrunner --- djcelery/contrib/test_runner.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/djcelery/contrib/test_runner.py b/djcelery/contrib/test_runner.py index e65de297..a6e03654 100644 --- a/djcelery/contrib/test_runner.py +++ b/djcelery/contrib/test_runner.py @@ -42,7 +42,9 @@ class CeleryTestSuiteRunnerStoringResult(DiscoverRunner): """Django test runner allowing testing of celery delayed tasks, and storing the results of those tasks in ``TaskMeta``. - Requires setting CELERY_RESULT_BACKEND = 'database'. + Requires setting CELERY_RESULT_BACKEND to be one of: + 'database' + 'djcelery.backends.database:DatabaseBackend' USAGE: @@ -65,5 +67,8 @@ def on_success_patched(self, retval, task_id, args, kwargs): **kwargs ) - settings.CELERY_RESULT_BACKEND = 'database' + if not hasattr(settings, 'CELERY_RESULT_BACKEND') or \ + settings.CELERY_RESULT_BACKEND not in \ + ['database', 'djcelery.contrib.test_runner:CeleryTestSuiteRunnerStoringResult']: + settings.CELERY_RESULT_BACKEND = 'database' _set_eager()