本文介绍了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()将触发您的模拟方法。
这篇关于Python返回MagicMock对象，而不是Return_Value的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！
本站部分内容来源互联网,如果有图片或者内容侵犯您的权益请联系我们删除！