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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
import importlib
from collections.abc import Generator
from typing import Any
import pytest
from tests.views.schema import schema
from .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]),
("AsyncDjangoHttpClient", "async_django", [pytest.mark.django]),
("AsyncFlaskHttpClient", "async_flask", [pytest.mark.flask]),
("ChannelsHttpClient", "channels", [pytest.mark.channels]),
("ChaliceHttpClient", "chalice", [pytest.mark.chalice]),
("DjangoHttpClient", "django", [pytest.mark.django]),
("FastAPIHttpClient", "fastapi", [pytest.mark.fastapi]),
("FlaskHttpClient", "flask", [pytest.mark.flask]),
("QuartHttpClient", "quart", [pytest.mark.quart]),
("SanicHttpClient", "sanic", [pytest.mark.sanic]),
("LitestarHttpClient", "litestar", [pytest.mark.litestar]),
(
"SyncChannelsHttpClient",
"channels",
[pytest.mark.channels, pytest.mark.django_db],
),
]:
try:
client_class = getattr(
importlib.import_module(f".{module}", package="tests.http.clients"),
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(schema)
|
