基于pytest+requests+allure+yaml实现接口主动化测试框架

1.项目背景和目的：

   这是一个基于Python的接口主动化测试框架，主要目的是：
  - 提供一个稳定、可维护的接口测试办理方案
  - 实现测试用例与测试数据的解耦
  - 提供具体的测试陈诉和日志记载
  - 支持多环境配置和机动的用例管理
  2.技术栈选型

  框架接纳了以下核心技术：
  pytest：作为测试实验引擎，提供了强大的用例管理和参数化能力
  requests：处理HTTP请求，支持各种接口调用场景
  allure：天生雅观的测试陈诉，支持具体的测试步骤记载
  yaml：管理配置文件和测试数据，提供更好的可读性和维护性
  logging：实现完备的日志记载体系
  
  3.项目分层架构计划

  让我具体介绍这个项目的分层架构计划：
  

  

• 测试用例层 (testcases/) - test_api.py：测试用例实现
     - 用例编写和管理
     - 测试流程控制
     - 测试数据参数化
     - 断言结果验证
  
  

1. import pytest
2. import allure
3. import json
4. import time
5. from datetime import datetime
6. from common.http_client import HTTPClient
7. from common.assertions import Assertions
8. from common.yaml_handler import YamlHandler
9. from common.variable_handler import VariableHandler
11. class TokenManager:
12.     def __init__(self):
13.         self.token = None
14.         self.expire_time = None
15.         self.yaml_handler = YamlHandler()
16.         self.token_config = self.yaml_handler.read_yaml("data/test_data.yaml")["token_config"]
18.     def set_token(self, token, expire_time=None):
19.         """
20.         设置token和过期时间
21.         :param token: token字符串
22.         :param expire_time: 过期时间戳（秒）
23.         """
24.         self.token = token
25.         if expire_time:
26.             self.expire_time = expire_time
27.         else:
28.             # 如果没有提供过期时间，使用配置的默认过期时间
29.             self.expire_time = time.time() + self.token_config["expire_time"]
31.     def is_token_valid(self):
32.         """
33.         检查token是否有效
34.         :return: bool
35.         """
36.         if not self.token or not self.expire_time:
37.             return False
38.         
39.         # 检查是否接近过期时间
40.         remaining_time = self.expire_time - time.time()
41.         return remaining_time > self.token_config["refresh_before_expire"]
43.     def get_token(self):
44.         """
45.         获取token
46.         :return: token字符串
47.         """
48.         return self.token
50. @allure.epic("API自动化测试")
51. class TestAPI:
52.     def setup_class(self):
53.         self.http_client = HTTPClient()
54.         self.assertions = Assertions()
55.         self.test_data = YamlHandler.read_yaml("data/test_data.yaml")
56.         self.token_manager = TokenManager()
57.         self.variable_handler = VariableHandler()
58.    
59.     def refresh_token(self):
60.         """
61.         刷新token
62.         """
63.         with allure.step("刷新token"):
64.             # 获取登录测试用例数据
65.             login_case = next(
66.                 case for case in self.test_data["test_cases"]
67.                 if case["path"] == "/api/login"
68.             )
69.             
70.             # 发送登录请求
71.             response = self.http_client.request(
72.                 method=login_case["method"],
73.                 path=login_case["path"],
74.                 json=login_case["data"]
75.             )
76.             
77.             # 验证登录响应
78.             assert response.status_code == 200, "登录失败，无法刷新token"
79.             
80.             # 获取新token和过期时间
81.             response_data = response.json()
82.             new_token = response_data.get("token")
83.             expire_time = response_data.get("expire_time")  # 假设接口返回过期时间戳
84.             
85.             # 更新token管理器
86.             self.token_manager.set_token(new_token, expire_time)
87.             allure.attach("Token已刷新", f"新token: {new_token}", allure.attachment_type.TEXT)
88.    
89.     @allure.feature("API测试用例")
90.     @pytest.mark.parametrize("case", YamlHandler.read_yaml("data/test_data.yaml")["test_cases"])
91.     def test_api(self, case):
92.         """API测试用例"""
93.         allure.dynamic.title(case["case_name"])
94.         allure.dynamic.description(f"测试接口: {case['path']}")
95.         
96.         # 如果是需要token的接口，检查token是否需要刷新
97.         if "headers" in case and "Authorization" in case["headers"]:
98.             if not self.token_manager.is_token_valid():
99.                 self.refresh_token()
100.         
101.         # 替换请求数据中的变量
102.         if "headers" in case:
103.             case["headers"] = self.variable_handler.replace_variables(case["headers"])
104.         if "data" in case:
105.             case["data"] = self.variable_handler.replace_variables(case["data"])
106.         
107.         with allure.step(f"准备请求数据"):
108.             headers = case.get("headers", {})
109.             if "Authorization" in headers:
110.                 headers["Authorization"] = headers["Authorization"].format(
111.                     token=self.token_manager.get_token()
112.                 )
113.             allure.attach(
114.                 json.dumps(case.get("data", {}), ensure_ascii=False, indent=2),
115.                 "请求数据",
116.                 allure.attachment_type.JSON
117.             )
118.         
119.         # 发送请求
120.         with allure.step(f"发送{case['method']}请求到 {case['path']}"):
121.             request_kwargs = {
122.                 'method': case["method"],
123.                 'path': case["path"],
124.                 'service': case.get("service"),
125.                 'headers': case.get("headers")
126.             }
127.             
128.             # 添加请求数据
129.             if case.get("data"):
130.                 if case["method"].upper() == "GET":
131.                     request_kwargs['data'] = case["data"]
132.                 else:
133.                     request_kwargs['json'] = case["data"]
134.             
135.             response = self.http_client.request(**request_kwargs)
136.             allure.attach(
137.                 json.dumps(response.json(), ensure_ascii=False, indent=2),
138.                 "响应数据",
139.                 allure.attachment_type.JSON
140.             )
141.         
142.         # 保存需要的变量
143.         if "save_variables" in case and response.status_code == 200:
144.             with allure.step("保存接口返回变量"):
145.                 self.variable_handler.save_variables(
146.                     response.json(),
147.                     case["save_variables"]
148.                 )
149.         
150.         # 断言处理
151.         expected = case["expected"]
152.         with allure.step("执行断言"):
153.             if "status_code" in expected:
154.                 with allure.step(f"验证状态码: 期望 {expected['status_code']}"):
155.                     self.assertions.assert_status_code(response, expected["status_code"])
156.             
157.             if "contains_fields" in expected:
158.                 for field in expected["contains_fields"]:
159.                     with allure.step(f"验证字段存在: {field}"):
160.                         self.assertions.assert_contains_field(response, field)
161.             
162.             if "field_values" in expected:
163.                 for field, value in expected["field_values"].items():
164.                     with allure.step(f"验证字段值: {field} = {value}"):
165.                         self.assertions.assert_field_value(response, field, value)
166.         
167.         # 如果是登录接口，保存token
168.         if case["path"] == "/api/login" and response.status_code == 200:
169.             with allure.step("保存登录token"):
170.                 response_data = response.json()
171.                 self.token_manager.set_token(

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。