selfprivacy-rest-api/tests/conftest.py

"""Tests configuration."""
# pylint: disable=redefined-outer-name
# pylint: disable=unused-argument
import pytest
from flask import testing
from selfprivacy_api.app import create_app


@pytest.fixture
def tokens_file(mocker, shared_datadir):
    """Mock tokens file."""
    mock = mocker.patch(
        "selfprivacy_api.utils.TOKENS_FILE", shared_datadir / "tokens.json"
    )
    return mock


@pytest.fixture
def app():
    """Flask application."""
    app = create_app(
        {
            "ENABLE_SWAGGER": "1",
        }
    )

    yield app


@pytest.fixture
def client(app, tokens_file):
    """Flask unauthorized test client."""
    return app.test_client()


class AuthorizedClient(testing.FlaskClient):
    """Flask authorized test client."""

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.token = "TEST_TOKEN"

    def open(self, *args, **kwargs):
        if "headers" not in kwargs:
            kwargs["headers"] = {}
        kwargs["headers"]["Authorization"] = f"Bearer {self.token}"
        return super().open(*args, **kwargs)


class WrongAuthClient(testing.FlaskClient):
    """Flask client with wrong token"""

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.token = "WRONG_TOKEN"

    def open(self, *args, **kwargs):
        if "headers" not in kwargs:
            kwargs["headers"] = {}
        kwargs["headers"]["Authorization"] = f"Bearer {self.token}"
        return super().open(*args, **kwargs)


@pytest.fixture
def authorized_client(app, tokens_file):
    """Authorized test client fixture."""
    app.test_client_class = AuthorizedClient
    return app.test_client()


@pytest.fixture
def wrong_auth_client(app, tokens_file):
    """Wrong token test client fixture."""
    app.test_client_class = WrongAuthClient
    return app.test_client()


@pytest.fixture
def runner(app, tokens_file):
    """Flask test runner."""
    return app.test_cli_runner()