Python返回MagicMock对象，而不是Return_Value

Python returns MagicMock object instead of return_value(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。

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

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

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