异步和等待方法的Python pytest案例

时间:2017-08-29 16:32:28

标签: python python-3.x async-await pytest aiohttp

我正在尝试为以下异步编写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

1 个答案:

答案 0 :(得分:7)

pip install pytest-aiohttp,然后像这样创建一个夹具

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