Python renvoie l'objet MagicMock au lieu de return_value

J'ai un fichier python a.pyqui contient deux classes Aet 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()

Je voudrais unittest method_ben classe en me Bmoquant A. Voici le contenu du fichier testa.pyà cet effet:

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

Je m'attends à entrer Mocked Adans la sortie. Mais ce que j'obtiens c'est:

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

Où vais-je mal?

Lorsque vous @mock.patch('a.A'), vous remplacez la classe Adans le code testé par mock_a.

En B.method_bvous définissez alors a = A(), qui est maintenant a = mock_a()- c'est à dire aest le return_valuede mock_a. Comme vous n'avez pas spécifié cette valeur, c'est une valeur régulière MagicMock; cela n'est pas configuré non plus, vous obtenez donc la réponse par défaut (encore une autre MagicMock) lorsque vous appelez des méthodes dessus.

, Vous voulez plutôt que de configurer le return_valuedemock_a avoir la méthode appropriée, que vous pouvez faire comme soit:

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

ou, peut-être plus explicitement:

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

Votre code aurait fonctionné dans le cas a = A(attribuer la classe, ne pas créer d'instance), comme alors a.method_a()aurait déclenché votre méthode fictive.