如何用pytest模拟字典行为来测试账户校验逻辑改写？

本文共计672个文字，预计阅读时间需要3分钟。

原文：

在单元测试中，若函数直接依赖全局或模块级字典（如 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

随后，编写清晰、正交的 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 等兼容结构，提升函数通用性。

通过将依赖显式化，测试代码变得简洁、可读、可组合——每个测试用例只关注一个行为维度，真正实现“快速反馈、稳定可靠”的单元测试目标。