def test_get_holidays_timeout(self):
with patch('my_calendar.requests', autospec=True) as mock_requests:
mock_requests.get.side_effect = Timeout
with self.assertRaises(Timeout):
get_holidays()
mock_requests.get.assert_called_once()
def test_get_holidays_timeout(self, mock_requests):
with self.assertRaises(Timeout):
get_holidays()
def test_get_holidays_timeout(self, mock_requests):
mock_requests.get.side_effect = Timeout
with self.assertRaises(Timeout):
get_holidays()
mock_requests.get.assert_called_once()
def test_get_holidays_200(self, mock_requests):
mock_requests.return_value.status_code = 200
response = get_holidays()
self.assertEqual(response.status_code, 200)
def test_get_holidays_return_value(self, mock_requests):
mock_requests.return_value.value = "Sunday"
response = get_holidays()
self.assertEqual(response.value, "Sunday")
import my_calendar as calendar from unittest.mock import create_autospec calendar = create_autospec(calendar) calendar.is_weekday() calendar.get_holidays()
import my_calendar as calendar
from unittest.mock import Mock
# Автоматически патчим те методы которые нашли в объекте
mocked = True
if mocked:
calendar = Mock(spec=calendar)
print(calendar.is_weekday())
print(calendar.get_holidays())
