如何模拟导入实例的方法

Question

我正在为一些 python 模块编写单元测试。但是，我无法找到模拟导入实例的方法的方法。下面是我要测试的python模块。

--bar.py--

class A():
    @classmethod
    def method_1(self):
        ...    
a = A()

--foo.py--

from bar import a
class B():
    @classmethod
    def b(cls):
        if a.method_1():
            return True
        else:
            return False

--test_foo.py--

from foo import B

class TestB(TestCase):
    @patch('foo.a.method_1')
    def test_b(self, mock_method_1):
          mock_method_1.return_value = True
          test_b = B.b()
          ...

这不起作用。我的测试用例仍然调用原始的method_1 而不是我嘲笑的那个。

Answer 1

请改用patch.object decorator。它修补对象的属性而不是修补全局方法。

如果这不起作用，请尝试修补 bar.a 而不是 foo.a，但我认为这不是你的问题。

更新

问题改为类方法，所以我认为这会起作用：

--test_foo.py--

from foo import B

class TestB(TestCase):
    @patch('bar.A.method_1')
    def test_b(self, mock_method_1):
          mock_method_1.return_value = True
          test_b = B.b()
          ...