Python返回MagicMock对象而不是return_value

我有一个Python文件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()可以触发您的模拟方法。