25 lines
887 B
Python
25 lines
887 B
Python
from collections.abc import AsyncIterator, Awaitable, Callable
|
|
|
|
import httpx
|
|
import pytest
|
|
from httpx import ASGITransport, AsyncClient
|
|
|
|
from app.checker import EndpointChecker
|
|
from app.main import app
|
|
|
|
|
|
@pytest.fixture
|
|
async def api_client() -> AsyncIterator[AsyncClient]:
|
|
async with app.router.lifespan_context(app):
|
|
async with AsyncClient(transport=ASGITransport(app=app), base_url="http://test") as client:
|
|
yield client
|
|
|
|
|
|
async def public_resolver(host: str, port: int) -> list[str]:
|
|
return ["93.184.216.34"]
|
|
|
|
|
|
def install_checker(handler: Callable[[httpx.Request], httpx.Response], resolver: Callable[[str, int], Awaitable[list[str]]] = public_resolver) -> None:
|
|
client = httpx.AsyncClient(transport=httpx.MockTransport(handler))
|
|
app.state.checker = EndpointChecker(client, timeout=1, max_redirects=2, max_response_bytes=1024, resolver=resolver)
|