无法使用unittest修补测试类实例化的类

时间:2016-02-23 20:36:17

标签: python python-3.x python-unittest

我正在尝试修补由我尝试测试的类实例化的类,但它不起作用。我已经阅读了各种文档,但仍然没有找到我做错了什么。以下是代码段:

tests/Test.py

from module.ClassToTest import ClassToTest

class Test(object):

    @mock.patch('module.ClassToPatch.ClassToPatch', autospec = False)
    def setUp(self, my_class_mock):
        self.instance = my_class_mock.return_value
        self.instance.my_method.return_value = "def"
        self.class_to_test = ClassToTest()

    def test(self):
        val = self.class_to_test.instance.my_method() #Returns 'abc' instead of 'def'
        self.assertEqual(val, 'def')

module/ClassToPatch.py

class ClassToPatch(object):
    def __init__(self):
        pass
    def my_method(self):
        return "abc"

module/ClassToTest.py

from module.ClassToPatch import ClassToPatch

class ClassToTest(object):
    def __init__:
        # Still instantiates the concrete class instead of the mock
        self.instance = ClassToPatch()

我知道在这种情况下我可以很容易地注入依赖项,但这只是一个例子。此外,我们使用每个文件策略的单个类,文件名称类,因此奇怪的导入命名。

1 个答案:

答案 0 :(得分:3)

正如诺伯特所提到的,修复是从

改变模拟线
@mock.patch('module.ClassToPatch.ClassToPatch', autospec = False)

@mock.patch('module.ClassToTest.ClassToPatch', autospec = False)

According to the docs

  

patch()装饰器/上下文管理器可以轻松模拟被测模块中的类或对象。您指定的对象将在测试期间替换为mock(或其他对象),并在测试结束时恢复。

您正在测试ClassToTest模块,而不是ClassToPatch模块。