mirror of
https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git
synced 2024-11-17 16:09:14 +00:00
38 lines
858 B
Python
38 lines
858 B
Python
|
import pytest
|
||
|
|
from flask import testing
|
||
|
|
from selfprivacy_api.app import create_app
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def app():
|
||
|
|
app = create_app({
|
||
|
|
"AUTH_TOKEN": "TEST_TOKEN",
|
||
|
|
"ENABLE_SWAGGER": "0",
|
||
|
|
})
|
||
|
|
|
||
|
|
yield app
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def client(app):
|
||
|
|
return app.test_client()
|
||
|
|
|
||
|
|
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)
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def authorized_client(app):
|
||
|
|
app.test_client_class = AuthorizedClient
|
||
|
|
return app.test_client()
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def runner(app):
|
||
|
|
return app.test_cli_runner()
|