def test_fetch_url_connection_error():
"""
Connection errors are treated as NotFoundErrors.
"""
with patch("cheddar.index.remote.get") as mocked:
mocked.side_effect = ConnectionError
with assert_raises(NotFoundError):
fetch_url("http://example.com", TIMEOUT, getLogger())
def test_fetch_url_timeout():
"""
Timeouts are treated as NotFoundErrors.
"""
with patch("cheddar.index.remote.get") as mocked:
mocked.side_effect = Timeout
with assert_raises(NotFoundError):
fetch_url("http://example.com", TIMEOUT, getLogger())
def test_fetch_url_not_ok():
"""
Non-200 status codes are treated as NotFoundErrors.
"""
with patch("cheddar.index.remote.get") as mocked:
mocked.return_value = MagicMock()
mocked.return_value.status_code = codes.bad_request
with assert_raises(NotFoundError):
fetch_url("http://example.com", TIMEOUT, getLogger())
def test_fetch_url_ok():
"""
200 status codes succeed.
"""
with patch("cheddar.index.remote.get") as mocked:
mocked.return_value = MagicMock()
mocked.return_value.status_code = codes.ok
response = fetch_url("http://example.com", TIMEOUT, getLogger())
eq_(codes.ok, response.status_code)
