Skip to content

Pytest Plugin

admin()

Get admin user.

Source code in saritasa_drf_tools/pytest_plugin.py
40
41
42
43
@pytest.fixture
def admin() -> typing.Any:
    """Get admin user."""
    raise NotImplementedError("Set up `admin` fixture")

admin_api_client(api_client_factory, admin)

Create api client.

Source code in saritasa_drf_tools/pytest_plugin.py
46
47
48
49
50
51
52
53
54
@pytest.fixture
def admin_api_client(  # noqa: ANN201
    api_client_factory,  # noqa: ANN001
    admin,  # noqa: ANN001
):
    """Create api client."""
    api_client = api_client_factory()
    api_client.force_authenticate(user=admin)
    return api_client

api_client(api_client_factory)

Create api client.

Source code in saritasa_drf_tools/pytest_plugin.py
17
18
19
20
@pytest.fixture
def api_client(api_client_factory):  # noqa: ANN001, ANN201
    """Create api client."""
    return api_client_factory()

api_client_factory()

Create factory which will generate api clients.

Source code in saritasa_drf_tools/pytest_plugin.py
 6
 7
 8
 9
10
11
12
13
14
@pytest.fixture
def api_client_factory():  # noqa: ANN201
    """Create factory which will generate api clients."""
    from rest_framework import test

    def _create_api_client() -> test.APIClient:
        return test.APIClient()

    return _create_api_client

user()

Get ordinary user.

Source code in saritasa_drf_tools/pytest_plugin.py
23
24
25
26
@pytest.fixture
def user() -> typing.Any:
    """Get ordinary user."""
    raise NotImplementedError("Set up `user` fixture")

user_api_client(api_client_factory, user)

Create api client.

Source code in saritasa_drf_tools/pytest_plugin.py
29
30
31
32
33
34
35
36
37
@pytest.fixture
def user_api_client(  # noqa: ANN201
    api_client_factory,  # noqa: ANN001
    user,  # noqa: ANN001
):
    """Create api client."""
    api_client = api_client_factory()
    api_client.force_authenticate(user=user)
    return api_client