77 lines
2.5 KiB
Python
77 lines
2.5 KiB
Python
|
import urllib.parse
|
||
|
|
from typing import Dict, Optional
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(scope="function")
|
||
|
|
async def response(app_client):
|
||
|
|
return await app_client.get("/")
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(scope="function")
|
||
|
|
async def response_json(response) -> Dict:
|
||
|
|
return response.json()
|
||
|
|
|
||
|
|
|
||
|
|
def get_link(landing_page, rel_type, method: Optional[str] = None):
|
||
|
|
return next(
|
||
|
|
filter(
|
||
|
|
lambda link: link["rel"] == rel_type
|
||
|
|
and (not method or link.get("method") == method),
|
||
|
|
landing_page["links"],
|
||
|
|
),
|
||
|
|
None,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_landing_page_health(response):
|
||
|
|
"""Test landing page"""
|
||
|
|
assert response.status_code == 200
|
||
|
|
assert response.headers["content-type"] == "application/json"
|
||
|
|
|
||
|
|
|
||
|
|
# Parameters for test_landing_page_links test below.
|
||
|
|
# Each tuple has the following values (in this order):
|
||
|
|
# - Rel type of link to test
|
||
|
|
# - Expected MIME/Media Type
|
||
|
|
# - Expected relative path
|
||
|
|
link_tests = [
|
||
|
|
("root", "application/json", "/"),
|
||
|
|
("conformance", "application/json", "/conformance"),
|
||
|
|
("service-doc", "text/html", "/api.html"),
|
||
|
|
("service-desc", "application/vnd.oai.openapi+json;version=3.0", "/api"),
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("rel_type,expected_media_type,expected_path", link_tests)
|
||
|
|
async def test_landing_page_links(
|
||
|
|
response_json: Dict, app_client, app, rel_type, expected_media_type, expected_path
|
||
|
|
):
|
||
|
|
link = get_link(response_json, rel_type)
|
||
|
|
|
||
|
|
assert link is not None, f"Missing {rel_type} link in landing page"
|
||
|
|
assert link.get("type") == expected_media_type
|
||
|
|
|
||
|
|
link_path = urllib.parse.urlsplit(link.get("href")).path
|
||
|
|
assert link_path == app.state.router_prefix + expected_path
|
||
|
|
|
||
|
|
resp = await app_client.get(link_path.rsplit("/", 1)[-1])
|
||
|
|
assert resp.status_code == 200
|
||
|
|
|
||
|
|
|
||
|
|
# This endpoint currently returns a 404 for empty result sets, but testing for this response
|
||
|
|
# code here seems meaningless since it would be the same as if the endpoint did not exist. Once
|
||
|
|
# https://github.com/stac-utils/stac-fastapi/pull/227 has been merged we can add this to the
|
||
|
|
# parameterized tests above.
|
||
|
|
def test_search_link(response_json: Dict, app):
|
||
|
|
for search_link in [
|
||
|
|
get_link(response_json, "search", "GET"),
|
||
|
|
get_link(response_json, "search", "POST"),
|
||
|
|
]:
|
||
|
|
assert search_link is not None
|
||
|
|
assert search_link.get("type") == "application/geo+json"
|
||
|
|
|
||
|
|
search_path = urllib.parse.urlsplit(search_link.get("href")).path
|
||
|
|
assert search_path == app.state.router_prefix + "/search"
|