Robot Framework中动态测试初始化与清理的实现方法咨询：按需执行数据库插入/删除操作

我刚经历过类似的转型阶段，太懂你这种复制粘贴改参数的痛苦了！咱们核心要解决的就是让每个测试用例能独立控制前置/后置的数据库操作，同时彻底消除冗余代码。下面给你几个落地的方案：

1. 用「测试用例级Setup/TearDown + 参数化关键字」替代套件级重复 #

这是最直接的解决方案，把MongoDB的增删逻辑封装成参数化关键字，然后让每个测试用例自己传递初始化数据：

第一步：封装通用数据库操作关键字 #

先写一个资源文件MongoDBHelpers.robot，把重复的MongoDB操作封装起来，重点是插入后记录文档ID，确保清理时精准删除：

*** Keywords ***
Create Test Document
    [Arguments]  ${document_data}  ${collection_name}=users
    # 插入文档并返回生成的_id
    ${insert_result}=  PyMongo Insert Document  ${collection_name}  ${document_data}
    Set Test Variable  ${TEST_DOC_ID}  ${insert_result.inserted_id}
    [Return]  ${TEST_DOC_ID}

Clean Up Test Document
    [Arguments]  ${collection_name}=users
    # 只删除当前测试用例插入的文档，避免误删其他数据
    IF  '${TEST_DOC_ID}' != ''
        PyMongo Delete Document  ${collection_name}  _id=${TEST_DOC_ID}
    END

第二步：在测试用例中直接传参 #

不用再复制整个套件文件了！只需要一个测试用例文件，每个用例定义自己的初始化数据：

*** Settings ***
Resource    MongoDBHelpers.robot
Test Teardown  Clean Up Test Document

*** Test Cases ***
记录存在时接口返回正常
    [Setup]  Create Test Document  {"user_id": "test_001", "status": "normal"}
    调用目标接口  /api/user/test_001
    校验响应状态码  200

记录存在数据缺陷时错误被处理
    [Setup]  Create Test Document  {"user_id": "", "status": "invalid"}
    调用目标接口  /api/user/
    校验响应状态码  400

记录不存在时错误被处理
    [Setup]  # 不需要插入数据，空Setup即可
    调用目标接口  /api/user/nonexist_001
    校验响应状态码  404
    [Teardown]  # 无数据需要清理，跳过

2. 用DataDriver库解决批量场景的变量作用域问题 #

你之前说DataDriver变量作用域有问题，大概率是没把数据驱动的变量正确传递给Setup关键字。试试这种写法：

第一步：准备测试场景数据源（比如CSV） #

新建test_scenarios.csv，每一行对应一个测试场景：

document_data,endpoint,expected_status
{"user_id": "test_001", "status": "normal"},/api/user/test_001,200
{"user_id": "", "status": "invalid"},/api/user/,400
"{}",/api/user/nonexist_001,404

第二步：在测试用例文件中关联数据源和Setup #

*** Settings ***
Resource    MongoDBHelpers.robot
Library     DataDriver    test_scenarios.csv
Test Setup  Create Test Document  ${document_data}  # 直接引用DataDriver的变量
Test Teardown  Clean Up Test Document

*** Test Cases ***
Template    执行接口测试场景

*** Keywords ***
执行接口测试场景
    [Arguments]  ${document_data}  ${endpoint}  ${expected_status}
    调用目标接口  ${endpoint}
    校验响应状态码  ${expected_status}

这里的关键是：DataDriver的变量在测试用例执行上下文里是全局可访问的，所以Setup关键字能直接拿到${document_data}，完美解决作用域问题。

3. 进阶：用标签动态匹配初始化逻辑 #

如果有些场景的初始化逻辑复杂，不想每个用例都写Setup参数，可以给用例打标签，在Suite Setup里根据标签动态执行对应操作：

*** Settings ***
Resource    MongoDBHelpers.robot
Suite Setup  根据标签执行动态初始化
Test Teardown  Clean Up Test Document

*** Test Cases ***
记录存在时接口返回正常
    [Tags]  RecordExists
    调用目标接口  /api/user/test_001
    校验响应状态码  200

记录存在数据缺陷时错误被处理
    [Tags]  RecordDefect
    调用目标接口  /api/user/
    校验响应状态码  400

*** Keywords ***
根据标签执行动态初始化
    ${current_tags}=  Get Tags
    IF  'RecordExists' in ${current_tags}
        Create Test Document  {"user_id": "test_001", "status": "normal"}
    ELSIF  'RecordDefect' in ${current_tags}
        Create Test Document  {"user_id": "", "status": "invalid"}
    END

核心原则总结 #

不管用哪种方案，记住这两点就能彻底解决重复问题：

• 封装通用操作：把MongoDB的增删改查全部封装成关键字，不要在测试用例里写原生操作
• 参数化测试数据：让测试用例只负责传递场景-specific的数据，逻辑全部放在关键字里

内容的提问来源于stack exchange，提问作者rook218