Блог разработчика: Golang, Криптовалюты, Автоматизация, Kubernetes
Продолжаем про юниты тесты в питоне. Удобная конструкция для проверки на вызов исключения в тесте:
|
1 2 |
def test_something_with_exception(self): self.assertRaises(ValueError, self.client.send_something_method, param='one_param') |
Тем самым мы тестируем метод send_something_method инциализрованного ранее объекта сlient с аргументами метода param и ожидаем что он выбросит исключение.
Встретился участок кода с контекстным менеджером. Потербовалось написать для этого юнит тест. Для справки, контекстным менеджером в питоне называется конструкция вида:
|
1 2 |
with self.statsd_client.timer('metric'): pass |
Для того чтобы сделать мок метода и не ловить ошибки
|
1 |
AtributeError __exit__ |
мы должны правильно описать мок:
|
1 2 3 4 5 6 |
timer = Mock() timer.__enter__ = Mock(return_value=(Mock(), None)) timer.__exit__ = Mock(return_value=None) self.statsd_client = Mock() self.statsd_client.timer.return_value = timer |