This issue tracker has been migrated to GitHub, and is currently read-only.
For more information, see the GitHub FAQs in the Python's Developer Guide.

Title: Add async magic method support to unittest.mock.Mock
Type: enhancement Stage: resolved
Components: Library (Lib) Versions: Python 3.8
Status: closed Resolution:
Dependencies: Superseder:
Assigned To: lisroach Nosy List: Dima.Tisnek, benjamin.peterson, berker.peksag, brett.cannon, butla, jmehnle, lisroach, michael.foord, miss-islington, rbcollins, tsutsumi, txomon, xtreak, yan12125, yselivanov
Priority: normal Keywords: patch

Created on 2016-03-01 22:51 by brett.cannon, last changed 2022-04-11 14:58 by admin. This issue is now closed.

Pull Requests
URL Status Linked Edit
PR 9296 merged lisroach, 2018-09-14 07:17
PR 17130 merged Ilya.Kulakov, 2019-11-12 18:52
PR 17138 merged miss-islington, 2019-11-13 02:33
Messages (12)
msg261092 - (view) Author: Brett Cannon (brett.cannon) * (Python committer) Date: 2016-03-01 22:51
Mock doesn't support stuff like __aexit__ which means you can't mock out something like an async context manager.
msg267338 - (view) Author: Yusuke Tsutsumi (tsutsumi) * Date: 2016-06-04 22:46
I'm interested in taking this on. I'm considering for this approach, creating a new MagicProxy class specifically for async, and assigning __aexit__ and __aenter__ to those.

The __call__ method itself has to be detected as asynchronous, and I don't believe it's possible for a __call__ to behave appropriately for both asynchronous and synchronous cases.
msg267479 - (view) Author: Yusuke Tsutsumi (tsutsumi) * Date: 2016-06-05 21:51
Taking a look at the code, this may require a bit more discussion.

Mock's classes create copies of themselves when an attribute is accessed. As such, I can't just add __aexit__ and have it generate a different mock type entirely, unless I know exactly what an async variant of this mock is.

If there was a way to specifically provide a function for asynchronous situations, like __acall__, that would make this a lot easier as well.

Anyway, my proposal now is:

create new classes MockAsync and MagicMockAsync, and have MockAsync and MagicMockAsync implement the __aexit__ and __aenter__ methods.

How is that as an approach?
msg267569 - (view) Author: Brett Cannon (brett.cannon) * (Python committer) Date: 2016-06-06 22:57
There's also __aiter__, but it's semantics might be changing (issue #27243).

Robert or Michael, any opinions on Yusuke's proposal?
msg283657 - (view) Author: Javier Domingo (txomon) Date: 2016-12-20 00:22
I found this while trying to test an async context manager. This is a critical feature to enable migrations to async code, as the impossibility to test something properly is not acceptable in many environments.

Implementing it in a way that __call__ returns an object capable of being coroutine or normal function would avoid having to implement Async specific Mocks, wouldn't it? I am not too confident, but would it be doable to have an implementation that depends on whether _is_coroutine is accessed or not?

I don't like it, but I really don't like the fact that we need to patch different all the methods depending on whether they are coroutine or not.

Obviously, having the __acall__ method would really help solving this issue.
msg293644 - (view) Author: Michał Bultrowicz (butla) * Date: 2017-05-14 10:53
I guess that it's worth noting that the same problem is being talked about in an issue on the asynctest library:
msg333208 - (view) Author: Dima Tisnek (Dima.Tisnek) * Date: 2019-01-08 04:16
Perhaps it's possible to vendor asynctest mock in the same vein as `mock` found it's way into unittest?

The real power of `asynctest` is in constructs like:

with asynctest.mock.patch("module.Class", autospec=True):

Where mock instances know what methods are async.
msg342926 - (view) Author: Lisa Roach (lisroach) * (Python committer) Date: 2019-05-20 16:19
New changeset 77b3b7701a34ecf6316469e05b79bb91de2addfa by Lisa Roach in branch 'master':
bpo-26467: Adds AsyncMock for asyncio Mock library support (GH-9296)
msg342929 - (view) Author: Lisa Roach (lisroach) * (Python committer) Date: 2019-05-20 16:36
Added and AsyncMock class which supports mocking async magic methods.
msg356485 - (view) Author: Brett Cannon (brett.cannon) * (Python committer) Date: 2019-11-12 19:43
Adding Lisa to potentially add the PR from Ilya.
msg356511 - (view) Author: Benjamin Peterson (benjamin.peterson) * (Python committer) Date: 2019-11-13 02:33
New changeset d6d6e2aa0249bb661541705335ddbb97a53d64c8 by Benjamin Peterson (Ilya Kulakov) in branch 'master':
Add Ilya Kulakov to Misc/ACKS. (GH-17130)
msg356513 - (view) Author: miss-islington (miss-islington) Date: 2019-11-13 02:40
New changeset e5125f7b3b88d8d4814ed7bddbd4f34d24d763dd by Miss Islington (bot) in branch '3.8':
Add Ilya Kulakov to Misc/ACKS. (GH-17130)
Date User Action Args
2022-04-11 14:58:28adminsetgithub: 70654
2019-11-13 02:40:34miss-islingtonsetnosy: + miss-islington
messages: + msg356513
2019-11-13 02:33:34benjamin.petersonsetnosy: + benjamin.peterson
messages: + msg356511
2019-11-13 02:33:17miss-islingtonsetpull_requests: + pull_request16648
2019-11-12 19:43:16brett.cannonsetassignee: lisroach
messages: + msg356485
2019-11-12 18:52:16Ilya.Kulakovsetpull_requests: + pull_request16639
2019-05-20 16:36:26lisroachsetstatus: open -> closed

messages: + msg342929
stage: patch review -> resolved
2019-05-20 16:19:58lisroachsetnosy: + lisroach
messages: + msg342926
2019-01-08 05:02:00xtreaksetnosy: + xtreak
2019-01-08 04:16:26Dima.Tisneksetnosy: + Dima.Tisnek

messages: + msg333208
versions: + Python 3.8, - Python 3.6, Python 3.7
2018-10-10 23:54:39yselivanovsetnosy: + yselivanov
2018-09-14 07:17:42lisroachsetkeywords: + patch
stage: test needed -> patch review
pull_requests: + pull_request8725
2018-08-24 00:28:36jmehnlesetnosy: + jmehnle
2017-11-04 17:45:18yan12125setnosy: + yan12125
2017-05-14 10:53:17butlasetnosy: + butla
messages: + msg293644
2017-02-14 09:38:38berker.peksagsetnosy: + berker.peksag

versions: + Python 3.7
2016-12-20 00:22:47txomonsetnosy: + txomon
messages: + msg283657
2016-06-06 22:57:03brett.cannonsetnosy: + rbcollins
messages: + msg267569
2016-06-05 21:51:27tsutsumisetmessages: + msg267479
2016-06-04 22:46:43tsutsumisetnosy: + tsutsumi
messages: + msg267338
2016-03-01 22:51:46brett.cannoncreate