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

CONTRACT_ADJUSTMENTS_SQL = """
SELECT
    CONTRACT_ID,
    CURRENCY_CODE,
    SUM(CURRENCY_AMOUNT) AS AMOUNT
FROM $db.$schema.ACCOUNTING_RUN_ADJUSTMENTS
WHERE ACCOUNTING_RUN_ID = $accounting_run_id
AND PAYEE_ID = $payee_id
GROUP BY
    CONTRACT_ID,
    CURRENCY_CODE
"""


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