From d920d9e9fe850369113798ff31987e23915447ec Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sybren=20A=2E=20St=C3=BCvel?= <sybren@stuvel.eu>
Date: Tue, 6 Jun 2017 17:35:49 +0200
Subject: [PATCH] Also mock .s() and .si() celery signature functions.

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

diff --git a/pillar/tests/__init__.py b/pillar/tests/__init__.py
index beebb6b1..70514cfd 100644
--- a/pillar/tests/__init__.py
+++ b/pillar/tests/__init__.py
@@ -74,7 +74,10 @@ class PillarTestServer(pillar.PillarServer):
             def decorator(f):
                 def delay(*args, **kwargs):
                     return f(*args, **kwargs)
+
                 f.delay = delay
+                f.si = unittest.mock.MagicMock()
+                f.s = unittest.mock.MagicMock()
                 return f
 
             return decorator