简介:pytest-asyncio是一个pytest插件。它便于测试使用异步库的代码。具体来说,pytest-asyncio提供了对作为测试函数的协同程序的支持。这允许用户在测试中等待代码。
历史攻略:
Python:aiomultiprocess实现协程与多进程的强强联合
安装:
pip install pytest-asyncio
使用案例:aysncio_demo.py
# -*- coding: utf-8 -*-
# time: 2024/4/5 12:06
# file: asyncio_demo.py
# 公众号: 玩转测试开发
import asyncio
import datetime
async def do_somethings(user):
# 1、定义协程函数
print(f"{user} is do_somethings. the time is {datetime.datetime.now()}")
await asyncio.sleep(0.1)
return user + " is finish do_somethings."
async def do_another(user):
print(f"{user} is do_another. the time is {datetime.datetime.now()}")
await asyncio.sleep(0.2)
return user + " is finish do_another."
async def do_others(user):
print(f"{user} is do_others. the time is {datetime.datetime.now()}")
return user + " is finish do_others."
if __name__ == '__main__':
# 3、加入事件循环。
tasks = []
for i in range(3):
tasks.append(do_somethings("tom"))
tasks.append(do_another("ken"))
tasks.append(do_others("lily"))
asyncio.run(asyncio.wait(tasks))
test_demo.py
# -*- coding: utf-8 -*-
# time: 2024/3/31 10:34
# file: test_demo.py
# 公众号: 玩转测试开发
import sys
import pytest
from lib.asyncio_demo import do_somethings, do_another, do_others
from logger import log
@pytest.mark.asyncio
async def test_do_somethings():
user = "tom"
res = await do_somethings(user)
log.info(f"{res}")
pytest.assume(user + " is finish do_somethings." in res)
@pytest.mark.asyncio
async def test_do_another():
user = "ken"
res = await do_another(user)
log.info(f"{res}")
pytest.assume(user + " is finish do_another." in res)
@pytest.mark.asyncio
async def test_do_others():
user = "lily"
res = await do_others(user)
log.info(f"{res}")
# 此次故意放一个错误试试。
pytest.assume(user + " is finish do_others." not in res)
运行结果: