在Python3.5中模拟异步调用

Mocking async call in python 3.5(在Python3.5中模拟异步调用)

问题描述

如何使用unittest.mock.patch模拟从一个本机协程到另一个本机协程的异步调用？

我目前有一个相当尴尬的解决方案：

class CoroutineMock(MagicMock):
    def __await__(self, *args, **kwargs):
        future = Future()
        future.set_result(self)
        result = yield from future
        return result

然后

class TestCoroutines(TestCase):
    @patch('some.path', new_callable=CoroutineMock)
    def test(self, mock):
        some_action()
        mock.assert_called_with(1,2,3)

这个不错，但看起来很难看。有没有更多的蟒蛇方式来做到这一点？

推荐答案

解决方案实际上非常简单： 我只需要将mock的__call__方法转换为协程：

class AsyncMock(MagicMock):
    async def __call__(self, *args, **kwargs):
        return super(AsyncMock, self).__call__(*args, **kwargs)

这非常有效，当调用mock时，代码接收本机协程

示例用法：

@mock.patch('my.path.asyncio.sleep', new_callable=AsyncMock)
def test_stuff(sleep):
    # code

这篇关于在Python3.5中模拟异步调用的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！