Python返回MagicMock对象而不是return_value

我有一个python文件a.py，它包含两个类A和B。Python返回MagicMock对象而不是return_value

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'>

我在哪里做错了？

来源

2016-07-05 Mehdi Jafarnia Jahromi

测试时，'A（）'从'mock_A'（一个普通的'MagicMock'，你没有指定任何东西）返回'return_value'，它不是类'A'的一个实例。您需要将'return_value'设置为具有已定义的'method_a'的东西。 – jonrsharpe

mock_a.method_a.return_value ='Mocked A'=> mock_a（）。method_a.return_value ='Mocked A'应该更好:) –

@AliSAIDOMAR正好是正确的，它是调用'mock_a'的返回值，应该有方法，而不是'mock_a'本身。 – jonrsharpe

当您使用@mock.patch('a.A')时，您正在使用mock_a替换正在测试的代码中的类A。

在B.method_b你再设置a = A()，这是现在a = mock_a() - 即a是mock_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()会触发你的模拟方法。

来源

2016-07-05 09:05:29 jonrsharpe

Python返回MagicMock对象而不是return_value

回答

相关问题