"""Utils for payment_holds db tests."""
from ows_accounting import config
from ows_accounting.models.vendor_payment_hold import VendorPaymentHold
from ows_accounting.models.vendor_payment_hold_log import VendorPaymentHoldLog
from ows_accounting.utils import mysql


session = mysql._holds_db_session()


def create_tables():
    """Create tables defined in models."""
    if config.ENVIRONMENT != config.TEST_ENVIRONMENT:
        return
    mysql.BaseModel.metadata.drop_all(mysql._holds_db_engine)
    mysql.BaseModel.metadata.create_all(mysql._holds_db_engine)


def seed_vendor_payment_hold(data):
    """Insert test data to vendor_payment_hold table.

    Args:
        data (list): multiple hold data.
    """
    if config.ENVIRONMENT != config.TEST_ENVIRONMENT:
        return
    session.execute(VendorPaymentHold.__table__.insert().values(data))
    session.commit()
    session.close()


def seed_vendor_payment_hold_log(data):
    """Insert test data to vendor_payment_hold_log table.

    Args:
        data (list): multiple hold data.
    """
    if config.ENVIRONMENT != config.TEST_ENVIRONMENT:
        return
    session.execute(VendorPaymentHoldLog.__table__.insert().values(data))
    session.commit()
    session.close()