У меня есть файл python, a.py
который содержит два класса A
и 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()
Я хотел бы UnitTest method_b
в классе B
по насмешливый A
. Вот содержимое файла testa.py
для этой цели:
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()
Я ожидаю получить Mocked A
результат. Но вот что я получаю:
<MagicMock name='A().method_a()' id='4326621392'>
Где я делаю не так?
python
python-unittest
magicmock
Мехди Джафарния Джахроми
источник
источник
A()
возвращаетreturn_value
frommock_A
(обычныйMagicMock
, поскольку вы ничего не указали), который не является экземпляром классаA
. Вам нужно установить этоreturn_value
как то, что имеет определенныйmethod_a
.mock_a
которое должно иметь метод, а не онmock_a
сам.mock_a().method_a.return_value = 'Mocked A'
иmock_a.return_value.method_a.return_value = 'Mocked A'
работали. Большое спасибо за ваши комментарии. Не могли бы вы сформулировать это как ответ?Ответы:
Когда
@mock.patch('a.A')
вы заменяете классA
в тестируемом коде наmock_a
.В
B.method_b
вас тогда наборa = A()
, который в настоящее времяa = mock_a()
- то естьa
этоreturn_value
изmock_a
. Поскольку вы не указали это значение, это обычноеMagicMock
; это тоже не настроено, поэтому вы получите ответ по умолчанию (еще одинMagicMock
) при вызове его методов.Вместо этого, вы хотите настроить от иметь соответствующий метод, который вы можете сделать либо как:
return_value
mock_a
mock_a().method_a.return_value = 'Mocked A' # ^ note parentheses
или, возможно, более явно:
mock_a.return_value.method_a.return_value = 'Mocked A'
Ваш код сработал бы в этом случае
a = A
(присвоение класса, а не создание экземпляра), поскольку тогдаa.method_a()
сработал бы ваш фиктивный метод.источник
mock_data.configure_mock(columns='my_column')
который решает эту проблему. Спасибо за ответ. ( Ссылка : bradmontgomery.net/blog/how-world-do-you-mock-name-attribute )