Python restituisce l'oggetto MagicMock invece di return_value

Ho un file python a.pyche contiene due classi Ae 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()

Vorrei unirmi method_bin classe prendendo in Bgiro A. Ecco il contenuto del file testa.pyper questo scopo:

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

Mi aspetto di entrare Mocked Ain uscita. Ma quello che ottengo è:

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

Dove sto sbagliando?

Quando tu @mock.patch('a.A'), stai sostituendo la classe Anel codice sotto test con mock_a.

In B.method_bte allora imposta a = A(), che è adesso a = mock_a()- cioè aè il return_valuedi mock_a. Poiché non hai specificato questo valore, è regolare MagicMock; nemmeno questo è configurato, quindi ottieni la risposta predefinita (ancora un'altra MagicMock) quando chiami i metodi su di esso.

Invece, vuoi configurare il return_valuedimock_a per avere il metodo appropriato, cosa che puoi fare come:

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

o, forse più esplicitamente:

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

Il tuo codice avrebbe funzionato nel caso a = A(assegnando la classe, non creando un'istanza), poiché allora a.method_a()avrebbe attivato il tuo metodo fittizio.