Python中如何断言复杂方法已调用另一个方法？遗留代码测试场景

我正在为一段测试友好性较差的现有代码添加测试，需验证复杂方法是否调用了另一个方法。示例代码如下：

class SomeView(...):
    def verify_permission(self, ...):
        # some logic to verify permission
        ...

    def get(self, ...):
        # some codes here I am not interested in this test case
        ...
        
        if some condition:
            self.verify_permission(...)
        
        # some other codes here I am not interested in this test case
        ...        

我需要编写测试用例，验证当满足特定条件时self.verify_permission方法会被调用。请问是否需要通过Mock一步步到该方法执行点？或是需要重构get方法以提升代码的测试友好性？

两种可行方案 #

1. 用Mock快速实现测试（无需重构现有代码） #

不需要一步步Mock到执行点，直接Mock目标方法就能完成验证。以Python的unittest.mock为例，测试代码可以这么写：

from unittest.mock import MagicMock
import unittest

class TestSomeView(unittest.TestCase):
    def test_get_calls_verify_permission_when_condition_met(self):
        # 创建View实例
        view = SomeView()
        # 替换原方法为Mock对象
        view.verify_permission = MagicMock()
        
        # 构造满足触发条件的参数/环境，让some condition为True
        # 比如传入特定请求参数、修改实例属性等
        ...
        
        # 调用get方法
        view.get(...)
        
        # 断言verify_permission按预期被调用
        view.verify_permission.assert_called_once_with(...)

如果get方法的前置逻辑依赖外部资源（比如数据库、请求对象），只需Mock这些依赖让代码能走到条件判断分支即可，不用完全模拟前置步骤的真实逻辑。

2. 重构代码提升测试友好性（长期更优） #

如果这段代码需要频繁维护或扩展，重构拆分逻辑会让后续测试和维护更简单。比如把条件判断逻辑抽成独立方法：

class SomeView(...):
    def verify_permission(self, ...):
        # some logic to verify permission
        ...

    def _should_check_permission(self, ...):
        # 把原有的some condition逻辑单独提取
        return some condition

    def get(self, ...):
        # some codes here I am not interested in this test case
        ...
        
        if self._should_check_permission(...):
            self.verify_permission(...)
        
        # some other codes here I am not interested in this test case
        ...        

这么做的好处：

• 可以单独测试_should_check_permission的逻辑是否正确
• 测试get方法时，只需Mock_should_check_permission就能轻松控制是否进入调用权限验证的分支
• 代码职责更清晰，降低后续维护的心智负担

总结 #

• 若只是临时加测试、不想改动现有代码，直接Mock目标方法并构造满足条件的环境即可快速完成验证
• 若代码属于长期维护的核心模块，重构拆分逻辑是更值得投入的方案，能从根本上提升测试友好性

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