import os from typing import Generator, Optional, Union import pytest from fittrackee import create_app, db from fittrackee.application.models import AppConfig from fittrackee.application.utils import update_app_config_from_database def get_app_config( with_config: Optional[bool] = False, max_workouts: Optional[int] = None, max_single_file_size: Optional[Union[int, float]] = None, max_zip_file_size: Optional[Union[int, float]] = None, max_users: Optional[int] = None, ) -> Optional[AppConfig]: if with_config: config = AppConfig() config.gpx_limit_import = 10 if max_workouts is None else max_workouts config.max_single_file_size = ( (1 if max_single_file_size is None else max_single_file_size) * 1024 * 1024 ) config.max_zip_file_size = ( (10 if max_zip_file_size is None else max_zip_file_size) * 1024 * 1024 ) config.max_users = 100 if max_users is None else max_users db.session.add(config) db.session.commit() return config return None def get_app( with_config: Optional[bool] = False, max_workouts: Optional[int] = None, max_single_file_size: Optional[Union[int, float]] = None, max_zip_file_size: Optional[Union[int, float]] = None, max_users: Optional[int] = None, ) -> Generator: app = create_app() with app.app_context(): try: db.create_all() app_db_config = get_app_config( with_config, max_workouts, max_single_file_size, max_zip_file_size, max_users, ) if app_db_config: update_app_config_from_database(app, app_db_config) yield app except Exception as e: print(f'Error with app configuration: {e}') finally: db.session.remove() db.drop_all() # close unused idle connections => avoid the following error: # FATAL: remaining connection slots are reserved for # non-replication superuser connections db.engine.dispose() return app @pytest.fixture def app(monkeypatch: pytest.MonkeyPatch) -> Generator: monkeypatch.setenv('EMAIL_URL', 'smtp://none:none@0.0.0.0:1025') if os.getenv('TILE_SERVER_URL'): monkeypatch.delenv('TILE_SERVER_URL') if os.getenv('MAP_ATTRIBUTION'): monkeypatch.delenv('MAP_ATTRIBUTION') yield from get_app(with_config=True) @pytest.fixture def app_with_max_workouts(monkeypatch: pytest.MonkeyPatch) -> Generator: monkeypatch.setenv('EMAIL_URL', 'smtp://none:none@0.0.0.0:1025') yield from get_app(with_config=True, max_workouts=2) @pytest.fixture def app_with_max_file_size_equals_0( monkeypatch: pytest.MonkeyPatch, ) -> Generator: monkeypatch.setenv('EMAIL_URL', 'smtp://none:none@0.0.0.0:1025') yield from get_app(with_config=True, max_single_file_size=0) @pytest.fixture def app_with_max_file_size(monkeypatch: pytest.MonkeyPatch) -> Generator: monkeypatch.setenv('EMAIL_URL', 'smtp://none:none@0.0.0.0:1025') yield from get_app(with_config=True, max_single_file_size=0.001) @pytest.fixture def app_with_max_zip_file_size(monkeypatch: pytest.MonkeyPatch) -> Generator: monkeypatch.setenv('EMAIL_URL', 'smtp://none:none@0.0.0.0:1025') yield from get_app(with_config=True, max_zip_file_size=0.001) @pytest.fixture def app_with_3_users_max(monkeypatch: pytest.MonkeyPatch) -> Generator: monkeypatch.setenv('EMAIL_URL', 'smtp://none:none@0.0.0.0:1025') yield from get_app(with_config=True, max_users=3) @pytest.fixture def app_no_config() -> Generator: yield from get_app(with_config=False) @pytest.fixture def app_ssl(monkeypatch: pytest.MonkeyPatch) -> Generator: monkeypatch.setenv('EMAIL_URL', 'smtp://none:none@0.0.0.0:1025?ssl=True') yield from get_app(with_config=True) @pytest.fixture def app_tls(monkeypatch: pytest.MonkeyPatch) -> Generator: monkeypatch.setenv('EMAIL_URL', 'smtp://none:none@0.0.0.0:1025?tls=True') yield from get_app(with_config=True) @pytest.fixture def app_wo_domain() -> Generator: yield from get_app(with_config=True) @pytest.fixture() def app_config() -> AppConfig: config = AppConfig() config.gpx_limit_import = 10 config.max_single_file_size = 1048576 config.max_zip_file_size = 10485760 config.max_users = 0 db.session.add(config) db.session.commit() return config