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

Я хотел бы UnitTest method_bв классе Bпо насмешливый A. Вот содержимое файла 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из mock_a. Поскольку вы не указали это значение, это обычное MagicMock; это тоже не настроено, поэтому вы получите ответ по умолчанию (еще один MagicMock) при вызове его методов.

Вместо этого, вы хотите настроить от иметь соответствующий метод, который вы можете сделать либо как:return_valuemock_a

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()сработал бы ваш фиктивный метод.