我已经使用 aiohttp 编写了一个简单的 HTTP 客户端,我正在尝试通过修补 aiohttp.ClientSession
和 aiohttp.ClientResponse
来测试它。但是,似乎 unittest.mock.patch
装饰器没有遵守我的异步代码。我猜测,这是某种命名空间不匹配。
这是一个最小的例子:
from aiohttp import ClientSession
async def is_ok(url:str) -> bool:
async with ClientSession() as session:
async with session.request("GET", url) as response:
return (response.status == 200)
我正在使用异步装饰器进行测试,如 this answer 中所述.所以这是我尝试的测试:
import unittest
from unittest.mock import MagicMock, patch
from aiohttp import ClientResponse
from my.original.module import is_ok
class TestClient(unittest.TestCase):
@async_test
@patch("my.original.module.ClientSession", spec=True)
async def test_client(self, mock_client):
mock_response = MagicMock(spec=ClientResponse)
mock_response.status = 200
async def _mock_request(*args, **kwargs):
return mock_response
mock_client.request = mock_response
status = await is_ok("foo")
self.assertTrue(status)
我的 is_ok
协同程序在用于 __main__
时工作正常,但是当我运行测试时,它给我一个错误,表明 session.request
函数未根据我的 patch
调用进行模拟。 (特别是它说“无法从 URL 'foo' 解析主机名”,如果它没有被 mock ,它应该是这样。)
我无法逃避这种行为。我试过:
- 在模拟完成后导入
is_ok
。 - 将mock分配给
mock_client
和mock_client.__aenter__
的各种组合,将mock_client.request
设置为MagicMock(return_value=mock_response)
,或者使用mock_client().request
等 - 使用特定的
__aenter__
和__aexit__
方法编写模拟ClientSession
并在new
参数中使用它补丁
。
这些似乎都没有什么不同。如果我将断言放入 is_ok
以测试 ClientSession
是 MagicMock
的实例,那么当我运行测试时这些断言会失败(再次,他们会在代码未打补丁时)。这引出了我的命名空间不匹配理论:也就是说,事件循环在 patch
所针对的不同命名空间中运行。
要么那样,要么我在做一些愚蠢的事情!
最佳答案
关于python - 使用 unittest.mock.patch 测试 aiohttp 客户端,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46450545/