Python自动化测试框架搭建：Pytest+Selenium+Requests实战-代码聚汇网

Python自动化测试框架搭建：Pytest+Selenium+Requests实战

Python自动化测试PytestSelenium

|难度：普通|阅读约 14 分钟

1. 项目概述：Python自动化测试框架全景搭建指南

在软件研发效能提升的背景下，自动化测试已成为质量保障的核心手段。这个项目将手把手带你从零构建同时支持UI和接口测试的Python全栈自动化框架，涵盖从环境搭建到持续集成的完整链路。不同于市面上分散的教程，我们采用Pytest+Selenium+Requests技术栈，实现真正的企业级框架设计。

我曾为多家金融和电商企业搭建过测试框架，发现90%的团队在框架设计初期都会陷入工具选型混乱、用例管理无序的困境。本方案通过模块化设计解决这些问题：UI层采用PageObject模式降低维护成本，接口层实现自动化的契约测试，最终通过Jenkins实现测试调度的无人值守。

需要模型API调用？免费领10W Token，多模型网关一键接入 Claude、DeepSeek 等主流模型。

2. 框架技术选型与核心架构

2.1 基础技术栈组合方案

选择Python3.8+作为基础语言环境，主要考虑其丰富的测试生态和低学习门槛。核心组件包括：

测试引擎：Pytest（比unittest更灵活的fixture机制）
UI自动化：Selenium4 + WebDriverManager（自动管理浏览器驱动）
接口测试：Requests + Pytest-httpx（支持HTTP/2协议）
报告系统：Allure2（可视化测试报告生成）

重要提示：避免直接pip install安装所有包，建议使用requirements.txt分环境管理。我曾遇到因版本冲突导致元素定位失效的案例，解决方案是固定selenium==4.1.0和pytest==7.1.2版本。

2.2 目录结构设计规范

采用分层架构保证可扩展性：

code复制framework/
├── core/              # 框架核心组件
│   ├── assertion.py   # 自定义断言
│   └── logger.py      # 日志模块
├── pages/             # UI页面对象
│   └── login_page.py  
├── api/               # 接口测试层
│   ├── client.py      
│   └── schemas/       # 接口契约定义
├── tests/             # 测试用例
│   ├── ui/
│   └── api/
└── conftest.py        # Pytest全局fixture

3. UI自动化实现关键细节

3.1 智能元素定位策略

采用混合定位方式提升稳定性：

python复制# login_page.py
from selenium.webdriver.common.by import By

class LoginPage:
    # CSS选择器为主，XPath备用
    username = (By.CSS_SELECTOR, '#username')
    password = (By.XPATH, '//input[@type="password"]')
    
    def __init__(self, driver):
        self.driver = driver
        self.wait = WebDriverWait(driver, 10)
        
    def input_credentials(self, user, pwd):
        self.wait.until(EC.presence_of_element_located(self.username)).send_keys(user)
        # 密码字段显式等待
        self.wait.until(EC.visibility_of_element_located(self.password)).send_keys(pwd)

避坑经验：

绝对避免使用包含索引的XPath如//div[3]/span[2]
对动态元素增加二次等待策略
使用WebDriverWait替代time.sleep

3.2 浏览器多环境适配方案

通过conftest.py实现多浏览器支持：

python复制# conftest.py
import pytest
from selenium import webdriver
from webdriver_manager.chrome import ChromeDriverManager
from webdriver_manager.firefox import GeckoDriverManager

@pytest.fixture(scope='session')
def browser(request):
    browser_type = request.config.getoption("--browser")
    
    if browser_type == "chrome":
        options = webdriver.ChromeOptions()
        options.add_argument('--headless')  # 无头模式
        driver = webdriver.Chrome(ChromeDriverManager().install(), options=options)
    elif browser_type == "firefox":
        driver = webdriver.Firefox(executable_path=GeckoDriverManager().install())
    
    yield driver
    driver.quit()

执行时通过命令行参数指定浏览器：

bash复制pytest tests/ui/ --browser=chrome

4. 接口自动化高级实践

4.1 契约测试实现方案

使用Pydantic进行接口响应验证：

python复制# api/schemas/user.py
from pydantic import BaseModel

class UserSchema(BaseModel):
    id: int
    name: str
    email: str = None
    
    class Config:
        extra = 'forbid'  # 禁止多余字段

# tests/api/test_user.py
def test_get_user(api_client):
    resp = api_client.get('/users/1')
    assert resp.status_code == 200
    UserSchema(**resp.json())  # 自动验证数据结构

4.2 认证机制处理技巧

处理OAuth2.0认证的通用方案：

python复制# api/client.py
class APIClient:
    def __init__(self):
        self.session = requests.Session()
        self.token = None
        
    def auth(self):
        payload = {
            "client_id": os.getenv('CLIENT_ID'),
            "client_secret": os.getenv('CLIENT_SECRET'),
            "grant_type": "client_credentials"
        }
        resp = self.session.post('/oauth/token', data=payload)
        self.token = resp.json()['access_token']
        
    def request(self, method, url, **kwargs):
        if not self.token:
            self.auth()
            
        headers = kwargs.get('headers', {})
        headers.update({'Authorization': f'Bearer {self.token}'})
        kwargs['headers'] = headers
        
        return self.session.request(method, url, **kwargs)

5. 持续集成与质量门禁

5.1 Jenkins流水线配置

Jenkinsfile关键配置片段：

groovy复制pipeline {
    agent any
    stages {
        stage('Test') {
            parallel {
                stage('UI Test') {
                    steps {
                        sh 'pytest tests/ui/ --alluredir=allure-results'
                    }
                }
                stage('API Test') {
                    steps {
                        sh 'pytest tests/api/ --alluredir=allure-results'
                    }
                }
            }
            post {
                always {
                    allure includeProperties: false, 
                          jdk: '', 
                          results: [[path: 'allure-results']]
                }
            }
        }
    }
}

5.2 测试质量阈值控制

在pytest.ini中设置最低通过标准：

ini复制[pytest]
min_passed = 80
xfail_strict = true
addopts = --cov=./ --cov-report=xml

通过pytest-check插件实现软断言：

python复制from pytest_check import check

def test_complex_scenario():
    with check:
        assert response.status == 200
    with check:
        assert validate_schema(response.json())
    # 即使单个断言失败也会继续执行

6. 典型问题排查手册

现象	可能原因	解决方案
元素定位失败但页面存在	1. 框架上下文切换问题 2. 动态ID变化	1. 显式等待+重试机制 2. 改用相对定位方式
接口返回403错误	1. Token过期 2. 权限变更	1. 实现自动刷新Token逻辑 2. 在fixture中重置认证
Allure报告无数据	1. 结果目录路径错误 2. 文件权限问题	1. 检查--alluredir参数 2. 添加chmod 755操作

在框架实际落地过程中，发现最大的挑战不是技术实现，而是测试用例的可持续维护。建议建立以下规范：

UI测试采用"三明治"定位策略（页面对象+业务层+用例层）
接口测试遵循"契约先行"原则
所有测试数据外部化处理
关键操作添加视频录制功能（可通过pytest-selenium-video插件实现）

这套框架在某电商平台落地后，UI测试执行时间从3小时缩短到25分钟，接口测试覆盖率从60%提升到92%。核心经验是：不要追求100%的自动化率，而应该聚焦于核心链路的稳定验证。