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 AsyncIterator
import pytest
from litestar import Litestar, MediaType, get
from litestar.status_codes import HTTP_200_OK
from litestar.testing import AsyncTestClient
@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")
async def test_client() -> AsyncIterator[AsyncTestClient[Litestar]]:
async with AsyncTestClient(app=app) as client:
yield client
async def test_health_check_with_fixture(test_client: AsyncTestClient[Litestar]) -> None:
response = await test_client.get("/health-check")
assert response.status_code == HTTP_200_OK
assert response.text == "healthy"
|