1个脚本跑1000组数据！美团测试总监亲授数据驱动实战

# ❌ 传统硬编码脚本（需求变更必崩溃）  
def test_login():  
    driver.find_element(By.ID, "username").send_keys("admin")  
    driver.find_element(By.ID, "password").send_keys("123456")  

# ✅ 数据驱动脚本（参数化应对变化）  
import csv  
@pytest.mark.parametrize("user,pwd", read_csv("login_data.csv"))  
def test_login(user, pwd):  
    driver.find_element(By.ID, "username").send_keys(user)  
    driver.find_element(By.ID, "password").send_keys(pwd)  

💡 企业收益：

# 支持多源数据的工厂模式实现  
def load_data(source):  
    if source.endswith(".csv"):  
        with open(source, encoding="utf-8-sig") as f:  # 解决BOM头问题  
            return list(csv.DictReader(f))  
    elif source.endswith(".json"):  
        return json.load(open(source))  
    elif source.startswith("mysql:"):  
        return query_db(source.split(":")[1])  # 企业级数据库连接器  

# 获取今日头条的测试数据  
test_data = load_data("testcases/search_toutiao.json")  

# ✅ 动态生成测试ID（报告更直观）  
@pytest.mark.parametrize("user", test_data, ids=lambda d: f"{d['role']}权限测试")  
def test_permission(user):  
    login(user["name"], user["pwd"])  
    assert check_access(user["expected_features"])  

# 报告显示：  
# ✓ 管理员权限测试  
# ✓ 运营人员权限测试  

[pytest]  
addopts = --tb=short --durations=0  
; 失败时重跑2次  
reruns = 2  
; 自动保存失败用例数据  
failed_data_file = reports/failed_cases.json  

// failed_cases.json  
{  
  "test_permission[运营人员权限测试]": {  
    "user": {"name": "ops01", "pwd": "*****", "role": "operator"},  
    "error": "AssertionError: 缺少'删除'权限"  
  }  
}  

from cryptography.fernet import Fernet  

# ✅ 数据加密存储  
key = Fernet.generate_key()  
cipher = Fernet(key)  
encrypted_pwd = cipher.encrypt(b"123456")  # 写入测试数据文件  

# ✅ 运行时动态解密  
@pytest.fixture(scope="function")  
def decrypt_data(request):  
    request.param["password"] = cipher.decrypt(request.param["enc_password"]).decode()  
    return request.param  

# 使用：  
@pytest.mark.parametrize("user", test_data, indirect=True)  
def test_payment(user):  
    pay(user["card_num"], user["password"])