如何在pytest生成JUnit XML结果文件后注入自定义逻辑？

核心思路是直接调用pytest junitxml插件的内部方法，在测试会话结束时强制生成完整的XML文件，之后再执行归档逻辑，完全在pytest内部完成，无需额外命令。

具体实现步骤 #

• 在项目的conftest.py中添加以下钩子函数：

  def pytest_sessionfinish(session):
      # 获取junitxml插件实例
      junit_plugin = session.config.pluginmanager.getplugin("_junitxml")
      # 检查插件是否启用且XML实例存在
      if junit_plugin and hasattr(junit_plugin, "xml"):
          # 强制触发XML文件的最终写入
          junit_plugin.xml.finalize(session)
          # 获取已生成的XML文件路径
          xml_file_path = junit_plugin.xml.log_file.name
          
          # 在这里添加你的私有CI归档逻辑
          # 例如：upload_to_private_ci(xml_file_path)
          # 替换成你实际的上传代码即可
  

原理说明 #

• pytest_sessionfinish是整个测试会话结束时触发的钩子，此时所有测试用例已经执行完毕。
• junitxml插件的finalize()方法原本会在pytest会话的最后阶段自动执行，负责将所有测试报告数据写入XML文件。我们手动调用这个方法，就能确保在执行归档逻辑前，XML文件已经完全生成并保存到磁盘。
• 该方案会自动识别用户是否指定了--junitxml参数，只有当参数存在时才会执行后续逻辑，不会影响正常的测试运行。

内容的提问来源于stack exchange，提问作者Leggy