test_can_handle_connection_timeout added

This commit is contained in:
Peter Pfeufer 2021-11-23 01:23:34 +01:00
parent 8faadc23b0
commit 49548d6f9f
No known key found for this signature in database
GPG Key ID: 6051D2C6AD4EBC27

View File

@ -1,6 +1,7 @@
from math import ceil from math import ceil
from unittest.mock import patch from unittest.mock import patch
import requests
import requests_mock import requests_mock
from packaging.version import Version as Pep440Version from packaging.version import Version as Pep440Version
@ -307,3 +308,9 @@ class TestFetchListFromGitlab(TestCase):
result = _fetch_list_from_gitlab(self.url, max_pages=max_pages) result = _fetch_list_from_gitlab(self.url, max_pages=max_pages)
self.assertEqual(result, GITHUB_TAGS[:4]) self.assertEqual(result, GITHUB_TAGS[:4])
self.assertEqual(requests_mocker.call_count, max_pages) self.assertEqual(requests_mocker.call_count, max_pages)
@requests_mock.mock()
def test_can_handle_connection_timeout(self, requests_mocker):
requests_mocker.get(self.url, exc=requests.exceptions.ConnectTimeout)
with self.assertRaises(requests.exceptions.ConnectTimeout):
result = _fetch_list_from_gitlab(self.url)