### 构建和使用 Python pytest YAML 接口自动化测试框架 #### 测试框架概述 为了创建一个高效且易于维护的接口自动化测试框架，可以采用`requests`库来发送HTTP请求，通过`pytest`管理测试用例并执行测试流程，而测试数据则可以通过YAML文件定义[^1]。 #### 框架设计原则 该框架的设计遵循了分层思想以及数据驱动的理念，实现了代码逻辑同测试数据的有效分离。这种架构不仅提高了项目的可读性和扩展性，还使得新成员能够快速上手操作[^2]。 #### 关键组件说明 ##### commons模块 这是整个框架的核心部分，包含了多个子模块用于支持不同功能需求： - **cases.py**: 负责加载由YAML编写的测试用例； - **databases.py**: 提供与数据库交互的能力； - **exchange.py**: 实现数据之间的转换处理； - **files.py**: 处理各类文件的操作如读取/写入等； - **funcs.py**: 支持动态加载自定义函数； - **models.py**: 明确规定了测试案例应该遵循的标准格式； - **session.py**: 对标准版`requests.Session()`进行了增强改造； - **settings.py**: 解析外部传入的各种参数设置； - **templates.py**: 嵌套模板渲染机制以便灵活调整输出内容； 以上所有这些都位于独立于具体业务场景之外的位置，从而方便后期迭代更新而不影响现有功能。 ##### requirements.txt 记录着项目所需安装的所有第三方包版本号信息，确保环境一致性的同时也便于他人部署运行本工具集。 ##### run.py 作为程序启动入口点，在这里初始化全局变量、导入必要的类方法，并最终调用`pytest.main()`触发实际测试过程。 ##### data目录 专门用来保存敏感资料比如密钥证书之类的静态资源文件夹。 ##### pytest.ini 配置项集合体，允许用户修改默认行为或者指定额外插件选项等等。 ##### testcases路径 放置所有的API描述文档即`.yml/.yaml`结尾形式的具体实例化表达式所在位置。 ```python import os from pathlib import Path import pytest import requests import yaml def load_yaml(file_path): with open(file_path, 'r', encoding='utf8') as f: return yaml.safe_load(f) @pytest.fixture(scope="module") def api_session(): session = requests.session() yield session session.close() if __name__ == "__main__": current_dir = str(Path(__file__).parent.absolute()) case_files = [ os.path.join(current_dir, "testcases", file_name) for file_name in os.listdir(os.path.join(current_dir, "testcases")) if file_name.endswith(".yaml") or file_name.endswith(".yml") ] for case_file in case_files: cases_data = load_yaml(case_file) @pytest.mark.parametrize( argnames=["case"], argvalues=[(item,) for item in cases_data], ids=[f"{i['id']}" for i in cases_data] ) class TestApi: def test_api(self, api_session, case): response = getattr(api_session, case["method"].lower())( url=case.get('url'), headers=case.get('headers'), params=case.get('params'), json=case.get('json') ) assert response.status_code == int(case["expected"]["status"]) ``` 此段脚本展示了如何遍历给定目录下的每一个.yaml/yml文件并将它们转化为对应的测试单元。对于每个这样的条目而言，则会依据其内部结构自动构造相应的断言语句来进行验证工作。