Python单元测试：如何Mock被测试函数导入的数据库调用函数？

核心问题：Patch路径错误 #

你当前patch的"db.calls.get_from_db"是函数定义的原始路径，但unittest.mock的patch规则是：必须patch代码实际运行时使用的那个引用，也就是被调用函数所在模块里的导入副本。

在你的代码逻辑里：

• competition.prizes.common模块的get_prizes函数，通过from .db import get_from_db导入了数据库调用函数，它实际使用的是competition.prizes.common.get_from_db这个本地引用。
• 你patch原始定义的路径根本碰不到这个本地引用，所以mock自然不会生效。

修复步骤 #

1. 修正Patch路径：把patch目标改成"competition.prizes.common.get_from_db"
2. 注意装饰器参数顺序：unittest.mock的装饰器是从下往上应用的，函数参数的顺序对应最后一个装饰器的mock在前，第一个在后。

修正后的测试代码 #

from unittest.mock import patch
import lambda_function

class TestLambdaFunction:
    # 装饰器顺序：先patch get_entries，再patch get_from_db
    @patch("competition.prizes.common.get_from_db", return_value=(200, 100))
    @patch.object(lambda_function, "get_entries")
    def test_level_one_not_reached(self, mock_get_entries, mock_get_from_db):
        # 调用lambda_handler而非lambda_function（原代码此处有误）
        result = lambda_function.lambda_handler()
        # 可添加断言验证mock调用情况
        mock_get_entries.assert_called_once()
        mock_get_from_db.assert_called_once()

额外代码错误修正 #

你的示例代码里还有几处笔误会导致运行异常：

• lambda_function里的导入：from competiton.prizes import get_prize → 拼写错误，应改为from competition.prizes import get_prizes（补全字母i，且函数名与调用的get_prizes()对应）
• lambda_function里的get_prizes函数：return get_prizes()会触发无限递归，应改为调用导入的函数，比如return get_prize()（若导入的是get_prize），或直接删除这个冗余函数
• competition.prizes.common里的get_prizes函数定义缺少冒号：def get_prizes() → def get_prizes():

关键规则总结 #

记住：要mock的是「被测试代码实际用到的那个模块里的函数引用」，而不是函数最初定义的模块路径。

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