Ho un file python a.py
che contiene due classi A
e 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_b
in classe prendendo in B
giro A
. Ecco il contenuto del file testa.py
per 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 A
in uscita. Ma quello che ottengo è:
<MagicMock name='A().method_a()' id='4326621392'>
Dove sto sbagliando?
A()
restituiscereturn_value
frommock_A
(un normaleMagicMock
, poiché non hai specificato nient'altro), che non è un'istanza della classeA
. È necessario impostarloreturn_value
in modo che abbia un valore definitomethod_a
.