5

我想模拟一个用于初始化类级(而不是实例)属性的模块级函数。这是一个简化的示例:

# 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() 中,以便模拟的 a.fn() 在 aC 进入范围之前“存在” - 不,仍然失败。

任何见解将不胜感激!

4

2 回答 2

5

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

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

如果您只是在方法中添加一个 print 语句,您甚至可以看到这种情况发生:

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_varPropertyMock您设置的范围内。

以下建议(可能不太理想,因为它需要更改设计)将需要修改您使用类属性的原因。因为如果您实际上将该类属性设置为实例属性,那么当您创建实例时,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)
于 2016-02-21T02:27:01.863 回答
0

甚至@idjaw 的回答也是正确的,并正确解释了发生的事情我认为最简洁、简单和直接的方法是直接a.C.cls_var通过值而不是模拟来修补属性。

@mock.patch('a.C.cls_var', '1234')

这足以满足您的所有需求:PropertyMock仅当您必须替换应该像属性一样的属性时,使用才很有用。

有关您的方法为何不起作用的所有详细信息,请查看@idjaw 的答案。

于 2016-02-22T13:08:54.217 回答