def test_http_error_5xx(self, mock_urlopen):
status_code = 500
mock_urlopen.return_value.getcode.return_value = status_code
fd = route.connect(mock_urlopen, self.url, None, self.timeout,
self.num_retries, self.retry_wait, False)
self.assertEqual(fd.getcode(), status_code)
mock_urlopen.assert_called_with(self.url, None, self.timeout)
def test_url_error(self, mock_urlopen):
status_code = 400
mock_urlopen.return_value.getcode.return_value = status_code
mock_urlopen.side_effect = urllib.request.URLError('Mock-URLError')
with self.assertRaises(urllib.request.URLError) as e:
fd = route.connect(mock_urlopen, self.url, None, self.timeout,
self.num_retries, self.retry_wait, False)
mock_urlopen.assert_called_with(self.url, None, self.timeout)
def test_http_error_4xx(self, mock_urlopen):
status_code = 400
mock_urlopen.return_value.getcode.return_value = status_code
mock_urlopen.side_effect = urllib.request.HTTPError(
self.url, status_code, 'Mock-HTTPError', {}, None)
with self.assertRaises(urllib.request.HTTPError) as e:
fd = route.connect(mock_urlopen, self.url, None, self.timeout,
self.num_retries, self.retry_wait, False)
self.assertEqual(e.exception.code, status_code)
mock_urlopen.assert_called_with(self.url, None, self.timeout)
def test_status_code_ok(self, mock_urlopen):
mock_urlopen.return_value.getcode.return_value = 305
fd = route.connect(mock_urlopen, self.url, None, self.timeout,
self.num_retries, self.retry_wait, False)
self.assertEqual(fd.getcode(), 305)
mock_urlopen.assert_called_with(self.url, None, self.timeout)