"""Test snowflake_check_ingest_sales queries."""

from dags.templates.sales_ingest.snowflake_check_ingest_sales import check_ingested_sales


def test_check_ingest_sales():
    """Test that template renders correctly."""
    actual_template = check_ingested_sales().render(
        env='test_env',
        sales_ingest_type='test_type',
        batch_id=123
    )

    expected_template = """
        SELECT
            COUNT(*) as count
        FROM
            ROYALTY_ACCOUNTING.test_env.STMT_DB_SALES_test_type_STAGING
        WHERE
            BATCH_ID = '123'
    """

    assert actual_template == expected_template