mirror of
https://github.com/mealie-recipes/mealie.git
synced 2024-10-17 20:50:34 -07:00
73 lines
1.6 KiB
Python
73 lines
1.6 KiB
Python
import contextlib
|
|
from collections.abc import Generator
|
|
|
|
from pytest import MonkeyPatch, fixture
|
|
|
|
|
|
def _clean_temp_dir():
|
|
with contextlib.suppress(Exception):
|
|
temp_dir = Path(__file__).parent / ".temp"
|
|
|
|
if temp_dir.exists():
|
|
import shutil
|
|
|
|
shutil.rmtree(temp_dir, ignore_errors=True)
|
|
|
|
|
|
_clean_temp_dir()
|
|
|
|
mp = MonkeyPatch()
|
|
mp.setenv("PRODUCTION", "True")
|
|
mp.setenv("TESTING", "True")
|
|
mp.setenv("ALLOW_SIGNUP", "True")
|
|
mp.setenv("OPENAI_API_KEY", "dummy-api-key")
|
|
from pathlib import Path
|
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
from mealie.app import app
|
|
from mealie.core import config
|
|
from mealie.db.db_setup import SessionLocal, generate_session
|
|
from mealie.db.init_db import main
|
|
from tests import data as test_data
|
|
from tests.fixtures import * # noqa: F403 F401
|
|
|
|
main()
|
|
|
|
|
|
def override_get_db():
|
|
try:
|
|
db = SessionLocal()
|
|
yield db
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
@fixture(scope="session")
|
|
def api_client():
|
|
app.dependency_overrides[generate_session] = override_get_db
|
|
|
|
yield TestClient(app)
|
|
|
|
with contextlib.suppress(Exception):
|
|
settings = config.get_app_settings()
|
|
settings.DB_PROVIDER.db_path.unlink() # Handle SQLite Provider
|
|
|
|
|
|
@fixture(scope="session")
|
|
def test_image_jpg():
|
|
return test_data.images_test_image_1
|
|
|
|
|
|
@fixture(scope="session")
|
|
def test_image_png():
|
|
return test_data.images_test_image_2
|
|
|
|
|
|
@fixture(scope="session", autouse=True)
|
|
def global_cleanup() -> Generator[None, None, None]:
|
|
"""Purges the .temp directory used for testing"""
|
|
|
|
yield None
|
|
_clean_temp_dir()
|