Skip to content
Snippets Groups Projects
Commit 7f655374 authored by fuzzyman's avatar fuzzyman
Browse files

Fix issue with _mock_ as varargs or varkwargs and mocksignature

parent 0a9900e2
No related branches found
No related tags found
No related merge requests found
......@@ -86,6 +86,10 @@ def _getsignature(func, skipfirst):
regargs = regargs[1:]
assert '_mock_' not in regargs, ("_mock_ is a reserved argument name, can't mock signatures using _mock_")
if varargs is not None:
assert '_mock_' not in varargs, ("_mock_ is a reserved argument name, can't mock signatures using _mock_")
if varkwargs:
assert '_mock_' not in varkwargs, ("_mock_ is a reserved argument name, can't mock signatures using _mock_")
if skipfirst:
regargs = regargs[1:]
signature = inspect.formatargspec(regargs, varargs, varkwargs, defaults, formatvalue=lambda value: "")
......
......@@ -149,5 +149,15 @@ class TestMockSignature(unittest2.TestCase):
def f(_mock_=None):
pass
self.assertRaises(AssertionError, lambda: mocksignature(f))
def f(*_mock_):
pass
self.assertRaises(AssertionError, lambda: mocksignature(f))
def f(**_mock_):
pass
self.assertRaises(AssertionError, lambda: mocksignature(f))
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment