如何模拟由@patch 装饰器创建的对象的方法？

Question

我正在尝试使用 @patch 装饰器和模拟对象来测试我的代码消耗一些库的行为。不幸的是，我看不到任何方法可以只使用装饰器来做我想做的事情，而且似乎我应该能够做到。我意识到通常我应该更关心方法的返回值，而不是它对库的特定调用序列，但是编写一个测试来比较一个图像和另一个图像是非常困难的。

我想也许我可以创建我的模拟对象的实例，然后使用@patch.object(...)，但这不起作用，显然@patch('MockObject.method') 只会模拟类方法。我可以使用with patch.object(someMock.return_value, 'method') as mockName 使代码正常工作，但这会使我的测试变得混乱。

from some.module import SomeClass


class MockObject:
    def method(self):
        pass


class TestClass:
    @patch('SomeClass.open', return_value=MockObject())
    @patch('??????.method')  # I can't find any value to put here that works
    def test_as_desired(self, mockMethod, mockOpen):
        instance = SomeClass.open()
        instance.method()

        mockOpen.assert_called_once()
        mockMethod.assert_called_once()

    @patch('SomeClass.open', return_value=MockObject())
    def test_messy(self, mockOpen):
        with patch.object(mockOpen.return_value, 'method') as mockMethod:
            instance = SomeClass.open()
            instance.method()

            mockOpen.assert_called_once()
            mockMethod.assert_called_once()

Answer 1

我认为你把事情复杂化了：

import unittest.mock

@unittest.mock.patch.object(SomeClass, open=mock.Mock())
def test_as_desired(self, mock_open):
    instance = SomeClass.open()  # SomeClass.open is a mock, so its return value is too
    instance.method()

    mock_ppen.assert_called_once()
    instance.method.assert_called_once()