From: Xtreak <tirkarthi@users.noreply.github.com>
Date: Mon, 3 Dec 2018 07:58:15 +0000 (+0530)
Subject: bpo-32153: Add unit test for create_autospec with partial function returned in getatt... 
X-Git-Tag: v3.8.0a1~370
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=c667b094ae37799a7e42ba5cd2ad501cc7920888;p=thirdparty%2FPython%2Fcpython.git

bpo-32153: Add unit test for create_autospec with partial function returned in getattr (#10398)

* Add create_autospec with partial function returned in getattr

* Use self.assertFalse instead of assert

* Use different names and remove object
---

diff --git a/Lib/unittest/test/testmock/testhelpers.py b/Lib/unittest/test/testmock/testhelpers.py
index 7919482ae99c..9edebf551660 100644
--- a/Lib/unittest/test/testmock/testhelpers.py
+++ b/Lib/unittest/test/testmock/testhelpers.py
@@ -8,6 +8,7 @@ from unittest.mock import (
 )
 
 from datetime import datetime
+from functools import partial
 
 class SomeClass(object):
     def one(self, a, b):
@@ -871,6 +872,19 @@ class SpecSignatureTest(unittest.TestCase):
         mocked.assert_called_once_with(4, 5, 6)
 
 
+    def test_autospec_getattr_partial_function(self):
+        # bpo-32153 : getattr returning partial functions without
+        # __name__ should not create AttributeError in create_autospec
+        class Foo:
+
+            def __getattr__(self, attribute):
+                return partial(lambda name: name, attribute)
+
+        proxy = Foo()
+        autospec = create_autospec(proxy)
+        self.assertFalse(hasattr(autospec, '__name__'))
+
+
 class TestCallList(unittest.TestCase):
 
     def test_args_list_contains_call_list(self):