From d0c30cfeca41454701dd217c43c3971170b59abf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sybren=20A=2E=20St=C3=BCvel?= Date: Thu, 1 Jun 2017 16:47:48 +0200 Subject: [PATCH] Mock Celery while testing, to prevent actual background task creation. --- pillar/tests/__init__.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/pillar/tests/__init__.py b/pillar/tests/__init__.py index 329a9556..8090a867 100644 --- a/pillar/tests/__init__.py +++ b/pillar/tests/__init__.py @@ -8,6 +8,7 @@ import logging import os import sys import typing +import unittest.mock try: from urllib.parse import urlencode @@ -58,6 +59,16 @@ class PillarTestServer(pillar.PillarServer): logging.getLogger('werkzeug').setLevel(logging.DEBUG) logging.getLogger('eve').setLevel(logging.DEBUG) + def _config_celery(self): + """Disables Celery by entirely mocking it. + + Without this, actual Celery tasks will be created while the tests are running. + """ + + from celery import Celery + + self.celery = unittest.mock.MagicMock(Celery) + class AbstractPillarTest(TestMinimal): pillar_server_class = PillarTestServer