"""Common operations for art_relations model testing."""

from assets.connectors import mysql
from assets.models.legacy import import_asset
from assets.models.legacy import import_asset_detail
from assets.models.legacy import trackdown_media_access
from tests.testutils import db

session = mysql.ar_database_session()


def create_tables():
    """Create table defined in art_relations models."""
    db._exit_if_not_test_environment(session)

    mysql.ArModel.metadata.drop_all(mysql.ar_db_engine)
    mysql.ArModel.metadata.create_all(mysql.ar_db_engine)


def seed_import_asset_table(data):
    """Seed asset_upload table."""
    db._exit_if_not_test_environment(session)

    session.execute(
        import_asset.ImportAsset.__table__.insert().values(
            data))
    session.commit()
    session.close()


def seed_trackdown_media_access(data):
    """Seed trackdown_media_access table."""
    db._exit_if_not_test_environment(session)

    session.execute(
        trackdown_media_access.TrackdownMediaAccess.__table__.insert().values(
            data))
    session.commit()
    session.close()


def seed_import_asset_detail_table(data):
    """Seed import_asset_detail table."""
    db._exit_if_not_test_environment(session)

    session.execute(
        import_asset_detail.ImportAssetDetail.__table__.insert().values(data))
    session.commit()
    session.close()