"""Test snowflake_copy_results queries."""

from dags.templates.accounting_run_commit import snowflake_copy_contract_transactions


def test_copy_contract_transaction_distro_template():
    """Test that the template renders the insert statement correctly."""
    template = snowflake_copy_contract_transactions.copy_contract_transaction_distro_template().render(
        environment='test',
        accounting_run_id=2
    )
    assert template == """
        INSERT INTO ROYALTY_ACCOUNTING.test.CONTRACT_TRANSACTION_DISTRO
        (
            CONTRACT_TXN_ID,
            TXN_ID,
            ACCOUNTING_RUN_ID,
            CONTRACT_ID,
            ACCOUNT_ID,
            PAYEE_CURRENCY_ID,
            CONTRACT_TERM_ID,
            CONTRACT_TERM_CONDITION_ID
        )
        SELECT
            CONTRACT_TXN_ID,
            TXN_ID,
            ACCOUNTING_RUN_ID,
            CONTRACT_ID,
            ACCOUNT_ID,
            PAYEE_CURRENCY_ID,
            CONTRACT_TERM_ID,
            CONTRACT_TERM_CONDITION_ID
        FROM ROYALTY_ACCOUNTING.test.CONTRACT_TRANSACTION_STAGING AS source
        WHERE source.ACCOUNTING_RUN_ID = 2
    """


def test_copy_contract_transaction_nr_template():
    """Test that the template renders the insert statement correctly."""
    template = snowflake_copy_contract_transactions.copy_contract_transaction_nr_template().render(
        environment='test',
        accounting_run_id=2
    )
    assert template == """
        INSERT INTO ROYALTY_ACCOUNTING.test.CONTRACT_TRANSACTION_NR
        (
            CONTRACT_TXN_ID,
            TXN_ID,
            ACCOUNTING_RUN_ID,
            CONTRACT_ID,
            ACCOUNT_ID,
            PAYEE_CURRENCY_ID,
            CONTRACT_TERM_ID,
            CONTRACT_TERM_CONDITION_ID,
            SCHEDULE_ID,
            TERM_RATE
        )
        SELECT
            CONTRACT_TXN_ID,
            TXN_ID,
            ACCOUNTING_RUN_ID,
            CONTRACT_ID,
            ACCOUNT_ID,
            PAYEE_CURRENCY_ID,
            CONTRACT_TERM_ID,
            CONTRACT_TERM_CONDITION_ID,
            SCHEDULE_ID,
            TERM_RATE
        FROM ROYALTY_ACCOUNTING.test.CONTRACT_TRANSACTION_NR_STAGING AS source
        WHERE source.ACCOUNTING_RUN_ID = 2
    """