以下是一个完整的接口自动化测试框架代码示例，包含请求封装、断言封装、日志记录、配置管理、报告生成等功能模块。该框架使用了`requests`库来发送HTTP请求，并结合`unittest`或`pytest`进行测试用例管理。 ### 接口自动化测试框架代码 ```python import os import logging import requests import unittest from configparser import ConfigParser from datetime import datetime # 配置日志 logging.basicConfig( level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s', handlers=[ logging.FileHandler(f"logs/{datetime.now().strftime('%Y-%m-%d')}.log"), logging.StreamHandler() ] ) # 配置管理 class ConfigManager: def __init__(self, config_file): self.config = ConfigParser() self.config.read(config_file) def get(self, section, option): return self.config.get(section, option) # 请求封装 class RequestHandler: def __init__(self, base_url): self.base_url = base_url def send_request(self, method, endpoint, headers=None, params=None, data=None, json=None, files=None): url = f"{self.base_url}{endpoint}" try: response = requests.request(method, url, headers=headers, params=params, data=data, json=json, files=files) response.raise_for_status() # 检查请求是否成功 return response.json() if response.headers.get('Content-Type', '').startswith('application/json') else response.text except requests.exceptions.RequestException as e: logging.error(f"Request failed: {e}") return None # 断言封装 class AssertionHandler: @staticmethod def assert_equal(actual, expected, message=""): assert actual == expected, f"{message} - Expected: {expected}, Actual: {actual}" @staticmethod def assert_in(substring, string, message=""): assert substring in string, f"{message} - Substring: {substring} not found in String: {string}" # 测试用例基类 class BaseTestCase(unittest.TestCase): def setUp(self): config = ConfigManager("config.ini") self.base_url = config.get("API", "base_url") self.request_handler = RequestHandler(self.base_url) def test_example(self): response = self.request_handler.send_request("GET", "/example_endpoint") AssertionHandler.assert_equal(response["status"], "success", "Status mismatch") # 报告生成（简单实现） def generate_report(test_results): report_file = f"reports/report_{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.txt" with open(report_file, "w") as f: f.write(f"Test Report Generated at {datetime.now()}\n\n") for result in test_results: f.write(f"{result}\n") logging.info(f"Report generated: {report_file}") if __name__ == "__main__": # 运行测试用例并生成报告 suite = unittest.TestLoader().loadTestsFromTestCase(BaseTestCase) test_results = [] runner = unittest.TextTestRunner(verbosity=2) result = runner.run(suite) for test, error in result.errors + result.failures: test_results.append(f"{test.id()}: Failed - {error}") for test in result.successes: test_results.append(f"{test.id()}: Passed") generate_report(test_results) ``` --- ### 解释 1. **日志管理**： - 使用`logging`模块记录运行日志，支持文件和控制台输出。 - 日志文件按日期命名，方便后续排查问题。 2. **配置管理**： - 使用`ConfigParser`读取配置文件（如`config.ini`），便于管理环境变量（如API基础URL）。 3. **请求封装**： - `RequestHandler`类封装了`requests`库的功能，简化了HTTP请求的发送。 - 支持GET、POST等常见方法，并能处理JSON响应或其他类型响应。 4. **断言封装**： - `AssertionHandler`类提供了通用的断言方法，避免在每个测试用例中重复编写断言逻辑。 5. **测试用例基类**： - 继承自`unittest.TestCase`，提供了一个标准的测试用例模板。 - 在`setUp`方法中初始化必要的资源（如请求处理器）。 6. **报告生成**： - 简单实现了测试报告功能，将测试结果写入文本文件。 - 可以扩展为HTML报告或其他格式。 --- ###