Python gibt das MagicMock-Objekt anstelle von return_value zurück

Ich habe eine Python-Datei, a.pydie zwei Klassen enthält Aund 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()

Ich würde gerne method_bim Unterricht Bdurch Verspotten unittest A. Hier ist der Inhalt der Datei testa.pyfür diesen Zweck:

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

Ich erwarte, Mocked Ain die Ausgabe zu kommen. Aber was ich bekomme ist:

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

Wo mache ich falsch?

Wenn Sie @mock.patch('a.A'), ersetzen Sie die Klasse Aim zu testenden Code durch mock_a.

In B.method_bdir dann setzen a = A(), was jetzt ist a = mock_a()- dh das aist return_valuevon mock_a. Da Sie diesen Wert nicht angegeben haben, handelt es sich um einen regulären Wert MagicMock. Dies ist auch nicht konfiguriert, daher erhalten Sie die Standardantwort (noch eine andere MagicMock), wenn Sie Methoden darauf aufrufen.

Stattdessen möchten Sie konfigurieren die return_valuevonmock_a haben , die entsprechende Methode, die Sie entweder tun können:

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

oder vielleicht expliziter:

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

Ihr Code hätte in diesem Fall funktioniert a = A(Zuweisen der Klasse, keine Instanz erstellen), da dann a.method_a()Ihre Scheinmethode ausgelöst worden wäre.