Also mock .s() and .si() celery signature functions.
This commit is contained in:
parent
c2bc52718a
commit
d920d9e9fe
@ -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
|
||||||
|
Loading…
x
Reference in New Issue
Block a user