我正在尝试修补由我尝试测试的类实例化的类,但它不起作用。我已经阅读了各种文档,但仍然没有找到我做错了什么。以下是代码段:
在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()
我知道在这种情况下我可以很容易地注入依赖项,但这只是一个例子。此外,我们使用每个文件策略的单个类,文件名称类,因此奇怪的导入命名。
答案 0 :(得分:3)
正如诺伯特所提到的,修复是从
改变模拟线@mock.patch('module.ClassToPatch.ClassToPatch', autospec = False)
到
@mock.patch('module.ClassToTest.ClassToPatch', autospec = False)
patch()装饰器/上下文管理器可以轻松模拟被测模块中的类或对象。您指定的对象将在测试期间替换为mock(或其他对象),并在测试结束时恢复。
您正在测试ClassToTest
模块,而不是ClassToPatch
模块。