python模拟：替换类方法

Question

考虑以下代码（这不是一个很好的设计，但这就是重点）：

class A(object):
    def __init__(self,filepath):
        self._access_file_system(filepath)

    def get(self):
        return self._result_dict


class B(object):
    def __init__(self,filepath1,filepath2):
        self._filepath1 = filepath1
        self._filepath2 = filepath2

    def foo(self):
        a1 = A(self._filepath1).get()
        a2 = A(self._filepath2).get()
        return a1['result']==a2['result']

现在，如果我想测试B.foo() ，我需要模拟A （因为它访问了构造函数中的文件系统）。

要编写一个测试，将确保B.foo()返回False的情况下， a1.get()和a2.get()提供不同的价值观，我还需要模拟B.get()

因此，测试功能可能应如下所示：

import mock
mock_get = mock.MagicMock(side_effect=[{'result': 0}, {'result': 1}])

@mock.patch('__main__.A')
def test_foo(MockA):

    b = B('/file1','/file2')
    res = b.foo()
    assert res
    MockA.assert_any_call('/file1')
    MockA.assert_any_call('/file2')

    #Doesn't work -
    #the assignment doesn't propagate into the objects instantiated inside foo()
    #A.get = mock_get

    #The assigned method propagates into the class definition,
    #so it works - BUT WHY?!
    a = A(None)
    a.get = mock_get

    b = B('/file1', '/file2')
    res = b.foo()
    assert not res

现在，奇怪的一点-从代码内的注释中可以看出，如果我们将mock_get分配给该类，它将不会传播，但是如果我们创建一个实例并将其分配给它，则它会传播给该类的其他实例。

我想这种行为与mock内部机制有关，因此对我来说很重要的一点是要理解它，并充分利用此库的所有丰富功能。

那么，有人知道吗？

Answer 1

在第一种情况下，我看不到您正在修补get方法的任何地方。 在调用B之前，应将模拟值分配给A的get方法。 例如，为什么以下测试失败？：

import mock
mock_get = mock.MagicMock(side_effect=[{'result': 0}, {'result': 1}])

@mock.patch('__main__.A')
def test_foo(MockA):

    MockA.get = mock_get

    b = B('/file1','/file2')
    res = b.foo()
    assert not res
    MockA.assert_any_call('/file1')
    MockA.assert_any_call('/file2')

先前行为的原因是我们忘记修补对象（A）的返回值，在本例中为MockA ，而不是对象本身（ MockA ）。 A对象是实例化A类的结果，您应该通过A类的return_value访问方法。 在您的示例中，将类似于以下内容：

import mock
mock_get = mock.MagicMock(side_effect=[{'result': 0}, {'result': 1}])

@mock.patch('__main__.A')
def test_foo(MockA):

    MockA.return_value.get = mock_get
    b = B('/file1','/file2')
    res = b.foo()

    assert res
    MockA.assert_any_call('/file1')
    MockA.assert_any_call('/file2')

您可以查看以下一些帖子，以获取有关常见python单元测试陷阱的更多信息：

• https://esaezgil.com/post/unittesting_pitfalls/ （由我撰写）
• http://alexmarandon.com/articles/python_mock_gotchas/
• 当然还有文档： https : //docs.python.org/3/library/unittest.mock.html