如何模拟aiohttp.client.ClientSession.get异步上下文管理器

如何模拟aiohttp.client.ClientSession.get异步上下文管理器

在您的链接中，有一个编辑：

编辑：这篇文章中提到的GitHub问题已得到解决，从0.11.1版开始，asynctest开箱即用地支持异步上下文管理器。

由于

asynctest==0.11.1

已更改，因此，一个有效的示例是：

import randomfrom aiohttp import ClientSessionfrom asynctest import CoroutineMock, patchasync def get_random_photo_url():    while True:        async with ClientSession() as session: async with session.get('random.photos') as resp:     json = await resp.json()        photos = json['photos']        if not photos: continue        return random.choice(photos)['img_src']@patch('aiohttp.ClientSession.get')async def test_call_api_again_if_photos_not_found(mock_get):       mock_get.return_value.__aenter__.return_value.json = CoroutineMock(side_effect=[        {'photos': []}, {'photos': [{'img_src': 'a.jpg'}]}    ])    image_url = await get_random_photo_url()    assert mock_get.call_count == 2    assert mock_get.return_value.__aenter__.return_value.json.call_count == 2    assert image_url == 'a.jpg'

关键问题是您需要正确模拟函数

json

，因为默认情况下它是一个

MagicMock

实例。要访问此功能，您需要

mock_get.return_value.__aenter__.return_value.json

。