在python 3.4中模拟内部对象的实例方法

Question

我的代码中有一个使用第三方库的类。 我想通过调用库类的实例方法来模拟获得的对象的实例方法。 我不清楚如何模拟此内部对象的实例方法。 以下是我的代码：

我的课：

from a import A

class MyClass(object):
    def __init__(self):
        self.obj = A()

    def do_something(self):
        b = self.obj.get_inner_object()
        result = b.do_inner()
        return result

这是我的测试课：

from unittest import TestCase
from unittest.mock import MagicMock
from unittest.mock import patch 
from a import A
class TestMyClass(TestCase):
    def __init__(self):
        self.my_class = MyClass()

    @patch.object(A,'get_inner_object')
    def test_do_something(self, mock_get_inner_object):
        test_res = self.my_class.do_something()

正如您在上面看到的，我想模拟掉我的库的2个方法-get_inner_object（）和do_inner（），它们是get_inner_object（）返回的对象的实例方法。 我能够模拟get_inner_object（），但不清楚如何模拟do_inner（）方法。 请澄清。 这是我正在关注的帮助： https : //www.toptal.com/python/an-introduction-to-mocking-in-python

Answer 1

只是模拟出所有A ：

@patch('a.A')
def test_do_something(self, mock_A):
    mock_b = mock_A.return_value.get_inner_object.return_value
    mock_b.do_inner.return_value = 'mocked return value'

    test_res = self.my_class.do_something()

    self.assertEqual(test_res, 'mocked return value')

毕竟，您在这里测试MyClass ，而不是A

无论哪种方式，无论您使用@patch.object(A, 'get_inner_object')还是修补所有A ， self.obj.get_inner_object()表达式都将调用 Mock实例，因此.return_value将返回.return_value属性。 do_inner方法只是此处返回的模拟的另一个链式调用，因此您可以通过将.return_value属性设置为要测试的内容来设置该方法返回的内容。

要将其转换回您的@patch.object()情况， mock_b然后是mock_inner_object.return_value对象：

@patch.object(A,'get_inner_object')
def test_do_something(self, mock_get_inner_object):
    mock_b = mock_get_inner_object.return_value
    mock_b.do_inner.return_value = 'mocked return value'

    test_res = self.my_class.do_something()

    self.assertEqual(test_res, 'mocked return value')