"""Test template module.""" from reserves_take.constants.snowflake import GET_TRANSACTIONS_SQL from reserves_take.template import get_formatted_query def test_get_formatted_query(): """Test get_formatted_query.""" result = get_formatted_query(GET_TRANSACTIONS_SQL, { 'db': 'ROYALTY_ACCOUNTING', 'schema': 'test', 'accounting_run_id': '35' }) assert result == """ SELECT ctd.CONTRACT_ID, ctd.ACCOUNT_ID, cmp.ISO_CODE AS CURRENCY_CODE, SUM( IFF(txn.TRANSACTION_TYPE in ('PS', 'TP'), arr.GROSS_REVENUE_PAYEE_CURRENCY, 0) ) AS PHYSICAL_SALES_ACCOUNT_CURRENCY, SUM( IFF(txn.TRANSACTION_TYPE in ('RE', 'TR'), arr.GROSS_REVENUE_PAYEE_CURRENCY, 0) ) AS PHYSICAL_RETURNS_ACCOUNT_CURRENCY FROM ROYALTY_ACCOUNTING.test.ACCOUNTING_RUN_RESULTS_DISTRO AS arr INNER JOIN ROYALTY_ACCOUNTING.test.STMT_DB_SALES_DISTRO AS txn ON arr.txn_id = txn.stmt_db_sales_distro_txn_id INNER JOIN ROYALTY_ACCOUNTING.test.CONTRACT_TRANSACTION_DISTRO AS ctd ON arr.txn_id = ctd.txn_id AND arr.accounting_run_id = ctd.accounting_run_id INNER JOIN ROYALTY_ACCOUNTING.test.CURRENCY_MAP AS cmp ON ctd.PAYEE_CURRENCY_ID = cmp.ISO_NUMBER WHERE arr.ACCOUNTING_RUN_ID = 35 AND txn.TRANSACTION_TYPE IN ('RE','PS', 'TP', 'TR') GROUP BY ctd.CONTRACT_ID, ctd.ACCOUNT_ID, CURRENCY_CODE HAVING ABS(PHYSICAL_SALES_ACCOUNT_CURRENCY) > ABS(PHYSICAL_RETURNS_ACCOUNT_CURRENCY); """