diff --git a/tests/conftest.py b/tests/conftest.py index 4016ff0..24ea336 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1 +1,36 @@ -# Env var setup that must run before app.config is imported. +from collections.abc import AsyncIterator, Awaitable, Callable + +import httpx +import pytest + +from app.api import create_app +from app.checker import EndpointChecker +from app.config import Settings + + +async def public_resolver(_: str, __: int) -> list[str]: + return ["93.184.216.34"] + + +@pytest.fixture +def client_factory() -> Callable[[httpx.AsyncBaseTransport], Awaitable[httpx.AsyncClient]]: + async def make(transport: httpx.AsyncBaseTransport) -> httpx.AsyncClient: + checker = EndpointChecker(1, 3, resolver=public_resolver, transport=transport) + app = create_app(Settings(), checker=checker) + return httpx.AsyncClient( + transport=httpx.ASGITransport(app=app), base_url="http://test" + ) + + return make + + +@pytest.fixture +async def client( + client_factory: Callable[[httpx.AsyncBaseTransport], Awaitable[httpx.AsyncClient]], +) -> AsyncIterator[httpx.AsyncClient]: + async def handler(_: httpx.Request) -> httpx.Response: + return httpx.Response(200) + + value = await client_factory(httpx.MockTransport(handler)) + async with value: + yield value