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()

我想通过模拟A来单元测试B类中的method_b。以下是该文件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。因为您没有指定这个值，所以它是一个常规的MagicMock；它也没有配置，所以当调用它的方法时，您会得到默认的响应(还有一个MagicMock)。

相反，您希望将mock_a的return_value配置为具有适当的方法，您可以这样做：

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()将触发您的模拟方法。