Testing: make Celery tasks execute immediately when called.

This commit is contained in:
Sybren A. Stüvel 2017-06-02 16:15:46 +02:00
parent 6f16e20bf6
commit 27ad3459c1

View File

@ -70,6 +70,17 @@ class PillarTestServer(pillar.PillarServer):
self.celery = unittest.mock.MagicMock(Celery)
def fake_task(*task_args, **task_kwargs):
def decorator(f):
def delay(*args, **kwargs):
return f(*args, **kwargs)
f.delay = delay
return f
return decorator
self.celery.task = fake_task
class AbstractPillarTest(TestMinimal):
pillar_server_class = PillarTestServer