import logging
import os
from datetime import datetime, timedelta, timezone
from time import sleep

import pytest
from db_schema.postgres import connection
from db_schema.postgres.connection import Connection
from db_schema.schemas import slz
from freezegun import freeze_time
from smelog.entities import LoggerConfig
from smelog.factory import LoggerFactory

from slz_uow_reset import Config


@pytest.fixture
def test_config():
    return Config(
        environment='test',
        rds_secrets_key='pg conn details',
    )


@pytest.fixture
def db() -> Connection:
    def creds_loader():
        return dict(
            active_endpoint=os.environ.get('PG_HOST', '0.0.0.0'),
            port=os.environ.get('PG_PORT', 5432),
            database=os.environ.get('PG_DB', 'slz'),
            username=os.environ.get('PG_USER', 'admin'),
            password=os.environ.get('PG_PASSWORD', 'admin'),
        )

    conn = connection.Connection(
        credentials_loader=creds_loader, name='s3_storage_test', version='v1'
    )
    # Simple hack for waiting migrations to be completed
    while True:
        result = conn.session.execute(  # pylint: disable=no-member
            'SELECT id FROM databasechangelog ORDER BY dateexecuted DESC LIMIT 1'
        )
        if result.first():
            break
        sleep(1)

    yield conn

    conn.session.rollback()
    for model in [slz.UnitOfWorkPipeline, slz.UnitOfWorkGroup, slz.ContentStatus, slz.UnitOfWork]:
        conn.session.query(model).delete()
        conn.session.commit()
    conn.disconnect()


@pytest.fixture(scope='function')
def db_licensors(db):
    return {item.licensor_name: item for item in db.session.query(slz.Licensor).all()}


@pytest.fixture(scope='function')
def db_reports(db):
    return {item.report_name: item for item in db.session.query(slz.Report).all()}


@pytest.fixture(scope='function')
def db_data_sources(db):
    return {item.data_source_name: item for item in db.session.query(slz.DataSource).all()}


@pytest.fixture(scope='function')
def logger_test():
    log_config = LoggerConfig(
        name='TEST',
        version='1',
        level=logging.DEBUG,
        environment='test',
        is_local=True,
    )
    return LoggerFactory(log_config).get_logger('TEST')


@pytest.fixture(scope='function')
def now():
    with freeze_time(datetime(2021, 11, 24, 3, 21, 34, tzinfo=timezone.utc)):
        yield datetime.now().astimezone(timezone.utc)


@pytest.fixture(scope='function')
def unit_of_work_stubs(db_licensors, now, db_reports):
    return {
        1:
            dict(
                unit_of_work_code='apple-20191124-theorchard-amEvent-v1_2',
                reprocess_id='1',
                licensor=db_licensors['theorchard'],
                report=db_reports['amEvent'],
                report_date='2019-11-24',
                version='v1_2',
                activity_status=slz.ActivityStatusEnum.IN_PROGRESS.value,
                completeness_status=slz.CompletenessStatusEnum.MIN_COMPLETE.value,
                is_force_complete=False,
                priority=slz.UnitOfWorkPriorityEnum.DEFAULT.value,
                next_run_at=now + timedelta(minutes=15),
                created_at=now,
                last_updated_at=now + timedelta(minutes=1),
            ),
    }