mirror of
https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git
synced 2024-11-17 16:09:14 +00:00
e125f3a4b1
except for one which will temporarily remain gitkeeps are to prevent shared_datadir from erroring out in a freshly cloned repo. for now huey database and jobs fixtures use shared_datadir
70 lines
1.7 KiB
Python
70 lines
1.7 KiB
Python
"""Tests configuration."""
|
|
# pylint: disable=redefined-outer-name
|
|
# pylint: disable=unused-argument
|
|
import os
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
from shutil import copy
|
|
import os.path as path
|
|
|
|
|
|
def pytest_generate_tests(metafunc):
|
|
os.environ["TEST_MODE"] = "true"
|
|
|
|
|
|
def global_data_dir():
|
|
return path.join(path.dirname(__file__), "data")
|
|
|
|
|
|
@pytest.fixture
|
|
def tokens_file(mocker, tmpdir):
|
|
"""Mock tokens file."""
|
|
tmp_file = tmpdir / "tokens.json"
|
|
source_file = path.join(global_data_dir(), "tokens.json")
|
|
copy(source_file, tmp_file)
|
|
mock = mocker.patch("selfprivacy_api.utils.TOKENS_FILE", tmp_file)
|
|
return mock
|
|
|
|
|
|
@pytest.fixture
|
|
def jobs_file(mocker, shared_datadir):
|
|
"""Mock tokens file."""
|
|
mock = mocker.patch("selfprivacy_api.utils.JOBS_FILE", shared_datadir / "jobs.json")
|
|
return mock
|
|
|
|
|
|
@pytest.fixture
|
|
def huey_database(mocker, shared_datadir):
|
|
"""Mock huey database."""
|
|
mock = mocker.patch(
|
|
"selfprivacy_api.utils.huey.HUEY_DATABASE", shared_datadir / "huey.db"
|
|
)
|
|
return mock
|
|
|
|
|
|
@pytest.fixture
|
|
def client(tokens_file, huey_database, jobs_file):
|
|
from selfprivacy_api.app import app
|
|
|
|
return TestClient(app)
|
|
|
|
|
|
@pytest.fixture
|
|
def authorized_client(tokens_file, huey_database, jobs_file):
|
|
"""Authorized test client fixture."""
|
|
from selfprivacy_api.app import app
|
|
|
|
client = TestClient(app)
|
|
client.headers.update({"Authorization": "Bearer TEST_TOKEN"})
|
|
return client
|
|
|
|
|
|
@pytest.fixture
|
|
def wrong_auth_client(tokens_file, huey_database, jobs_file):
|
|
"""Wrong token test client fixture."""
|
|
from selfprivacy_api.app import app
|
|
|
|
client = TestClient(app)
|
|
client.headers.update({"Authorization": "Bearer WRONG_TOKEN"})
|
|
return client
|