"""Common operations for direct_delivery model testing."""

from assets.connectors import mysql
from assets.models.legacy import asset
from assets.models.legacy import asset_location
from assets.models.legacy import asset_location_detail
from assets.models.legacy import \
    asset_received_destination_details as destination_details
from assets.models.legacy import asset_received_queue
from assets.models.legacy import asset_type
from assets.models.legacy import storage
from assets.models.legacy import storage_drive
from assets.models.legacy import storage_drive_asset_folder
from tests.testutils import db


session = mysql.dd_database_session()


def create_tables():
    """Create table defined in asset_received_queue model."""
    db._exit_if_not_test_environment(session)

    mysql.DdModel.metadata.drop_all(mysql.dd_db_engine)
    mysql.DdModel.metadata.create_all(mysql.dd_db_engine)


def seed_asset_table(data):
    """Seed asset table."""
    db._exit_if_not_test_environment(session)

    session.execute(asset.Asset.__table__.insert().values(data))
    session.commit()
    session.close()


def seed_asset_location_table(data):
    """Seed asset location table."""
    db._exit_if_not_test_environment(session)

    session.execute(
        asset_location.AssetLocation.__table__.insert().values(data))
    session.commit()
    session.close()


def seed_asset_location_detail_table(data):
    """Seed asset location detail table."""
    db._exit_if_not_test_environment(session)

    session.execute(
        asset_location_detail.AssetLocationDetail.__table__.insert().values(
            data))
    session.commit()
    session.close()


def seed_asset_received_queue_table(data):
    """Seed asset_received_queue table."""
    db._exit_if_not_test_environment(session)

    session.execute(
        asset_received_queue.AssetReceivedQueue.__table__.insert().values(
            data))
    session.commit()
    session.close()


def seed_asset_received_destination_details_table(data):
    """Seed asset_received_destination_details table."""
    db._exit_if_not_test_environment(session)

    session.execute(
        destination_details.AssetReceivedDestinationDetails.__table__.insert(
            ).values(data))
    session.commit()
    session.close()


def seed_asset_type_table(data):
    """Seed asset_type table."""
    db._exit_if_not_test_environment(session)

    session.execute(
        asset_type.AssetType.__table__.insert(
            ).values(data))
    session.commit()
    session.close()


def seed_storage_table(data):
    """Seed storage table."""
    db._exit_if_not_test_environment(session)

    session.execute(
        storage.Storage.__table__.insert().values(data))
    session.commit()
    session.close()


def seed_storage_drive_table(data):
    """Seed storage drive table."""
    db._exit_if_not_test_environment(session)

    session.execute(
        storage_drive.StorageDrive.__table__.insert().values(data))
    session.commit()
    session.close()


def seed_storage_drive_asset_folder_table(data):
    """Seed storage drive asset folder table."""
    db._exit_if_not_test_environment(session)

    session.execute(
        storage_drive_asset_folder.StorageDriveAssetFolder.__table__.insert(
            ).values(data))
    session.commit()
    session.close()