Mock Celery while testing, to prevent actual background task creation.

This commit is contained in:
Sybren A. Stüvel 2017-06-01 16:47:48 +02:00
parent 5af54237b9
commit d0c30cfeca

View File

@ -8,6 +8,7 @@ import logging
import os import os
import sys import sys
import typing import typing
import unittest.mock
try: try:
from urllib.parse import urlencode from urllib.parse import urlencode
@ -58,6 +59,16 @@ class PillarTestServer(pillar.PillarServer):
logging.getLogger('werkzeug').setLevel(logging.DEBUG) logging.getLogger('werkzeug').setLevel(logging.DEBUG)
logging.getLogger('eve').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): class AbstractPillarTest(TestMinimal):
pillar_server_class = PillarTestServer pillar_server_class = PillarTestServer