这里实际发生的是当你实际导入模块时,fn()已经执行了.因此,在您已经评估了存储在类属性中的方法之后,模拟就会出现.
因此,当您尝试模拟方法时,您尝试进行的测试为时已晚.
如果只是在方法中添加print语句,甚至可以看到这种情况:
def fn():
print("I have run")
return "asdf"
在您的测试模块中,当您导入并且只是在没有运行测试的情况下运行时,您将看到我已经运行将出现在您的控制台输出中,而无需从模块中明确地运行任何内容.
所以,你可以采取两种方法.您可以使用PropertyMock将类属性模拟为您希望存储的内容,如下所示:
@mock.patch('a.C.cls_var', new_callable=PropertyMock)
def test_mock_fn(self, mocked_p):
mocked_p.return_value = '1234'
self.assertEqual('1234', a.C.cls_var)
现在,你必须要知道,通过这样做,你仍然在运行fn,但是通过这种嘲弄,你现在在cls_var中使用你设置的PropertyMock持有’1234′.
以下建议(可能不太理想,因为它需要进行设计更改)将需要修改您使用类属性的原因.因为如果您实际上将该类属性设置为实例属性,那么当您创建C实例时,您的方法将执行,此时它将使用您的模拟.
所以,你的班级看起来像:
class C:
def __init__(self):
self.var = fn()
你的测试看起来像:
@mock.patch('a.fn', return_value='1234')
def test_mock_fn(self, mocked_p):
self.assertEqual('1234', a.C().var)