[3.8] bpo-39082: Allow AsyncMock to correctly patch static/class methods (GH-18190)

(cherry picked from commit 62865f4532)

Co-authored-by: Matthew Kokotovich <mkokotovich@gmail.com>
This commit is contained in:
Matthew Kokotovich 2020-01-26 09:30:27 -06:00 committed by Chris Withers
parent eebcff8c07
commit 19be85c765
3 changed files with 26 additions and 0 deletions

View File

@ -48,6 +48,8 @@ _safe_super = super
def _is_async_obj(obj):
if _is_instance_mock(obj) and not isinstance(obj, AsyncMock):
return False
if hasattr(obj, '__func__'):
obj = getattr(obj, '__func__')
return asyncio.iscoroutinefunction(obj) or inspect.isawaitable(obj)

View File

@ -19,6 +19,15 @@ class AsyncClass:
def normal_method(self):
pass
@classmethod
async def async_class_method(cls):
pass
@staticmethod
async def async_static_method():
pass
class AwaitableClass:
def __await__(self):
yield
@ -71,6 +80,20 @@ class AsyncPatchDecoratorTest(unittest.TestCase):
test_async()
def test_is_AsyncMock_patch_staticmethod(self):
@patch.object(AsyncClass, 'async_static_method')
def test_async(mock_method):
self.assertIsInstance(mock_method, AsyncMock)
test_async()
def test_is_AsyncMock_patch_classmethod(self):
@patch.object(AsyncClass, 'async_class_method')
def test_async(mock_method):
self.assertIsInstance(mock_method, AsyncMock)
test_async()
def test_async_def_patch(self):
@patch(f"{__name__}.async_func", return_value=1)
@patch(f"{__name__}.async_func_args", return_value=2)

View File

@ -0,0 +1 @@
Allow AsyncMock to correctly patch static/class methods