Python zwraca obiekt MagicMock zamiast return_value

Mam plik Pythona, a.pyktóry zawiera dwie klasy Ai 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()

Chciałbym odpocząć method_bw klasie B, kpiąc A. Oto zawartość pliku testa.pydo tego celu:

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()

Spodziewam się, że wyjdę Mocked A. Ale otrzymuję:

<MagicMock name='A().method_a()' id='4326621392'>

Gdzie ja robię źle?

Kiedy @mock.patch('a.A')zamieniasz klasę Aw testowanym kodzie na mock_a.

W B.method_btobie więc ustaw a = A(), co jest teraz a = mock_a()- czyli ajest return_valuez mock_a. Ponieważ nie określiłeś tej wartości, jest to zwykła MagicMock; to też nie jest skonfigurowane, więc otrzymujesz domyślną odpowiedź (jeszcze inną MagicMock) podczas wywoływania metod na niej.

Zamiast tego, chcesz skonfigurować of mieć odpowiednią metodę, którą można zrobić jako:return_valuemock_a

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

lub, być może bardziej szczegółowo:

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

Twój kod zadziałałby w tym przypadku a = A(przypisując klasę, a nie tworząc instancję), ponieważ wtedy a.method_a()wywołałby twoją metodę pozorowaną.