From 27ad3459c18035f35285e4458eb335009208dd8b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sybren=20A=2E=20St=C3=BCvel?= <sybren@stuvel.eu>
Date: Fri, 2 Jun 2017 16:15:46 +0200
Subject: [PATCH] Testing: make Celery tasks execute immediately when called.

---
 pillar/tests/__init__.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/pillar/tests/__init__.py b/pillar/tests/__init__.py
index 1e902259..beebb6b1 100644
--- a/pillar/tests/__init__.py
+++ b/pillar/tests/__init__.py
@@ -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