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)