"""db.py.

Setup database testing structures for SQLlite which is used in testing env.

"""

import datetime

from blacklist_manager.connectors.mysql import ar_db_engine
from blacklist_manager.connectors.mysql import ar_db_session
from blacklist_manager.connectors.mysql import BaseModel
from blacklist_manager.models.blacklist_reasons import BlacklistReasons
from blacklist_manager.models.blacklist_words import BlacklistWords
from blacklist_manager.models.orchadmin_users import OrchadminUsers


def create_all():
    """Create all tables."""
    return BaseModel.metadata.create_all(ar_db_engine)


def drop_all():
    """Drop all tables."""
    return BaseModel.metadata.drop_all(ar_db_engine)


def insert_mock_records():
    """Insert mock_records."""
    with ar_db_session() as session:
        session.add(
            OrchadminUsers(orchadmin_users_id=532, login='test', password='test', f_name='test', l_name='user')
        )
        session.add(
            OrchadminUsers(orchadmin_users_id=6499, login='test2', password='test2', f_name='test2', l_name='user2')
        )
        session.commit()
        session.add(
            BlacklistReasons(
                blacklist_reason_id=3,
                reason='Valid Reason',
                added_by=532,
                alert='Alert of reason',
                contact='schauhan@theorchard.com',
                date_added=datetime.date(2018, 9, 2),
                date_updated=datetime.date(2018, 9, 2)
            )
        )
        session.add(
            BlacklistReasons(
                blacklist_reason_id=243323,
                reason='reasonable reason',
                added_by=6499,
                alert='abc alert',
                contact='asdf@theorchard.com',
                date_added=datetime.date(2019, 10, 1),
                date_updated=datetime.date(2019, 10, 1)
            )
        )
        session.commit()
        session.add(
            BlacklistWords(
                blacklist_word_id=1,
                word='Blacklist Word',
                notes='Blacklist Notes',
                blacklist_reason_id=3,
                vendor_id=123,
                added_by=532,
                date_added=datetime.date(2018, 9, 2),
                added_by_identity='a1b2c3d4-e5f6-7890-abcd-ef1234567890'
            )
        )
        session.add(
            BlacklistWords(
                blacklist_word_id=32,
                word='abc',
                notes='abc notes',
                blacklist_reason_id=243323,
                vendor_id=None,
                added_by=6499,
                date_added=datetime.date(2019, 10, 1)
            )
        )
        session.commit()