"""Test snowflake_copy_mech_deductions queries."""

from dags.templates.accounting_run_commit import snowflake_copy_mech_deductions


def test_copy_mech_deductions_template():
    """Test that the template renders the insert statement correctly."""
    template = snowflake_copy_mech_deductions.copy_mech_deductions_template().render(
        environment='TEST',
        accounting_run_id=2
    )
    assert template == """
        INSERT INTO ROYALTY_ACCOUNTING.TEST.ACCOUNTING_RUN_RESULTS_DISTRO_MECH
        (
            CONTRACT_TXN_ID,
            MECHANICAL_TRANSACTION_ID,
            CONTRACT_MECHANICAL_DEDUCTION_ID,
            TERRITORY,
            MECHANICAL_DEDUCTION_ADMIN_TYPE,
            MECHANICAL_DEDUCTION_RATE,
            PUBLISHER_ADMIN_FEE_RATE,
            MECHANICAL_ROYALTY_AMOUNT_SALE_CURRENCY,
            GROSS_REVENUE_AFTER_WITHHOLDING_TAX_SALE_CURRENCY,
            NET_REVENUE_SALE_CURRENCY,
            MECHANICAL_DEDUCTION_AMOUNT_SALE_CURRENCY,
            PUBLISHER_ADMIN_FEE_SALE_CURRENCY,
            NET_REVENUE_AFTER_MECHANICAL_SALE_CURRENCY,
            MECHANICAL_ROYALTY_AMOUNT_PAYEE_CURRENCY,
            GROSS_REVENUE_AFTER_WITHHOLDING_TAX_PAYEE_CURRENCY,
            NET_REVENUE_PAYEE_CURRENCY,
            MECHANICAL_DEDUCTION_AMOUNT_PAYEE_CURRENCY,
            PUBLISHER_ADMIN_FEE_PAYEE_CURRENCY,
            NET_REVENUE_AFTER_MECHANICAL_PAYEE_CURRENCY
        )
        SELECT
            source.CONTRACT_TXN_ID,
            MECHANICAL_TRANSACTION_ID,
            CONTRACT_MECHANICAL_DEDUCTION_ID,
            TERRITORY,
            MECHANICAL_DEDUCTION_ADMIN_TYPE,
            MECHANICAL_DEDUCTION_RATE,
            PUBLISHER_ADMIN_FEE_RATE,
            MECHANICAL_ROYALTY_AMOUNT_SALE_CURRENCY,
            GROSS_REVENUE_AFTER_WITHHOLDING_TAX_SALE_CURRENCY,
            NET_REVENUE_SALE_CURRENCY,
            MECHANICAL_DEDUCTION_AMOUNT_SALE_CURRENCY,
            PUBLISHER_ADMIN_FEE_SALE_CURRENCY,
            NET_REVENUE_AFTER_MECHANICAL_SALE_CURRENCY,
            MECHANICAL_ROYALTY_AMOUNT_PAYEE_CURRENCY,
            GROSS_REVENUE_AFTER_WITHHOLDING_TAX_PAYEE_CURRENCY,
            NET_REVENUE_PAYEE_CURRENCY,
            MECHANICAL_DEDUCTION_AMOUNT_PAYEE_CURRENCY,
            PUBLISHER_ADMIN_FEE_PAYEE_CURRENCY,
            NET_REVENUE_AFTER_MECHANICAL_PAYEE_CURRENCY
        FROM ROYALTY_ACCOUNTING.TEST.ACCOUNTING_RUN_RESULTS_DISTRO_MECH_STAGING AS source
            INNER JOIN  ROYALTY_ACCOUNTING.TEST.CONTRACT_TRANSACTION_DISTRO as ct
                    ON source.CONTRACT_TXN_ID = ct.CONTRACT_TXN_ID
        WHERE ct.ACCOUNTING_RUN_ID = 2
        """