34 lines
985 B
Python
34 lines
985 B
Python
|
import logging
|
||
|
|
import os
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from alembic import config
|
||
|
|
from pytest_alembic.config import Config
|
||
|
|
from streamlit.testing.v1 import AppTest
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
@pytest.fixture(autouse=True)
|
||
|
|
def setup_database(alembic_runner):
|
||
|
|
logger.info("Running database migrations")
|
||
|
|
alembic_runner.migrate_up_to('heads')
|
||
|
|
yield
|
||
|
|
logger.info("Resetting database")
|
||
|
|
alembic_runner.migrate_down_to('base')
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def alembic_config() -> Config:
|
||
|
|
logging.info("Setting up alembic config")
|
||
|
|
alembic_cfg = config.Config(toml_file="pyproject.toml")
|
||
|
|
alembic_cfg.set_main_option("sqlalchemy.url", os.getenv("DATABASE_URL", ""))
|
||
|
|
return Config(alembic_config=alembic_cfg)
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def app() -> AppTest:
|
||
|
|
return AppTest.from_file("app/streamlit_app.py")
|
||
|
|
|
||
|
|
def delete_database():
|
||
|
|
file = os.getenv("DATABASE_FILE")
|
||
|
|
if file and os.path.isfile(file):
|
||
|
|
logger.info(f"Deleting database file {file}")
|
||
|
|
os.remove(file)
|