Единичное тестирование: как утверждать несколько вызовов одного и того же метода?
У меня есть метод, который вызывает другой метод дважды, с разными аргументами.
class A(object):
def helper(self, arg_one, arg_two):
"""Return something which depends on arguments."""
def caller(self):
value_1 = self.helper(foo, bar) # First call.
value_2 = self.helper(foo_bar, bar_foo) # Second call!
Использование assert_called_with помогает мне утверждать только первый вызов, а не второй. Даже assert_called_once_with не кажется полезным. Что мне здесь не хватает? Есть ли способ проверить такие звонки?
Ответы
Ответ 1
Вы можете использовать mock_calls который содержит все вызовы, сделанные для метода. Этот список содержит первый вызов, второй вызов и все последующие вызовы. Таким образом, вы можете писать утверждения с помощью mock_calls[1] чтобы указать что-то о втором вызове.
Например, если m = mock.Mock() и код выполняет m.method(123) то вы пишете:
assert m.method.mock_calls == [mock.call(123)]
который утверждает, что список вызовов m.method - это ровно один вызов, а именно вызов с аргументом 123.
Ответ 2
Чтобы добавить к ответу Саймона Виссера, вы можете использовать метод unittest.TestCase self.assertEqual() вместо синтаксиса assert, который, я бы сказал, является лучшей практикой в контексте Unit Test, поскольку вы также можете добавлять к нему комментарии, которые будут отображаться всякий раз, когда что-то идет не так.
Так, например:
self.assertEqual(
[
mock.call(1, 'ValueA', True)),
mock.call(2, 'ValueB', False)),
mock.call(3, 'ValueC', False))
],
mock_cur.execute.mock_calls,
"The method was not called with the correct arguments."
)
