37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
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
|