Python 10个接口自动化测试提效示例！

在接口自动化测试中，提高效率通常涉及减少重复代码、增加测试的可维护性和重用性，以及充分利用现有工具和框架。以下是一些关键场景及其示例代码，展示如何在Python中实现高效的接口自动化测试。为了简化，我们将使用requests库来处理HTTP请求，请确保已安装此库。

1. 使用Session对象减少请求延迟

场景: 重复使用Session对象以减少每次请求的连接时间。

import requestsdef test_api_with_session():    session = requests.Session()    response = session.get('https://api.example.com/users')    assert response.status_code == 200, f"Failed with status code {response.status_code}"    # 更多请求可以使用同一个session...# 调用示例# test_api_with_session()# 预期输出: 成功状态码200或错误信息

2. 参数化测试

场景: 使用参数化运行相同测试逻辑但不同的输入。

import pytestfrom requests import get@pytest.mark.parametrize("endpoint", ['/users', '/posts', '/comments'])def test_endpoints(endpoint):    response = get(f"https://api.example.com/{endpoint}")    assert response.status_code == 200, f"{endpoint} failed with status {response.status_code}"# 调用示例 (需要pytest框架)# pytest -v test_script.py# 预期输出: 各个endpoint的测试结果

3. 依赖注入模拟

场景: 在测试中模拟外部依赖，避免真实调用。

def test_api_with_mock(mock_get):    mock_get.return_value.status_code = 200    response = requests.get('https://api.example.com/users')    assert response.status_code == 200# 使用unittest.mock或pytest-mock进行模拟# 示例未包含完整mock设置，仅展示思路

4. 异常处理

场景: 确保接口正确处理异常情况。

def test_error_handling():    try:        response = requests.get('https://api.example.com/users/99999')  # 故意请求不存在的用户        assert response.status_code != 404, "Expected 404 for non-existent user"    except requests.exceptions.RequestException as e:        print(f"Request failed: {e}")# 调用示例# test_error_handling()# 预期输出: 404错误或请求失败信息

5. 使用Fixture共享测试资源

场景: 利用pytest Fixtures来共享初始化和清理资源。

import pytest@pytest.fixturedef auth_token():    # 获取或模拟一个认证令牌    return "mock-token"def test_protected_api(auth_token):    headers = {'Authorization': f'Bearer {auth_token}'}    response = requests.get('https://api.example.com/protected', headers=headers)    assert response.status_code == 200# pytest -v test_script.py# 预期输出: 成功状态码或错误信息

6. 并发测试

场景: 模拟多用户并发访问以测试接口性能。

import concurrent.futuresdef concurrent_requests():    urls = ['https://api.example.com/user/1', 'https://api.example.com/user/2']    with concurrent.futures.ThreadPoolExecutor() as executor:        results = list(executor.map(requests.get, urls))        for resp in results:            assert resp.status_code == 200, f"Failed with status {resp.status_code}"# 调用示例# concurrent_requests()# 预期输出: 多个请求的成功或失败状态

7. 数据驱动测试

场景: 使用外部数据文件（如CSV、JSON）驱动测试。

import csvimport requestsdef test_with_csv_data(filename):    with open(filename, newline='') as csvfile:        reader = csv.reader(csvfile)        next(reader)  # Skip header        for row in reader:            url, expected_status = row            response = requests.get(url)            assert response.status_code == int(expected_status), f"{url} failed"# 调用示例# test_with_csv_data('testdata.csv')# 预期输出: 每个URL的测试结果

8. 响应验证

场景: 不仅验证状态码，还验证响应体中的具体数据。

def test_response_content():    response = requests.get('https://api.example.com/users/1')    data = response.json()    assert response.status_code == 200    assert data['id'] == 1, "Incorrect user ID returned"    # 更多验证...# 调用示例# test_response_content()# 预期输出: 验证成功或失败信息

9. 基于条件的测试

场景: 根据环境变量或配置决定是否执行某些测试。

import osdef test_optional_feature():    if os.getenv('ENABLE_FEATURE_X', 'false').lower() == 'true':        response = requests.post('https://api.example.com/feature_x')        assert response.status_code == 201, "Feature X failed"# 调用示例# 设置环境变量 ENABLE_FEATURE_X=true 后运行测试# 预期输出: 成功或跳过测试的提示

10. 定时任务自动化

场景: 定时执行接口测试，如夜间自动回归测试。

import scheduleimport timedef run_daily_tests():    # 包含所有测试函数的调用    test_api_with_session()    test_endpoints()  # 假设已经定义好pytest相关函数    # 其他测试...schedule.every().day.at("00:00").do(run_daily_tests)  # 每天凌晨执行while True:    schedule.run_pending()    time.sleep(60)  # 每分钟检查一次# 实际部署时，此代码块应运行在一个长期运行的进程或服务器上# 预期输出: 日志记录每天的测试结果

这些示例涵盖了接口自动化测试中提高效率和可维护性的多种策略，包括代码复用、参数化、异常处理、并发测试、数据驱动等。在实际应用中，还需要结合具体的测试框架和工具进行调整和完善。

下面是配套资料，对于做【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴我走过了最艰难的路程，希望也能帮助到你！

最后： 可以在公众号：自动化测试老司机！ 免费领取一份216页软件测试工程师面试宝典文档资料。以及相对应的视频学习教程免费分享！，其中包括了有基础知识、Linux必备、Shell、互联网程序原理、Mysql数据库、抓包工具专题、接口测试工具、测试进阶-Python编程、Web自动化测试、APP自动化测试、接口自动化测试、测试高级持续集成、测试架构开发测试框架、性能测试、安全测试等。

如果我的博客对你有帮助、如果你喜欢我的博客内容，请 “点赞” “评论” “收藏” 一键三连哦！