如何Mock类属性拦截内部方法调用，验证self.a.set_name参数正确性？

我需要确保类方法中的self.a.set_name被传入正确的参数值，由于存在15个类似方法，希望通过patch整个对象而非逐个方法来实现。但运行测试时出现断言错误：

AssertionError: Expected 'set_name' to be called once. Called 0 times.

代码示例 #

src/resources.py #

from package_a import ClassA


class MyClassB:
    def __init__(self):
        self.a = None

    def get_a(self, name: str):
        self.a = ClassA()
        self.a.set_name(name=name)

test/test_resources.py #

from src.resources import MyClassB
from unittest import mock

class TestClassB:
    def test_get_a(self):
        b = MyClassB()
        # with mock.patch("src.resources.package_a.ClassA") as mocked_a: # this also doesn't work
        with mock.patch.object(b, "a") as mocked_a:
            b.get_a(name="test")
            mocked_a.set_name.assert_called_once_with(name="test")

问题根源 #

1. patch对象被覆盖：你尝试patchb.a，但get_a方法里会重新执行self.a = ClassA()，直接覆盖了预patch的mock对象，导致原mocked_a从未被调用。
2. patch路径错误：之前尝试的mock.patch("src.resources.package_a.ClassA")路径不正确——因为src/resources.py中是直接from package_a import ClassA，所以在该模块内ClassA是本地引用，需要patch这个本地名称，而非原包路径。

正确测试代码 #

修改测试逻辑，patchsrc.resources.ClassA，并通过mock实例验证调用：

from src.resources import MyClassB
from unittest import mock

class TestClassB:
    def test_get_a(self):
        with mock.patch("src.resources.ClassA") as mocked_class_a:
            # 获取ClassA构造返回的mock实例
            mocked_a_instance = mocked_class_a.return_value
            b = MyClassB()
            b.get_a(name="test")
            # 验证set_name的调用参数
            mocked_a_instance.set_name.assert_called_once_with(name="test")

多方法适配 #

这种方式无需逐个patch每个方法内的self.a，只要所有15个方法都是通过ClassA()创建实例，这个patch就能覆盖所有场景，统一验证set_name的参数正确性。

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