43 lines
1.2 KiB
Diff
43 lines
1.2 KiB
Diff
---
|
|
Lib/unittest/mock.py | 18 ++++++++++
|
|
Misc/NEWS.d/next/Library/2022-10-08-19-39-27.gh-issue-98086.y---WC.rst | 1
|
|
2 files changed, 19 insertions(+)
|
|
|
|
--- a/Lib/unittest/mock.py
|
|
+++ b/Lib/unittest/mock.py
|
|
@@ -1595,6 +1595,12 @@ class _patch_dict(object):
|
|
def __call__(self, f):
|
|
if isinstance(f, type):
|
|
return self.decorate_class(f)
|
|
+ if inspect.iscoroutinefunction(f):
|
|
+ return self.decorate_async_callable(f)
|
|
+ return self.decorate_callable(f)
|
|
+
|
|
+
|
|
+ def decorate_callable(self, f):
|
|
@wraps(f)
|
|
def _inner(*args, **kw):
|
|
self._patch_dict()
|
|
@@ -1603,6 +1609,18 @@ class _patch_dict(object):
|
|
finally:
|
|
self._unpatch_dict()
|
|
|
|
+ return _inner
|
|
+
|
|
+
|
|
+ def decorate_async_callable(self, f):
|
|
+ @wraps(f)
|
|
+ async def _inner(*args, **kw):
|
|
+ self._patch_dict()
|
|
+ try:
|
|
+ return await f(*args, **kw)
|
|
+ finally:
|
|
+ self._unpatch_dict()
|
|
+
|
|
return _inner
|
|
|
|
|
|
--- /dev/null
|
|
+++ b/Misc/NEWS.d/next/Library/2022-10-08-19-39-27.gh-issue-98086.y---WC.rst
|
|
@@ -0,0 +1 @@
|
|
+Make sure ``patch.dict()`` can be applied on async functions.
|