pytest自动化测试
1 pytest自动化测试 - 我对测试用例超时处理的一点看法
1.1 背景
用例在执行过程中，可能由于网络等待，或者等待一些特殊的文件，而又由于一些异常，导致这些条件一直不能满足，用例卡死，这种情况在自动化测试中是不允许的，会浪费大量的时间，影响测试任务的进度，甚至导致版本发布周期的延长。
为了给每个用例都设置上超时时间，有以下几种方法：
1.2 方法1：给每个用例添加@pytest.mark.timeout修饰器
@pytest.mark.timeout在Windows下并不能很好的工作，1 个用例出现超时，整个测试任务就中断了，非常扯淡！
import pytestimport time@pytest.mark.timeout(5)def test_case1(): time.sleep(2) assert True@pytest.mark.timeout(5)def test_case2(): time.sleep(6)# 此用例会超时 assert Truedef test_case3(): print("用例3：Windows下不会执行") assert 3 == 3
1.3 方法2：使用钩子函数给每个用例添加一个timeout修饰器
然而，下面这个做法，在Windows下并没有什么卵用！，超时的那个用例也执行成功了！
import pytestimport timedef pytest_collection_modifyitems(session, config, items): for item in items: item.add_marker(pytest.mark.timeout(3))def setup_function(): print()def test_case3(): print("用例3：开始") time.sleep(3) assert True print("用例3：结束")def test_case4(): print("用例4：开始") time.sleep(5)# 此用例会超时，但Windows不超时，神奇！ assert True print("用例4：结束")def test_case5(): print("用例5: 开始") assert 3 == 3 print("用例5: 结束")
Windows下的执行结果：
============================= test session starts =============================platform win32 -- Python 3.13.1, pytest-8.3.4, pluggy-1.5.0rootdir: D:\TYYSOFT\Study\Python\pytestconfigfile: pytest.iniplugins: check-2.4.1, html-4.1.1, metadata-3.1.1, timeout-2.3.1, xdist-3.6.1collected 3 itemstest_timeout_004.py 用例3：开始用例3：结束.用例4：开始用例4：结束.用例5: 开始用例5: 结束.-- Generated html report: file:///D:/TYYSOFT/Study/Python/pytest/report.html --============================== 3 passed in 8.03s ==============================
其他所谓的使用timeout标记的超时处理在Windows下都没有什么卵用！
1.4 方法3：介绍一个自己写的使用线程处理用例超时的方法
用例的超时，其实主要是监控每条任务的超时情况，通过创建1个基础函数base_func这个函数每次调用时都创建1个线程，并给线程设置超时时间，用例中的执行语句都通过这个线程函数base_func来执行，base_func中对超时任务会设置失败。
import pytestimport threadingimport time# 核心函数：该函数创建1个线程，并执行给定的task，超时后会把用例设置为失败def base_func(task, timeout=5): thread = threading.Thread(target=task) thread.start() thread.join(timeout) #设置超时 if thread.is_alive(): print(f"::{task.__name__}: 语句执行超时...") assert False# 1个长时间任务def long_time_task(): time.sleep(2)# 模块发送消息的基础函数def send_msg(msg): print(msg)# 预置条件def setup_function(): print() base_func(print("预置条件！"))# 这是用例1def test_timeout_001(): base_func(send_msg("用例1开始")) base_func(long_time_task) base_func(send_msg("用例1结束"))def test_timeout_002(): base_func(send_msg("用例2开始")) base_func(send_msg("用例2结束"))
输出结果：
============================= test session starts =============================platform win32 -- Python 3.13.1, pytest-8.3.4, pluggy-1.5.0rootdir: D:\TYYSOFT\Study\Python\pytestconfigfile: pytest.iniplugins: check-2.4.1, html-4.1.1, metadata-3.1.1, timeout-2.3.1, xdist-3.6.1collected 2 itemstest_timeout_002.py 预置条件！用例1开始::long_time_task: 语句执行超时...F预置条件！用例2开始用例2结束.================================== FAILURES ===================================______________________________ test_timeout_001 _______________________________ def test_timeout_001(): base_func(send_msg("用例1开始"))> base_func(long_time_task)test_timeout_002.py:31: _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _task = <function long_time_task at 0x000001AAE4112DE0>, timeout = 5 def base_func(task, timeout=5): thread = threading.Thread(target=task) thread.start() thread.join(timeout) #设置超时 if thread.is_alive(): print(f"::{task.__name__}: 语句执行超时...")> assert FalseE assert Falsetest_timeout_002.py:13: AssertionError-- Generated html report: file:///D:/TYYSOFT/Study/Python/pytest/report.html --=========================== short test summary info ===========================FAILED test_timeout_002.py::test_timeout_001 - assert False========================= 1 failed, 1 passed in 5.11s =========================
两个用例都执行了， 第2个用例并没有因为第1个用例超时而中止执行！
使用基础函数来包装测试执行语句的必要性， 因为实际在测试执行时， 每个语句都有可能执行超时，对其进行封装执行可以保证测试脚本的严谨性。
<hr>作者声明：本文用于记录和分享作者的学习心得，可能有部分文字或示例来自AI平台，如：豆包、DeepSeek(硅基流动)(注册链接)等，由于本人水平有限，难免存在表达错误，欢迎留言交流和指教！
Copyright © 2022~2025 All rights reserved.
页:
[1]