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

Я хотів би пройти тестування 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()це спричинило б ваш макетний метод.