在python 3.5中模拟异步调用

Question

如何使用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)

这可以用，但看起来很难看。有没有更像Python的方法

Answer 1

解决方案其实很简单： 我只需要将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