diff options
Diffstat (limited to 'test_utils/spectrum/conftest.py')
| -rw-r--r-- | test_utils/spectrum/conftest.py | 18 |
1 files changed, 12 insertions, 6 deletions
diff --git a/test_utils/spectrum/conftest.py b/test_utils/spectrum/conftest.py index b7887f6..d737730 100644 --- a/test_utils/spectrum/conftest.py +++ b/test_utils/spectrum/conftest.py @@ -1,24 +1,30 @@ import logging - import time +from datetime import datetime, timezone +from typing import TYPE_CHECKING import pytest -from datetime import datetime, timezone + from generalresearch.managers.spectrum.survey import ( - SpectrumSurveyManager, SpectrumCriteriaManager, + SpectrumSurveyManager, ) from generalresearch.models.spectrum.survey import SpectrumSurvey from generalresearch.sql_helper import SqlHelper -from .surveys_json import SURVEYS_JSON, CONDITIONS +from .surveys_json import CONDITIONS, SURVEYS_JSON + +if TYPE_CHECKING: + from generalresearch.config import GRLBaseSettings @pytest.fixture(scope="session") -def spectrum_rw(settings) -> SqlHelper: - print(f"{settings.spectrum_rw_db=}") +def spectrum_rw(settings: "GRLBaseSettings") -> SqlHelper: logging.info(f"{settings.spectrum_rw_db=}") + + assert settings.spectrum_rw_db is not None assert "/unittest-" in settings.spectrum_rw_db.path + return SqlHelper( dsn=settings.spectrum_rw_db, read_timeout=2, |
