1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
|
from typing import Iterator
import pytest
from litestar import Litestar, MediaType, get
from litestar.status_codes import HTTP_200_OK
from litestar.testing import TestClient
@get(path="/health-check", media_type=MediaType.TEXT, sync_to_thread=False)
def health_check() -> str:
return "healthy"
app = Litestar(route_handlers=[health_check], debug=True)
@pytest.fixture(scope="function")
def test_client() -> Iterator[TestClient[Litestar]]:
with TestClient(app=app) as client:
yield client
def test_health_check_with_fixture(test_client: TestClient[Litestar]) -> None:
response = test_client.get("/health-check")
assert response.status_code == HTTP_200_OK
assert response.text == "healthy"
|