"""Test utilities."""
from commit_royalty.template import get_formatted_query

CONTRACT_SALES_SQL = """
SELECT
    CONTRACT_ID,
    PAYEE_CURRENCY_CODE AS CURRENCY_CODE,
    SUM(NET_REVENUE_PAYEE_CURRENCY) AS AMOUNT
FROM $db.$schema.ACCOUNTING_RUN_RESULTS
WHERE ACCOUNTING_RUN_ID = $accounting_run_id
AND PAYEE_ID = $payee_id
GROUP BY
    CONTRACT_ID,
    PAYEE_CURRENCY_CODE
"""


def test_get_formatted_query(mock_event):
    """Test get_formatted_query."""
    result = get_formatted_query(CONTRACT_SALES_SQL, {
        'db': 'ROYALTY_ACCOUNTING',
        'schema': 'dev',
        'accounting_run_id': '1337',
        'payee_id': '1'
    })
    assert result == """
SELECT
    CONTRACT_ID,
    PAYEE_CURRENCY_CODE AS CURRENCY_CODE,
    SUM(NET_REVENUE_PAYEE_CURRENCY) AS AMOUNT
FROM ROYALTY_ACCOUNTING.dev.ACCOUNTING_RUN_RESULTS
WHERE ACCOUNTING_RUN_ID = 1337
AND PAYEE_ID = 1
GROUP BY
    CONTRACT_ID,
    PAYEE_CURRENCY_CODE
"""