# standard imports
import os

# third-party imports
import pytest
import alembic
from alembic.config import Config as AlembicConfig

# local imports
from cic_notify.db import SessionBase
from cic_notify.db import dsn_from_config


@pytest.fixture(scope='session')
def database_engine(
    load_config,
        ):
    dsn = dsn_from_config(load_config)
    SessionBase.connect(dsn)
    return dsn


@pytest.fixture(scope='function')
def init_database(
        load_config,
        database_engine,
        ):

    rootdir = os.path.dirname(os.path.dirname(__file__))
    dbdir = os.path.join(rootdir, 'cic_notify', 'db')
    migrationsdir = os.path.join(dbdir, 'migrations', load_config.get('DATABASE_ENGINE'))
    if not os.path.isdir(migrationsdir):
        migrationsdir = os.path.join(dbdir, 'migrations', 'default')

    session = SessionBase.create_session()

    ac = AlembicConfig(os.path.join(migrationsdir, 'alembic.ini'))
    ac.set_main_option('sqlalchemy.url', database_engine)
    ac.set_main_option('script_location', migrationsdir)

    alembic.command.downgrade(ac, 'base')
    alembic.command.upgrade(ac, 'head')

    yield session
    session.commit()
    session.close()