Also mock .s() and .si() celery signature functions.

This commit is contained in:
Sybren A. Stüvel 2017-06-06 17:35:49 +02:00
parent c2bc52718a
commit d920d9e9fe

View File

@ -74,7 +74,10 @@ class PillarTestServer(pillar.PillarServer):
def decorator(f): def decorator(f):
def delay(*args, **kwargs): def delay(*args, **kwargs):
return f(*args, **kwargs) return f(*args, **kwargs)
f.delay = delay f.delay = delay
f.si = unittest.mock.MagicMock()
f.s = unittest.mock.MagicMock()
return f return f
return decorator return decorator