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

回答

相關問題