test_runner.py 556 B

12345678910111213141516171819
  1. from django.conf import settings
  2. from django.test.simple import run_tests as run_tests_orig
  3. USAGE = """\
  4. Custom test runner to allow testing of celery delayed tasks.
  5. """
  6. def run_tests(test_labels, *args, **kwargs):
  7. """Django test runner allowing testing of celery delayed tasks.
  8. All tasks are run locally, not in a worker.
  9. To use this runner set ``settings.TEST_RUNNER``::
  10. TEST_RUNNER = "celery.contrib.test_runner.run_tests"
  11. """
  12. settings.CELERY_ALWAYS_EAGER = True
  13. return run_tests_orig(test_labels, *args, **kwargs)