我想模拟一个用于初始化类级(非实例)属性的模块级函数.这是一个简化的例子:
# a.py
def fn():
return 'asdf'
class C:
cls_var = fn()
这是一个试图模拟a.fn()的单元测试:
# test_a.py
import unittest,mock
import a
class TestStuff(unittest.TestCase):
# we want to mock a.fn so that the class variable
# C.cls_var gets assigned the output of our mock
@mock.patch('a.fn',return_value='1234')
def test_mock_fn(self,mocked_fn):
print mocked_fn()," -- as expected,prints '1234'"
self.assertEqual('1234',a.C.cls_var) # fails! C.cls_var is 'asdf'
我相信这个问题是where to patch但是我在导入时尝试了两种变化而没有运气.我甚至尝试将import语句移动到test_mock_fn()中,以便在a.C进入作用域之前,模拟的a.fn()将“存在” – nope,仍然失败.
任何见解将不胜感激!
最佳答案
这里实际发生的是当你实际导入模块时,fn()已经执行了.因此,在您已经评估了存储在类属性中的方法之后,模拟就会出现.
因此,当您尝试模拟方法时,您尝试进行的测试为时已晚.
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)