在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)

这可行，但看起来很丑。有更多的pythonic方法可以做到这一点吗？

Answer 1

解决方案实际上非常简单：我只需要将__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