用于 async 和 await 方法的 Python pytest 案例

Question

我正在尝试为以下异步、等待方法编写 pytest，但我无处可去。

   class UserDb(object):
    async def add_user_info(self,userInfo):
      return await self.post_route(route='users',json=userInfo)

    async def post_route(self,route=None,json=None,params=None):
      uri = self.uri + route if route else self.uri      
      async with self.client.post(uri,json=json,params=params) as resp:            
      assert resp.status == 200
      return await resp.json()

有人可以帮我解决这个问题吗？ TIA

Answer 1

pip install pytest-aiohttp，然后像这样创建一个fixture

from pytest import fixture

def make_app():
    app = Application()
    # Config your app here
    return app

@fixture
def test_fixture(loop, test_client):
    """Test fixture to be used in test cases"""
    app = make_app()
    return loop.run_until_complete(test_client(app))

现在写你的测试

f = test_fixture

async def test_add_user_info(f):
    resp = await f.get('/')
    assert resp.status == 200
    assert await resp.json() == {'some_key': 'some_value'}

我还注意到您的 add_user_info 协程没有返回任何内容。 更多信息是here