Python Mock side_effect模拟方法行为报错，求正确实现方式

哥们，你的用法确实存在几个问题，才触发了这个报错，我帮你一步步拆解并修正：

首先梳理你代码里的核心错误 #

• @patch的参数写法完全错误：你当前写的@patch("MyClass", "foo", side_effect='_side_effect_foo')不符合patch的参数规则。patch的第一个参数必须是要Mock对象的完整路径（比如如果MyClass在my_module.py里，就得写"my_module.MyClass.foo"）；其次不能把目标方法名foo作为第二个参数传递，且side_effect需要传入函数对象，不是字符串形式的方法名。
• Side effect函数参数不匹配：原foo是实例方法，第一个参数是self，你的_side_effect_foo没有对应这个参数，会导致调用时参数不匹配。
• 原类方法逻辑错误：MyClass的bar方法里，my_var = foo(1)既没有用self.调用foo，也没有赋值给类/实例属性，这会导致Mock后的方法根本不会被触发，也无法修改my_var。
• 测试实例初始化位置不当：直接在TestClass里定义myClass = MyClass()会让所有测试用例共享同一个实例，容易产生测试污染。

修正后的完整代码示例 #

首先修正原业务类

class MyClass:
    my_var = None
    def foo(self, var1):
        return "original_value"  # 给默认返回值方便测试
    def bar(self):
        self.my_var = self.foo(1)  # 修正：用self调用foo，赋值给实例属性

然后修正测试类

这里提供两种可行的写法：

写法一：在测试方法内绑定side effect

import unittest
from unittest.mock import patch

class TestClass(unittest.TestCase):
    def setUp(self):
        self.my_class = MyClass()  # 每个测试用例初始化新实例，避免污染

    def _side_effect_foo(self, var1):
        # 匹配原foo的参数逻辑，这里可以写你的condition判断
        if var1 == 1:
            return "new_mocked_value"
        return "original_value"

    @patch("__main__.MyClass.foo")  # 路径根据你的实际模块调整，比如"my_module.MyClass.foo"
    def test_foo(self, mock_foo):
        # 将side effect绑定到Mock对象
        mock_foo.side_effect = self._side_effect_foo
        # 执行测试逻辑
        self.my_class.bar()
        # 验证结果是否符合预期
        self.assertEqual(self.my_class.my_var, "new_mocked_value")
        # 验证foo方法是否被正确调用
        mock_foo.assert_called_once_with(1)

写法二：直接在patch装饰器中传入静态方法作为side effect

import unittest
from unittest.mock import patch

class TestClass(unittest.TestCase):
    def setUp(self):
        self.my_class = MyClass()

    @staticmethod
    def _side_effect_foo(var1):
        if var1 == 1:
            return "new_mocked_value"
        return "original_value"

    @patch("__main__.MyClass.foo", side_effect=_side_effect_foo)
    def test_foo(self, mock_foo):
        self.my_class.bar()
        self.assertEqual(self.my_class.my_var, "new_mocked_value")
        mock_foo.assert_called_once_with(1)

关键注意点 #

• patch的第一个参数必须是目标对象的完整导入路径，Python需要通过这个路径定位到要替换的方法。
• side_effect必须传递可调用的函数/方法对象，不能是字符串。
• 确保原业务类中方法的调用是正确的（比如加self.），否则Mock不会生效。
• 测试实例尽量在setUp方法中初始化，保证每个测试用例的独立性。

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