python - Python返回MagicMock对象而不是return_value

我有一个pythont文件a.py，其中包含两个类A和B。

class A(object):
    def method_a(self):
        return "Class A method a"

class B(object):
    def method_b(self):
        a = A()
        print a.method_a()

我想通过模拟method_b在B类中对A进行单元测试。这是用于此目的的testa.py文件的内容:

import unittest
import mock
import a


class TestB(unittest.TestCase):

    @mock.patch('a.A')
    def test_method_b(self, mock_a):
        mock_a.method_a.return_value = 'Mocked A'
        b = a.B()
        b.method_b()


if __name__ == '__main__':
    unittest.main()

我希望在输出中得到Mocked A。但是我得到的是:

<MagicMock name='A().method_a()' id='4326621392'>

我在哪里做错了？

最佳答案

当您使用@mock.patch('a.A')时，您要用A替换被测代码中的mock_a类。

然后在B.method_b中设置a = A()，现在为a = mock_a()-即a是return_value的mock_a。由于您尚未指定此值，因此它是常规的MagicMock；这也没有配置，因此在调用方法时会得到默认响应(还有另一个MagicMock)。

相反，您希望将return_value的mock_a配置为具有适当的方法，可以按照以下任一方法进行操作:

mock_a().method_a.return_value = 'Mocked A'
    # ^ note parentheses

或者，也许更明确地:

mock_a.return_value.method_a.return_value = 'Mocked A'

您的代码在a = A(分配类，不创建实例)的情况下会起作用，因为a.method_a()会触发您的模拟方法。