Python返回MagicMock对象而不是return_value

Question

我有一个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'>

我哪里做错了？

Answer 1

当你@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()会触发您的mock方法.