"""Common operations for asset_upload model testing."""

from assets.connectors import mysql
from assets.models import asset_status
from assets.models import asset_upload
from tests.testutils import db


session = mysql.au_database_session()


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

    mysql.AuModel.metadata.drop_all(mysql.au_db_engine)
    mysql.AuModel.metadata.create_all(mysql.au_db_engine)


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

    session.execute(
        asset_upload.AssetUpload.__table__.insert().values(
            data))
    session.commit()
    session.close()


def seed_asset_status_table(data):
    """Seed asset_status table."""
    db._exit_if_not_test_environment(session)

    session.execute(
        asset_status.AssetStatus.__table__.insert().values(data))
    session.commit()
    session.close()