Python, return_value yerine MagicMock nesnesini döndürür

a.pyİki sınıf içeren bir python dosyam var Ave 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()

Ben unittest istiyorum method_bsınıfta Balay ile A. İşte testa.pybu amaç için dosyanın içeriği :

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

Çıktıya girmeyi bekliyorum Mocked A. Ama aldığım şey:

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

Nerede yanlış yapıyorum

Siz @mock.patch('a.A'), Atest edilen koddaki sınıfı ile değiştiriyorsunuz mock_a.

In B.method_bsize daha sonra ayarlamak a = A()şimdi hangi a = mock_a()- yani aolduğu return_valuebir mock_a. Bu değeri belirtmediğiniz için normaldir MagicMock; bu da yapılandırılmamıştır, bu nedenle MagicMocküzerinde yöntemler çağırırken varsayılan yanıtı (yine başka bir ) alırsınız .

Bunun yerine, yapılandırmak istediğiniz bir ya kadar yapabilirsiniz uygun yöntemi, sahip olmak:return_valuemock_a

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

veya belki daha açık bir şekilde:

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

Kodunuz a = A, o a.method_a()zaman sahte yönteminizi tetikleyeceği gibi durumda (sınıf atamak, bir örnek oluşturmak değil) işe yarardı.