"""Test update_sales_file_metadata queries."""

from dags.templates.sales_get_eligible import update_sales_file_metadata


def test_get_distro_sales_file_metadata():
    """Test that template renders query to get sales metadata correctly."""
    template = update_sales_file_metadata.get_distro_sales_file_metadata().render(
        env='test',
        sales_file_id=123
    )

    assert template == """
        SELECT
            SUM(UNIT_PRICE_USD * QUANTITY) AS amount_usd,
            COUNT(*) AS row_count
        FROM ROYALTY_ACCOUNTING.test.STMT_DB_SALES_DISTRO_TEMP
        WHERE SALES_FILE_ID = 123;
    """


def test_get_nr_sales_file_metadata():
    """Test that template renders query to get NR sales metadata correctly."""
    template = update_sales_file_metadata.get_nr_sales_file_metadata().render(
        env='test',
        sales_file_id=123
    )

    assert template == """
        SELECT
            SUM(ZEROIFNULL(UNIT_PRICE_USD) * ZEROIFNULL(QUANTITY)) AS amount_usd,
            COUNT(*) AS row_count
        FROM ROYALTY_ACCOUNTING.test.STMT_DB_SALES_NR_TEMP
        WHERE SALES_FILE_ID = 123;
    """