"""Test snowflake delete/truncate statements.""" from dags.templates.accounting_period_sales_approve import snowflake_truncate_temp_sales def test_truncate_temp_sales(): """Test that template renders truncate statement correctly.""" template = snowflake_truncate_temp_sales.truncate_temp_sales().render( env='test', temp_table='STMT_DB_SALES_DISTRO_TEMP' ) assert template == """TRUNCATE ROYALTY_ACCOUNTING.test.STMT_DB_SALES_DISTRO_TEMP;"""