[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:
parent
eebcff8c07
commit
19be85c765
|
@ -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)
|
||||
|
||||
|
||||
|
|
|
@ -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)
|
||||
|
|
|
@ -0,0 +1 @@
|
|||
Allow AsyncMock to correctly patch static/class methods
|
Loading…
Reference in New Issue