selfprivacy-rest-api/tests/conftest.py

44 lines
888 B
Python
Raw Normal View History

import pytest
from flask import testing
from selfprivacy_api.app import create_app
@pytest.fixture
def app():
2021-11-30 21:53:39 +00:00
app = create_app(
{
"AUTH_TOKEN": "TEST_TOKEN",
"ENABLE_SWAGGER": "0",
}
)
yield app
@pytest.fixture
def client(app):
return app.test_client()
2021-11-30 21:53:39 +00:00
class AuthorizedClient(testing.FlaskClient):
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)
2021-11-30 21:53:39 +00:00
@pytest.fixture
def authorized_client(app):
app.test_client_class = AuthorizedClient
return app.test_client()
2021-11-30 21:53:39 +00:00
@pytest.fixture
def runner(app):
2021-11-30 21:53:39 +00:00
return app.test_cli_runner()