模拟流API在python的单元测试

Question

我有一个调用流api的异步函数。为这个函数编写单元测试的最佳方法是什么？必须模拟api响应。

我尝试使用aiounittest，并使用unittest的mock。但是这调用了实际的api，而不是得到模拟的响应。也尝试pytest.mark.asyncio注释，但这一直给我的错误-协程从未等待。我已经验证pyest-asyncio已经安装。

我正在使用VS代码和Python 3.6.6

以下是相关的代码片段：

async def method1():
    response = requests.get(url=url, params=params, stream=True)
    for data in response.iter_lines():
        # processing logic here
        yield data

粘贴一些我尝试过的测试。

def mocked_get(*args, **kwargs):
#implementation of mock

class TestClass (unittest.TestCase):
    @patch("requests.get", side_effect=mocked_get)
    async def test_method (self, mock_requests):
        resp = []
        async for data in method1:
            resp.append (data)
    
        #Also tried await method1
    
        assert resp
    

还尝试了TestClass类（aiounittest.AsyncTestCase）：

Answer 1

使用asynctest代替aiounittest。

1. 更换unittest。带有异步测试的TestCase。测试用例

import asynctest
from asynctest.mock import patch


class TestClass(asynctest.TestCase):
    @patch("requests.get", side_effect=mocked_get)
    async def test_method(self, mock_requests):
        resp = []
        async for data in method1():
            resp.append(data)

        assert resp