Python：如何模拟类属性初始化函数

我想模拟一个用于初始化类级(非实例)属性的模块级函数.这是一个简化的例子：

# 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()已经执行了.因此,在您已经评估了存储在类属性中的方法之后,模拟就会出现.

因此,当您尝试模拟方法时,您尝试进行的测试为时已晚.

如果只是在方法中添加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)