"""Test snowflake_export_sales queries."""

from dags.templates.sales_approve import snowflake_export_sales


def test_export_approved_distro_sales_to_s3():
    """Test that the template renders the export statement correctly."""
    template = snowflake_export_sales.export_approved_distro_sales_to_s3().render(
        env='test',
        s3_path='1-acct-period/eligible-sales/2-sales/results/',
        sales_file_id=2,
        stage='ABACUS_PARQUET_FILE'
    )
    assert template == """
        COPY INTO @ABACUS_PARQUET_FILE/1-acct-period/eligible-sales/2-sales/results/
        FROM (
            SELECT
                CAST(txn.STMT_DB_SALES_DISTRO_TXN_ID AS BIGINT) AS txnId,
                TO_DECIMAL(txn.ACTIVITY_RATE, 7, 6) AS activityRate,
                CAST(txn.COUNTRY_ID AS INT) AS countryId,
                txn.ISRC AS isrc,
                CAST(txn.LABEL_ID AS INT) AS labelId,
                CAST(txn.QUANTITY AS INT) AS quantity,
                CAST(txn.SALE_CURRENCY_CODE_ID AS INT) AS saleCurrencyId,
                CAST(txn.STORE_ID AS INT) AS storeId,
                txn.TRANSACTION_DATE AS transactionDate,
                CAST(txn.TRANSACTION_TYPE_ID AS INT) AS transactionTypeId,
                TO_DECIMAL(txn.UNIT_PRICE_USD, 18, 6) AS unitPriceUSD,
                CAST(txn.UPC AS BIGINT) AS upc,
                TO_DECIMAL(txn.WITHHOLDING_TAX_USD, 18, 6) AS withholdingTaxUSD
            FROM ROYALTY_ACCOUNTING.test.STMT_DB_SALES_DISTRO AS txn
            WHERE txn.SALES_FILE_ID = 2
        )
        FILE_FORMAT = (TYPE = PARQUET)
        HEADER = TRUE
    """


def test_export_approved_nr_sales_to_s3():
    """Test that the template renders the export statement correctly."""
    template = snowflake_export_sales.export_approved_nr_sales_to_s3().render(
        env='test',
        s3_path='1-acct-period/eligible-sales/2-sales/results/',
        sales_file_id=2,
        stage='ABACUS_PARQUET_FILE'
    )
    assert template == """
        COPY INTO @ABACUS_PARQUET_FILE/1-acct-period/eligible-sales/2-sales/results/
        FROM (
            SELECT
                CAST(txn.STMT_DB_SALES_NR_TXN_ID AS BIGINT) AS txnId,
                TO_DECIMAL(txn.ACTIVITY_RATE, 7, 6) AS activityRate,
                CAST(txn.COUNTRY_ID AS INT) AS countryId,
                txn.ISRC AS isrc,
                txn.SOUND_RECORDING_ID AS soundRecordingId,
                txn.CONTRIBUTOR_ID AS contributorId,
                CAST(txn.QUANTITY AS INT) AS quantity,
                CAST(txn.SALE_CURRENCY_CODE_ID AS INT) AS saleCurrencyId,
                CAST(txn.STORE_ID AS INT) AS storeId,
                txn.START_DATE AS startDate,
                txn.END_DATE AS endDate,
                CAST(txn.TRANSACTION_TYPE_ID AS INT) AS transactionTypeId,
                TO_DECIMAL(txn.UNIT_PRICE_USD, 18, 6) AS unitPriceUSD,
                txn.CONTRIBUTION_ID AS contributionId,
                TO_DECIMAL(txn.WITHHOLDING_TAX_USD, 18, 6) AS withholdingTaxUSD
            FROM ROYALTY_ACCOUNTING.test.STMT_DB_SALES_NR AS txn
            WHERE txn.SALES_FILE_ID = 2
        )
        FILE_FORMAT = (TYPE = PARQUET)
        HEADER = TRUE
    """