conftest.py

import importlib
from collections.abc import Generator
from typing import Any

import pytest

from tests.http.clients.base import HttpClient


def _get_http_client_classes() -> Generator[Any, None, None]:
    for client, module, marks in [
        ("AioHttpClient", "aiohttp", [pytest.mark.aiohttp]),
        ("AsgiHttpClient", "asgi", [pytest.mark.asgi]),
        ("ChannelsHttpClient", "channels", [pytest.mark.channels]),
        ("FastAPIHttpClient", "fastapi", [pytest.mark.fastapi]),
        ("LitestarHttpClient", "litestar", [pytest.mark.litestar]),
        ("QuartHttpClient", "quart", [pytest.mark.quart]),
    ]:
        try:
            client_class = getattr(
                importlib.import_module(f"tests.http.clients.{module}"), client
            )
        except ImportError:
            client_class = None

        yield pytest.param(
            client_class,
            marks=[
                *marks,
                pytest.mark.skipif(
                    client_class is None, reason=f"Client {client} not found"
                ),
            ],
        )


@pytest.fixture(params=_get_http_client_classes())
def http_client_class(request: Any) -> type[HttpClient]:
    return request.param


@pytest.fixture
def http_client(http_client_class: type[HttpClient]) -> HttpClient:
    return http_client_class()