如何使用以下命令模拟从一个本机协程到另一个协程的异步调用unittest.mock.patch?
我目前有一个相当尴尬的解决方案：
class CoroutineMock(MagicMock):
def __await__(self, *args, **kwargs):
future = Future()
future.set_result(self)
result = yield from future
return result
Then
class TestCoroutines(TestCase):
@patch('some.path', new_callable=CoroutineMock)
def test(self, mock):
some_action()
mock.assert_called_with(1,2,3)
这可行但看起来很丑。有没有更多的Pythonic方法来做到这一点？
解决方案实际上很简单：
我只需要转换__call__模拟进入协程的方法：
class AsyncMock(MagicMock):
async def __call__(self, *args, **kwargs):
return super(AsyncMock, self).__call__(*args, **kwargs)
这完美地工作，当调用模拟时，代码接收本机协程
用法示例：
@mock.patch('my.path.asyncio.sleep', new_callable=AsyncMock)
def test_stuff(sleep):
# code
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
