接口自动化测试实战：用 unittest+requests+pymysql 搭建数据驱动框架并接入 TaoToken

1. 接口自动化测试从零搭建为什么选 unittestrequestspymysql接口自动化测试说白了就三件事发请求、断言响应、校验数据。听起来简单但真正落到项目里很多人卡在第一步——框架怎么搭。我见过太多人上来就写一堆requests.get()脚本跑几次就乱了用例散落、数据写死、数据库校验靠手动查。这篇要解决的就是这个问题。用unittest组织用例、requests发请求、pymysql校验落库数据搭一个数据驱动的接口自动化框架。适合谁写过一点 Python、知道 HTTP 请求是怎么回事、但还没搭过完整测试框架的同学。如果你已经在用 Postman 手动点接口想往自动化走这篇也能直接跟做。框架的核心思路是分层配置层管环境地址和数据库连接工具层封装请求和数据库操作用例层只关心业务逻辑数据层用 YAML 或 JSON 驱动参数。这样换环境只改配置加用例只写数据不用动框架代码。另外接口测试绕不开鉴权。很多团队会把模型调用、AI 能力封装成内部接口测试时需要统一 Key 通道。这篇会演示怎么通过 TaoToken 的统一 API 通道完成鉴权配置把 Key 管理从用例里抽出来避免硬编码。官网在 https://taotoken.net/?utm_sourcetaotoken_aicg_blog_end API 入口是 https://taotoken.net/api 后面配置章节会具体写。先把目录结构定下来这是整个框架的骨架ihrm_auto_test/ ├── config/ │ ├── __init__.py │ └── settings.py # 环境地址、数据库配置、Key 配置 ├── common/ │ ├── __init__.py │ ├── db_util.py # pymysql 封装 │ ├── request_util.py # requests 封装 │ └── assert_util.py # 断言模板 ├── data/ │ └── login_case.yaml # 数据驱动用例数据 ├── testcase/ │ ├── __init__.py │ └── test_login.py # unittest 用例 ├── report/ # 测试报告输出 ├── run.py # 入口 └── requirements.txt这个结构不复杂但每一层职责清晰。下面逐个拆。2. TaoToken 前置统一 Key 与 API 通道准备在写请求封装之前先把鉴权通道理清楚。接口自动化测试里最烦的就是 Key 散落在各个用例里换一次 Key 要全局搜索替换。TaoToken 提供统一 Key 和 API 通道正好解决这个问题。你需要准备的东西第一一个可用的 Key。登录 TaoToken 控制台在 API Keys 页面创建。地址是 https://taotoken.net/console/api-keys 创建后复制保存页面只显示一次。第二确认 API 基础地址。TaoToken 的 API 入口是 https://taotoken.net/api 所有请求走这个 base_url具体路径按接口文档拼。文档在 https://taotoken.net/doc 。第三如果你要测的是模型对话类接口可以在 https://taotoken.net/models 先手动验证一下 Key 是否可用确认能正常返回再写进自动化用例。长期跑编码类 Agent 任务的话Coding Plan 页面 https://taotoken.net/coding-plan 有对应的套餐说明。Key 的管理原则绝对不写进用例代码统一放配置文件通过环境变量或配置文件读取。下面 config 层会具体实现。注意Key 属于敏感信息提交代码前确认.gitignore已经排除配置文件或者用环境变量注入。3. 可复制配置config 骨架与数据库工具封装3.1 settings.py 配置骨架配置层要解决三件事环境切换、数据库连接参数、鉴权 Key。用类属性加环境变量的方式既方便本地调试也方便 CI 注入。# config/settings.py import os class Config: # 被测服务基础地址 BASE_URL os.getenv(BASE_URL, https://taotoken.net/api) # TaoToken 统一 Key优先从环境变量读取 API_KEY os.getenv(TAOTOKEN_API_KEY, ) # 请求超时秒 TIMEOUT 10 # 数据库配置 DB_HOST os.getenv(DB_HOST, 127.0.0.1) DB_PORT int(os.getenv(DB_PORT, 3306)) DB_USER os.getenv(DB_USER, root) DB_PASSWORD os.getenv(DB_PASSWORD, 123456) DB_NAME os.getenv(DB_NAME, test_db) DB_CHARSET utf8mb4 class DevConfig(Config): BASE_URL https://taotoken.net/api class TestConfig(Config): BASE_URL https://taotoken.net/api # 当前环境通过环境变量切换 ENV os.getenv(ENV, dev) current_config DevConfig if ENV dev else TestConfig这样切换环境只需要改ENV环境变量不用动代码。Key 从环境变量读本地调试时在终端export TAOTOKEN_API_KEY你的key即可。3.2 pymysql 数据库工具封装数据库校验是接口测试的关键一环。接口返回成功不代表数据真的落库了必须查库确认。用 pymysql 封装一个工具类支持查询、事务提交和回滚。# common/db_util.py import pymysql from config.settings import current_config class DBUtil: _conn None classmethod def get_conn(cls): if cls._conn is None or not cls._conn.open: cls._conn pymysql.connect( hostcurrent_config.DB_HOST, portcurrent_config.DB_PORT, usercurrent_config.DB_USER, passwordcurrent_config.DB_PASSWORD, databasecurrent_config.DB_NAME, charsetcurrent_config.DB_CHARSET, autocommitFalse, cursorclasspymysql.cursors.DictCursor ) return cls._conn classmethod def query_one(cls, sql): conn cls.get_conn() cursor conn.cursor() try: cursor.execute(sql) return cursor.fetchone() finally: cursor.close() classmethod def query_all(cls, sql): conn cls.get_conn() cursor conn.cursor() try: cursor.execute(sql) return cursor.fetchall() finally: cursor.close() classmethod def execute(cls, sql): conn cls.get_conn() cursor conn.cursor() try: rows cursor.execute(sql) conn.commit() return rows except Exception as e: conn.rollback() raise e finally: cursor.close() classmethod def close(cls): if cls._conn and cls._conn.open: cls._conn.close() cls._conn None这里用了DictCursor查询结果直接是字典断言时按字段名取值比元组下标可读性好太多。事务默认不自动提交写操作手动commit出错rollback避免测试数据污染。3.3 requests 请求封装请求封装要处理 base_url 拼接、统一鉴权头、超时和日志。把 Key 注入放在这一层用例里就不用管鉴权了。# common/request_util.py import requests from config.settings import current_config class RequestUtil: _session None classmethod def get_session(cls): if cls._session is None: cls._session requests.Session() cls._session.headers.update({ Authorization: fBearer {current_config.API_KEY}, Content-Type: application/json }) return cls._session classmethod def request(cls, method, path, **kwargs): url current_config.BASE_URL.rstrip(/) / path.lstrip(/) kwargs.setdefault(timeout, current_config.TIMEOUT) session cls.get_session() resp session.request(method.upper(), url, **kwargs) return resp classmethod def get(cls, path, **kwargs): return cls.request(GET, path, **kwargs) classmethod def post(cls, path, **kwargs): return cls.request(POST, path, **kwargs) classmethod def close(cls): if cls._session: cls._session.close() cls._session None用Session的好处是连接复用和 Cookie 保持登录态接口测试特别有用。鉴权头统一在 Session 初始化时注入用例代码干净。3.4 断言模板断言不要只断状态码要断业务码、关键字段、数据库记录。封装几个常用断言方法# common/assert_util.py class AssertUtil: staticmethod def assert_status_code(resp, expected200): assert resp.status_code expected, \ f状态码不符期望 {expected}实际 {resp.status_code} staticmethod def assert_json_field(resp, field, expected): body resp.json() actual body.get(field) assert actual expected, \ f字段 {field} 不符期望 {expected}实际 {actual} staticmethod def assert_db_record(sql, field, expected): from common.db_util import DBUtil record DBUtil.query_one(sql) assert record is not None, f数据库未查到记录SQL: {sql} actual record.get(field) assert actual expected, \ f数据库字段 {field} 不符期望 {expected}实际 {actual}这三个断言覆盖了接口测试最常见的校验场景HTTP 层、业务层、数据层。4. 验证请求unittest 用例与冒烟全链路跑通4.1 数据驱动用例数据把用例参数抽到 YAML 里加用例只改数据不改代码。先装依赖pip install requests pymysql pyyaml数据文件# data/login_case.yaml - case_name: 正常登录 path: /v1/chat/completions method: post payload: model: gpt-4o-mini messages: - role: user content: 你好 expected_status: 200 expected_field: object expected_value: chat.completion4.2 unittest 用例编写用例层继承unittest.TestCase用setUpClass做一次初始化tearDownClass清理连接。# testcase/test_login.py import unittest import yaml import os from common.request_util import RequestUtil from common.assert_util import AssertUtil from common.db_util import DBUtil def load_cases(): path os.path.join(os.path.dirname(__file__), ../data/login_case.yaml) with open(path, r, encodingutf-8) as f: return yaml.safe_load(f) class TestLogin(unittest.TestCase): classmethod def setUpClass(cls): cls.cases load_cases() classmethod def tearDownClass(cls): RequestUtil.close() DBUtil.close() def test_smoke_chat(self): case self.cases[0] resp RequestUtil.post(case[path], jsoncase[payload]) AssertUtil.assert_status_code(resp, case[expected_status]) AssertUtil.assert_json_field(resp, case[expected_field], case[expected_value]) print(响应体:, resp.json()) if __name__ __main__: unittest.main()4.3 运行与结果设置好环境变量后运行export TAOTOKEN_API_KEY你的key python -m unittest testcase.test_login -v预期输出test_smoke_chat (testcase.test_login.TestLogin) ... ok ---------------------------------------------------------------------- Ran 1 test in 1.234s OK如果响应体里能看到object: chat.completion说明请求链路通了。这一步验证的是配置读取正常、Key 注入正常、请求发送正常、断言逻辑正常。4.4 加上数据库校验的完整冒烟如果你的接口有落库行为在用例里追加数据库断言。假设有个订单接口插入后查库确认def test_create_order_and_check_db(self): payload {user_id: 1001, amount: 99.9} resp RequestUtil.post(/v1/order/create, jsonpayload) AssertUtil.assert_status_code(resp, 200) order_id resp.json().get(data, {}).get(order_id) self.assertIsNotNone(order_id, 接口未返回 order_id) sql fSELECT amount FROM orders WHERE id {order_id} AssertUtil.assert_db_record(sql, amount, 99.9)这条用例把请求、响应断言、数据库校验串成一条链路跑通就说明框架核心能力没问题。5. 本篇常见错排查报错一pymysql.err.OperationalError: (2003, Cant connect to MySQL server)数据库地址或端口不对。检查DB_HOST和DB_PORT本地 MySQL 默认 3306。如果是容器环境确认端口映射。另外确认 MySQL 服务已启动systemctl status mysql看一眼。报错二requests.exceptions.MissingSchema: Invalid URLbase_url 拼接出问题。检查current_config.BASE_URL是否带协议头path是否以/开头。封装里已经做了rstrip(/)和lstrip(/)处理如果还报错打印拼接后的完整 URL 定位。报错三AssertionError: 字段 object 不符期望 chat.completion实际 None响应结构和你预期的不一样。先print(resp.text)看原始返回确认字段路径。有些接口返回是嵌套的比如data.object断言时要用resp.json()[data][object]封装里的assert_json_field只取顶层嵌套字段需要自己扩展。报错四KeyError: TAOTOKEN_API_KEY或鉴权 401环境变量没设置。确认export命令在当前终端生效或者直接在settings.py里临时写死调试记得提交前改回来。401 的话检查 Key 是否过期、是否复制完整、Bearer 前缀有没有多空格。报错五数据库查询返回 None但接口明明成功了事务没提交。如果被测服务用的是手动提交接口返回成功但数据还在事务里没落库查库自然查不到。确认服务端事务提交逻辑或者查库前加短暂等待。另外检查查询条件是否和插入数据一致比如自增 ID 对不上。报错六RuntimeError: Working outside of application context或连接池耗尽DBUtil用了类变量持有连接多线程跑用例时会冲突。如果要用unittest并发把连接改成每次请求新建、用完即关或者引入连接池。单线程跑冒烟没问题并发场景要注意。6. 接入文档与 Key 管理把鉴权从用例里彻底抽离框架跑通之后最后一步是把鉴权配置规范化。前面settings.py已经从环境变量读 Key但实际项目里还需要考虑多环境 Key 切换、Key 轮换、CI 注入。推荐做法本地开发用.env文件加python-dotenv加载CI 环境用平台的环境变量注入生产测试环境用配置中心。核心原则是 Key 永远不进代码仓库。TaoToken 的接入文档在 https://taotoken.net/doc 里面有各接口的路径、参数、返回结构说明。API Keys 管理页面在 https://taotoken.net/console/api-keys 创建和吊销 Key 都在这里操作。如果你要验证模型对话类接口https://taotoken.net/models 可以手动试跑确认 Key 有效。长期跑编码 Agent 自动化任务的话https://taotoken.net/coding-plan 有对应的方案说明。把 Key 管理抽离之后你的用例代码里不应该出现任何硬编码的 Key 或地址。所有环境相关的配置都从settings.py走这样换环境、换 Key、加用例都不需要动用例代码。框架的价值就在这里让写用例的人只关心业务让配置的事归配置。