Python返回MagicMock对象，而不是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。

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

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

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

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