81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
|
from httpx import ASGITransport, AsyncClient
|
||
|
|
from stac_fastapi.api.app import StacApi
|
||
|
|
|
||
|
|
from stac_fastapi.pgstac.config import PostgresSettings, Settings
|
||
|
|
from stac_fastapi.pgstac.core import CoreCrudClient, health_check
|
||
|
|
from stac_fastapi.pgstac.db import close_db_connection, connect_to_db
|
||
|
|
|
||
|
|
|
||
|
|
async def test_ping_no_param(app_client):
|
||
|
|
"""
|
||
|
|
Test ping endpoint with a mocked client.
|
||
|
|
Args:
|
||
|
|
app_client (TestClient): mocked client fixture
|
||
|
|
"""
|
||
|
|
res = await app_client.get("/_mgmt/ping")
|
||
|
|
assert res.status_code == 200
|
||
|
|
assert res.json() == {"message": "PONG"}
|
||
|
|
|
||
|
|
|
||
|
|
async def test_health(app_client):
|
||
|
|
"""
|
||
|
|
Test health endpoint
|
||
|
|
|
||
|
|
Args:
|
||
|
|
app_client (TestClient): mocked client fixture
|
||
|
|
|
||
|
|
"""
|
||
|
|
res = await app_client.get("/_mgmt/health")
|
||
|
|
assert res.status_code == 200
|
||
|
|
body = res.json()
|
||
|
|
assert body["status"] == "UP"
|
||
|
|
assert body["pgstac"]["status"] == "UP"
|
||
|
|
assert body["pgstac"]["pgstac_version"]
|
||
|
|
|
||
|
|
|
||
|
|
async def test_health_503(database):
|
||
|
|
"""Test health endpoint error."""
|
||
|
|
|
||
|
|
# No lifespan so no `get_connection` is application state
|
||
|
|
api = StacApi(
|
||
|
|
settings=Settings(testing=True),
|
||
|
|
extensions=[],
|
||
|
|
client=CoreCrudClient(),
|
||
|
|
health_check=health_check,
|
||
|
|
)
|
||
|
|
|
||
|
|
async with AsyncClient(
|
||
|
|
transport=ASGITransport(app=api.app), base_url="http://test"
|
||
|
|
) as client:
|
||
|
|
res = await client.get("/_mgmt/health")
|
||
|
|
assert res.status_code == 503
|
||
|
|
body = res.json()
|
||
|
|
assert body["status"] == "DOWN"
|
||
|
|
assert body["lifespan"]["status"] == "DOWN"
|
||
|
|
assert body["lifespan"]["message"] == "application lifespan wasn't run"
|
||
|
|
assert body["pgstac"]["status"] == "DOWN"
|
||
|
|
assert body["pgstac"]["message"] == "Could not connect to database"
|
||
|
|
|
||
|
|
# No lifespan so no `get_connection` is application state
|
||
|
|
postgres_settings = PostgresSettings(
|
||
|
|
pguser=database.user,
|
||
|
|
pgpassword=database.password,
|
||
|
|
pghost=database.host,
|
||
|
|
pgport=database.port,
|
||
|
|
pgdatabase=database.dbname,
|
||
|
|
)
|
||
|
|
# Create connection pool but close it just after
|
||
|
|
await connect_to_db(api.app, postgres_settings=postgres_settings)
|
||
|
|
await close_db_connection(api.app)
|
||
|
|
|
||
|
|
async with AsyncClient(
|
||
|
|
transport=ASGITransport(app=api.app), base_url="http://test"
|
||
|
|
) as client:
|
||
|
|
res = await client.get("/_mgmt/health")
|
||
|
|
assert res.status_code == 503
|
||
|
|
body = res.json()
|
||
|
|
assert body["status"] == "DOWN"
|
||
|
|
assert body["lifespan"]["status"] == "UP"
|
||
|
|
assert body["pgstac"]["status"] == "DOWN"
|
||
|
|
assert body["pgstac"]["message"] == "pool is closed"
|