经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » 程序设计 » Python3 » 查看文章
Pytest 框架执行用例流程浅谈
来源:cnblogs  作者:空慧居士  时间:2023/8/30 9:12:06  对本文有异议

背景:

  根据以下简单的代码示例,我们将从源码的角度分析其中的关键加载执行步骤,对pytest整体流程架构有个初步学习。

代码示例:

  1. import pytest
  2.  
  3. def test_add():
  4.  
  5. assert 1 + 1 == 2
  6.  
  7. def test_sub():
  8.  
  9. assert 2 - 1 == 1

  通过 pytest test_example.py 运行此代码示例后,会触发pytest的入口函数main(),这个函数定义在src/pytest/__main__.py中,它的作用是创建一个PytestConfig对象,并调用其

do_configure()和do_main()方法。PytestConfig对象是pytest的核心配置类,它负责解析命令行参数、读取配置文件、注册插件、创建Session对象等。PytestConfig对象定义在

src/_pytest/config/__init__.py中,它继承了pluggy.HookimplMarker类,也就是说它可以作为一个插件管理器,调用各种hook函数。

  1. ```python
  2. # src/pytest/__main__.py
  3.  
  4. def main():
  5. # 创建PytestConfig对象
  6. config = PytestConfig()
  7. # 调用config.do_configure()方法
  8. config.do_configure()
  9. # 调用config.do_main()方法
  10. config.do_main()
  11. ```
  12.  
  13. ```python
  14. # src/_pytest/config/__init__.py
  15.  
  16. class PytestConfig(pluggy.HookimplMarker):
  17. def __init__(self):
  18. # 解析命令行参数
  19. self.parse_args()
  20. # 读取配置文件
  21. self.read_config_files()
  22. # 注册插件
  23. self.register_plugins()
  24. # 创建Session对象
  25. self.session = Session(self)
  26. def do_configure(self):
  27. # 调用hook函数pytest_configure
  28. self.hook.pytest_configure(config=self)
  29. def do_main(self):
  30. # 调用hook函数pytest_sessionstart
  31. self.hook.pytest_sessionstart(session=self.session)
  32. # 调用Session对象的main()方法
  33. self.session.main()
  34. ```
 

  Session对象是pytest的核心上下文类,它负责管理整个测试过程的信息,包括收集测试用例、执行测试用例、生成测试报告等。Session对象定义在src/_pytest/main.py中,它继承了

Collector类,也就是说它可以作为一个测试用例收集器。Session对象的main()方法是执行测试用例的主要入口,它会调用perform_collect()方法来收集测试用例,并返回一个列表items;然后

调用runtestloop()方法来循环执行items中的每个Item对象;最后调用hook函数pytest_sessionfinish来结束测试会话,并返回一个退出码exitstatus。

  1. ```python
  2. # src/_pytest/main.py
  3.  
  4. class Session(Collector):
  5. def __init__(self, config):
  6. # 初始化一些属性和状态信息
  7. def main(self):
  8. # 调用perform_collect()方法收集测试用例,并返回items列表
  9. items = self.perform_collect()
  10. # 调用runtestloop()方法循环执行items中的每个Item对象,并返回退出码exitstatus
  11. exitstatus = self.runtestloop(items)
  12. # 调用hook函数pytest_sessionfinish来结束测试会话,并返回退出码exitstatus
  13. self.hook.pytest_sessionfinish(session=self, exitstatus=exitstatus)
  14. return exitstatus
  15. def perform_collect(self):
  16. # 调用hook函数pytest_collectstart表示开始收集测试用例
  17. self.hook.pytest_collectstart(collector=self)
  18. # 调用自身的collect()方法来递归遍历指定的测试文件或目录,并返回一个列表items
  19. items = self.collect()
  20. # 调用hook函数pytest_collectreport表示收集测试用例结束,并生成收集报告
  21. self.hook.pytest_collectreport(report=CollectReport(self, "passed", items))
  22. # 调用hook函数pytest_collection_modifyitems允许对收集到的Item对象进行修改
  23. self.hook.pytest_collection_modifyitems(session=self, config=self.config, items=items)
  24. # 调用hook函数pytest_deselected表示从收集到的Item对象中筛选出需要执行的Item对象
  25. self.hook.pytest_deselected(items=self.deselected)
  26. # 调用hook函数pytest_collection_finish表示收集和筛选测试用例完成,并返回最终要执行的Item对象列表
  27. self.hook.pytest_collection_finish(session=self)
  28. return items
  29. def runtestloop(self, items):
  30. # 调用hook函数pytest_runtestloop表示开始循环执行测试用例
  31. self.hook.pytest_runtestloop(session=self)
  32. # 遍历items列表,依次取出每个Item对象
  33. for item in items:
  34. # 调用Item对象的runtestprotocol()方法来执行单个测试用例的协议
  35. item.runtestprotocol()
  36. # 返回退出码0表示成功
  37. return 0
  38. ```

  

  Item对象是pytest的核心测试类,它负责封装和执行单个测试用例的信息,包括名称、位置、参数化信息、标记信息等。Item对象定义在src/_pytest/python.py中,它继承了Node类,也

就是说它可以作为一个测试节点。Item对象的runtestprotocol()方法是执行单个测试用例的主要入口,它会调用hook函数pytest_runtest_logstart来开始记录日志信息;然后调用runtest()方法

来执行测试用例的前置、主体和后置部分;最后调用hook函数pytest_runtest_logfinish来结束记录日志信息,并生成日志报告。

 

  1. ```python
  2. # src/_pytest/python.py
  3.  
  4. class Item(Node):
  5. def __init__(self, name, parent, config, session):
  6. # 初始化一些属性和状态信息
  7. def runtestprotocol(self):
  8. # 调用hook函数pytest_runtest_logstart表示开始记录日志信息
  9. self.ihook.pytest_runtest_logstart(nodeid=self.nodeid, location=self.location)
  10. # 调用runtest()方法来执行测试用例的前置、主体和后置部分,并返回一个列表reports
  11. reports = self.runtest()
  12. # 调用hook函数pytest_runtest_logfinish表示结束记录日志信息,并生成日志报告
  13. self.ihook.pytest_runtest_logfinish(nodeid=self.nodeid, location=self.location)
  14. return reports
  15. def runtest(self):
  16. # 创建一个空列表reports
  17. reports = []
  18. # 调用_setup()方法来执行测试用例的前置操作,并将返回的报告添加到reports列表中
  19. reports.append(self._setup())
  20. # 如果前置操作没有失败或跳过,则调用_call()方法来执行测试用例的主体部分,并将返回的报告添加到reports列表中
  21. if reports[-1].passed:
  22. reports.append(self._call())
  23. # 调用_teardown()方法来执行测试用例的后置操作,并将返回的报告添加到reports列表中
  24. reports.append(self._teardown())
  25. # 返回reports列表
  26. return reports
  27. def _setup(self):
  28. # 调用hook函数pytest_runtest_setup表示开始执行前置操作,并返回一个报告setup_report
  29. setup_report = self.ihook.pytest_runtest_setup(item=self)
  30. return setup_report
  31. def _call(self):
  32. # 调用hook函数pytest_runtest_call表示开始执行主体部分,并返回一个报告call_report
  33. call_report = self.ihook.pytest_runtest_call(item=self)
  34. return call_report
  35. def _teardown(self):
  36. # 调用hook函数pytest_runtest_teardown表示开始执行后置操作,并返回一个报告teardown_report
  37. teardown_report = self.ihook.pytest_runtest_teardown(item=self)
  38. return teardown_report
  39. ```

 

总结:

Pytest的加载流程大致如下:

- Pytest首先会解析命令行参数,确定要执行的测试文件、测试目录、测试类、测试函数等,以及一些配置选项。
- Pytest会根据配置文件(pytest.ini、setup.cfg、tox.ini等)和命令行参数,创建一个Config对象,用于存储配置信息。
- Pytest会创建一个Session对象,用于管理整个测试过程的上下文信息,包括收集测试用例、执行测试用例、生成测试报告等。
- Pytest会调用hook函数pytest_sessionstart,表示测试会话开始。
- Pytest会调用hook函数pytest_collectstart,表示开始收集测试用例。
- Pytest会根据Config对象中的信息,递归遍历指定的测试文件或目录,寻找符合pytest约定的测试用例(以test_开头的函数或方法,以Test开头的类等)。
- Pytest会将找到的测试用例封装成Item对象,并添加到Session对象的items列表中。Item对象包含了测试用例的名称、位置、参数化信息、标记信息等。
- Pytest会调用hook函数pytest_collectreport,表示收集测试用例结束,并生成收集报告。
- Pytest会调用hook函数pytest_collection_modifyitems,允许对收集到的Item对象进行修改,例如重新排序、添加或删除标记等。
- Pytest会调用hook函数pytest_deselected,表示从收集到的Item对象中筛选出需要执行的Item对象,并将不需要执行的Item对象放入Session对象的deselected列表中。
- Pytest会调用hook函数pytest_collection_finish,表示收集和筛选测试用例完成,并返回最终要执行的Item对象列表。
- Pytest会根据是否使用多进程或多线程模式,创建相应的WorkerController对象,用于管理多个Worker对象。Worker对象负责执行具体的测试用例,并将结果返回给WorkerController对象。
- Pytest会调用hook函数pytest_runtestloop,表示开始循环执行测试用例。
- Pytest会遍历Session对象中的items列表,依次取出每个Item对象,并调用hook函数pytest_runtest_protocol,表示开始执行单个测试用例的协议。
- Pytest会调用hook函数pytest_runtest_logstart,表示开始记录单个测试用例的日志信息。
- Pytest会调用hook函数pytest_runtest_setup,表示开始执行单个测试用例的前置操作(例如setup函数或方法)。
- Pytest会调用hook函数pytest_runtest_call,表示开始执行单个测试用例的主体部分(例如测试函数或方法)。
- Pytest会调用hook函数pytest_runtest_teardown,表示开始执行单个测试用例的后置操作(例如teardown函数或方法)。
- Pytest会调用hook函数pytest_runtest_logfinish,表示结束记录单个测试用例的日志信息,并生成日志报告。
- Pytest会调用hook函数pytest_runtest_makereport,表示根据单个测试用例的执行结果,生成测试报告(包括setup、call和teardown三个阶段的报告)。
- Pytest会重复上述步骤,直到所有的Item对象都被执行完毕。
- Pytest会调用hook函数pytest_sessionfinish,表示测试会话结束,并生成最终的测试报告(包括所有Item对象的报告)。
- Pytest会调用hook函数pytest_terminal_summary,表示在终端输出最终的测试结果和统计信息。

原文链接:https://www.cnblogs.com/beyond-tester/p/17665993.html

 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号