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

Question

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

# 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'

我相信问题是在哪里修补,但我已尝试导入两个变种没有运气.我甚至尝试将import语句移动到test_mock_fn()中,以便在aC进入作用域之前,模拟的a.fn()将"存在" - nope仍然失败.

任何见解将不胜感激!

Answer 1

这里实际发生的是，当你真正导入你的模块时，fn()已经执行了。因此，在您已经评估了存储在您的类属性中的方法之后，模拟就会出现。

因此，当您尝试模拟该方法时，对于您正在尝试进行的测试来说已经太晚了。

如果您只是在方法中添加一个打印语句，您甚至可以看到这种情况发生：

def fn():
    print("I have run")
    return "asdf"

在您的测试模块中，当您导入a并简单地运行甚至不运行您的测试时，您将看到I have run将出现在您的控制台输出中，而无需从您的a模块中明确运行任何内容。

因此，您可以在这里采取两种方法。您可以使用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，但是通过这种模拟，您现在将 '1234'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)