use sys.modules.get correctly and reference DEFAULT with respect to it

This commit is contained in:
Chris Gilling 2014-03-26 09:36:02 -07:00
parent c3f4eb6d57
commit e3b9382122
1 changed files with 3 additions and 3 deletions

View File

@ -1836,10 +1836,10 @@ def getfuncargnames(function, startindex=None):
if startindex is None:
startindex = inspect.ismethod(function) and 1 or 0
if realfunction != function:
mock_default = sys.modules.get('mock.DEFAULT')
if mock_default:
mock = sys.modules.get('mock')
if mock is not None:
for patching in getattr(function, "patchings", []):
if not patching.attribute_name and patching.new is DEFAULT:
if not patching.attribute_name and patching.new is mock.DEFAULT:
startindex += 1
else:
startindex += len(getattr(function, "patchings", []))