summaryrefslogtreecommitdiff
path: root/Lib/unittest/mock.py
diff options
context:
space:
mode:
authorAaron Gallagher <habnabit@users.noreply.github.com>2017-07-19 17:01:14 -0700
committerBerker Peksag <berker.peksag@gmail.com>2017-07-20 03:01:14 +0300
commit856cbcc12f2e4cca93af5dc7ed6bcea4dd942f10 (patch)
tree16b9a0093cb7e8fe3a8f8c6886679fcea7ad3596 /Lib/unittest/mock.py
parented014f7e135fe021837208960237d6c37afde5be (diff)
downloadcpython-git-856cbcc12f2e4cca93af5dc7ed6bcea4dd942f10.tar.gz
bpo-29403: Fix mock's broken autospec behavior on method-bound builtin functions (GH-3)
Cython will, in the right circumstances, offer a MethodType instance where im_func is a builtin function. Any instance of MethodType is automatically assumed to be a Python-defined function (more specifically, a function that has an inspectable signature), but _set_signature was still conservative in its assumptions. As a result _set_signature would return early with None instead of a mock since the im_func had no inspectable signature. This causes problems deeper inside mock, as _set_signature is assumed to _always_ return a mock, and nothing checked its return value. In similar corner cases, autospec will simply not check the spec of the function, so _set_signature is amended to now return early with the original, not-wrapped mock object. Patch by Aaron Gallagher.
Diffstat (limited to 'Lib/unittest/mock.py')
-rw-r--r--Lib/unittest/mock.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/Lib/unittest/mock.py b/Lib/unittest/mock.py
index 6989dc792e..94e3442348 100644
--- a/Lib/unittest/mock.py
+++ b/Lib/unittest/mock.py
@@ -155,7 +155,7 @@ def _set_signature(mock, original, instance=False):
skipfirst = isinstance(original, type)
result = _get_signature_object(original, instance, skipfirst)
if result is None:
- return
+ return mock
func, sig = result
def checksig(*args, **kwargs):
sig.bind(*args, **kwargs)