本文介绍如何通过重构函数签名并传入可控制的字典参数，在 pytest 中高效、可靠地测试依赖外部字典（如 accounts_dict）的业务逻辑，避免全局状态干扰，提升测试隔离性与可维护性。
本文介绍如何通过重构函数签名并传入可控制的字典参数，在 pytest 中高效、可靠地测试依赖外部字典（如 `accounts_dict`）的业务逻辑，避免全局状态干扰，提升测试隔离性与可维护性。
在单元测试中，若函数直接依赖全局或模块级字典（如 accounts_dict），会导致测试难以控制、结果不可预测——例如原函数 validate_account(number) 在调用时因 accounts_dict 为空而始终抛出异常，无法覆盖“有效账号”分支。根本解法不是用 Mock “打补丁”，而是重构函数设计，将字典作为显式参数注入。这不仅符合依赖倒置原则，也使测试完全可控。
✅ 推荐方案：参数化字典依赖（推荐）
首先，修改被测函数，显式接收字典参数：
def validate_account(number, accounts_dict):
try:
number = int(number)
except ValueError:
raise ValueError("Please enter a valid account number")
if number not in accounts_dict: # 注意：无需重复 int(number)，已转换过
raise Exception("Pleaseeee enter a valid account number.")
return number? 关键改进：
- 移除对全局 accounts_dict 的隐式依赖；
- 修复重复类型转换（原代码 int(number) 出现两次）；
- 异常类型更精准（ValueError 用于解析失败，自定义 Exception 用于业务校验）。
随后，编写清晰、正交的 pytest 测试用例：
import pytest
def test_validate_account_valid():
# 当账号存在时，应成功返回整数
accounts = {1: "active", 42: "suspended"}
assert validate_account("1", accounts) == 1
def test_validate_account_invalid():
# 当账号不存在时，应抛出 Exception
accounts = {99: "active"}
with pytest.raises(Exception, match="Pleaseeee"):
validate_account("1", accounts)
def test_validate_account_non_integer():
# 输入非数字字符串，应抛 ValueError
with pytest.raises(ValueError, match="Please enter a valid account number"):
validate_account("abc", {1: True})⚠️ 注意事项与进阶建议
-
避免滥用 Mock 模拟字典：对 dict 类型使用 Mock(return_value=...) 无效，因为 in 操作符调用的是 __contains__，而非 __call__。若必须 Patch 全局字典（不推荐），应使用 unittest.mock.patch.dict：
from unittest.mock import patch def test_with_patch(): with patch.dict('module_name.accounts_dict', {1: True}): assert validate_account("1") == 1 # 前提是函数仍引用全局变量但此方式耦合模块路径、破坏测试独立性，仅作临时兼容方案。
生产环境字典初始化建议：可在应用启动时（如 main() 或 app.init()）一次性加载 accounts_dict，测试中则完全绕过该流程，专注验证逻辑本身。
增强健壮性（可选）：可将 accounts_dict 类型约束为 Mapping（来自 collections.abc），支持传入 defaultdict、ChainMap 等兼容结构，提升函数通用性。
通过将依赖显式化，测试代码变得简洁、可读、可组合——每个测试用例只关注一个行为维度，真正实现“快速反馈、稳定可靠”的单元测试目标。
