Eu tenho um arquivo python a.py
que contém duas classes 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()
Eu gostaria de fazer um teste de unidade method_b
em classe B
zombando A
. Aqui está o conteúdo do arquivo testa.py
para esta finalidade:
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()
Espero obter Mocked A
na saída. Mas o que eu consigo é:
<MagicMock name='A().method_a()' id='4326621392'>
Onde estou fazendo errado?
A()
retorna oreturn_value
frommock_A
(um regularMagicMock
, já que você não especificou mais nada), que não é uma instância da classeA
. Você precisa definir issoreturn_value
para ser algo definidomethod_a
.