使用 pytest.fixture 实现测试前后自动登录与登出的正确方式

3次阅读

本文详解如何通过 pytest 的 fixture 机制，在每个测试用例执行前自动调用登录逻辑、执行后自动调用登出逻辑，无需将测试函数拆分为独立 test_* 方法，也无需修改 conftest.py 或破坏测试结构。

本文详解如何通过 pytest 的 fixture 机制，在每个测试用例执行前自动调用登录逻辑、执行后自动调用登出逻辑，无需将测试函数拆分为独立 test_* 方法，也无需修改 conftest.py 或破坏测试结构。

在 pytest 中，@pytest.fixture 的核心价值并非“复用测试函数”，而是 声明式地管理测试依赖与生命周期 。用户提出的诉求——“让 test_login 在每个测试前运行、test_logout 在每个测试后运行”——本质上不是要复用两个测试用例，而是需要一个具备 前置 setup + 后置 teardown 行为 的可复用上下文。此时，正确做法是定义一个 yield fixture，而非尝试把测试函数当作 fixture 使用。

以下是一个完整、可直接运行的实践示例：

import pytest  # ✅ 正确：定义一个带 setup/teardown 行为的 fixture @pytest.fixture def logged_in_session(username, password):     # ? 前置动作：模拟登录（可替换为真实 Appium 登录逻辑）print(f"[SETUP] Logging in as {username}/{password}")     # 这里可调用 driver.find_element 等实际登录操作     assert username and password  # 简单校验，实际中应验证登录成功状态      yield  # 测试用例在此处执行      # ? 后置动作：模拟登出     print("[TEARDOWN] Logging out")     # 这里可调用 driver.find_element(……).click() 执行登出操作     # 注意：若登出失败，fixture 会报错，但不会影响其他测试的执行（pytest 默认隔离）# ✅ 测试类继承自基类（如 base_test），保持原有结构 class Test_Login:     @pytest.mark.parametrize("username,password", [         ("admin", "123456"),         ("testuser", "password123")     ])     def test_dashboard_access(self, username, password, logged_in_session):         # 此处已处于「已登录」状态，可直接测试业务逻辑         print(f"✅ Running test with {username} — dashboard access verified")         assert True  # 替换为真实断言，如 assert "Dashboard" in driver.title      def test_profile_update(self, logged_in_session):         # 同样自动获得登录上下文；注意：此用例不依赖 parametrize，故无需传 username/password         # 若需参数，可在 fixture 内部或通过其他 fixture 注入         print("✅ Running profile update test")         assert True

? 关键说明与最佳实践：

不要把测试函数当 fixture 用：test_login() 和 test_logout() 是测试行为的入口，而 fixture 是资源 / 状态的提供者。强行将其作为 fixture 会导致 pytest 报错（Function ‘test_login’ has no signature）或语义混乱。
yield 是核心：yield 前为 setup，yield 后为 teardown，pytest 自动保证其执行顺序与作用域（如 scope=”function” 即每个测试函数独享一次完整生命周期）。
参数化兼容性：fixture 可安全与 @pytest.mark.parametrize 共存。如上例所示，username 和 password 可由 parametrize 提供，并被 fixture 直接消费。
作用域控制：默认 scope=”function” 已满足“每个测试前后执行”的需求；若需跨多个测试共享登录态（不推荐，因状态耦合），可设为 scope=”class” 或 “module”，但务必确保登出逻辑幂等且异常安全。
错误处理建议：在 teardown 部分（yield 后）添加 try/except 可避免登出失败导致整个测试套件中断，例如：
```
yield try:     driver.find_element(*LOGOUT_BTN).click() except Exception as e:     print(f"[WARNING] Logout failed but ignored: {e}")
```