Add __dbt_tmp suffix to specified location for temporary tables

Author: yakovlevvs

.changes/unreleased/Fixes-20250220-121852.yaml

@@ -0,0 +1,7 @@
+kind: Fixes
+body: Add __dbt_tmp suffix to specified location for temporary tables
+time: 2025-02-20T12:18:52.200467772+03:00
+custom:
+    Author: yakovlevvs
+    Issue: "467"
+    PR: "468"

dbt/include/trino/macros/adapters.sql

@@ -82,7 +82,7 @@
   {{ return(sql) }}
 {% endmacro %}
 
-{% macro properties() %}
+{% macro properties(temporary=False) %}
   {%- set _properties = config.get('properties') -%}
   {%- set table_format = config.get('table_format') -%}
   {%- set file_format = config.get('file_format') -%}
@@ -117,6 +117,14 @@
     {%- endif -%}
   {%- endif -%}
 
+  {%- if temporary -%}
+    {%- if _properties -%}
+      {%- if _properties.location -%}
+          {%- do _properties.update({'location': _properties.location[:-1] ~ "__dbt_tmp'"}) -%}
+      {%- endif -%}
+    {%- endif -%}
+  {%- endif -%}
+
   {%- if _properties is not none -%}
       WITH (
           {%- for key, value in _properties.items() -%}
@@ -154,7 +162,7 @@
     {{ get_assert_columns_equivalent(sql) }}
     {%- set sql = get_select_subquery(sql) %}
     {{ comment(model.get('description')) }}
-    {{ properties() }}
+    {{ properties(temporary) }}
   ;
 
   insert into {{ relation }}
@@ -167,7 +175,7 @@
 
     create{{ or_replace }} table {{ relation }}
       {{ comment(model.get('description')) }}
-      {{ properties() }}
+      {{ properties(temporary) }}
     as (
       {{ sql }}
     );

tests/functional/adapter/materialization/test_incremental_delete_insert.py

@@ -13,6 +13,7 @@
     models__unary_unique_key_list_sql,
     seeds__seed_csv,
 )
+from dbt.tests.util import run_dbt_and_capture
 
 seeds__duplicate_insert_sql = """
 -- Insert statement which when applied to seed.csv triggers the inplace
@@ -93,6 +94,36 @@
 
 """
 
+models__location_specified = """
+{{
+    config(
+        materialized='incremental',
+        incremental_strategy='delete+insert',
+        unique_key=['state', 'county', 'city'],
+        properties= {
+            "location": "'s3a://datalake/model'"
+        }
+    )
+}}
+
+select
+    'CT' as state,
+    'Hartford' as county,
+    'Hartford' as city,
+    cast('2022-02-14' as date) as last_visit_date
+union all
+select 'MA','Suffolk','Boston',DATE '2020-02-12'
+union all
+select 'NJ','Mercer','Trenton',DATE '2022-01-01'
+union all
+select 'NY','Kings','Brooklyn',DATE '2021-04-02'
+union all
+select 'NY','New York','Manhattan',DATE '2021-04-01'
+union all
+select 'PA','Philadelphia','Philadelphia',DATE '2021-05-21'
+
+"""
+
 
 class TrinoIncrementalUniqueKey(BaseIncrementalUniqueKey):
     @pytest.fixture(scope="class")
@@ -146,3 +177,30 @@ def project_config_update(self):
             "models": {"+on_table_exists": "drop", "+incremental_strategy": "delete+insert"},
             "seeds": {"incremental": {"seed": {"+column_types": {"some_date": "date"}}}},
         }
+
+
+@pytest.mark.iceberg
+@pytest.mark.skip_profile("starburst_galaxy")
+class TestIcebergIncrementalDeleteInsertWithLocation:
+    @pytest.fixture(scope="class")
+    def models(self):
+        return {
+            "model.sql": models__location_specified,
+        }
+
+    def test_temporary_table_location(self, project):
+        # Create model with properties
+        results, logs = run_dbt_and_capture(["--debug", "run"], expect_pass=True)
+        assert len(results) == 1
+        assert f'create table "{project.database}"."{project.test_schema}"."model"' in logs
+        assert "location = 's3a://datalake/model'" in logs
+
+        # Temporary table is created on the second run
+        # So, now we check if the second run is successful and location
+        # is patched correctly
+        results, logs = run_dbt_and_capture(["--debug", "run"], expect_pass=True)
+        assert len(results) == 1
+        assert (
+            f'create table "{project.database}"."{project.test_schema}"."model__dbt_tmp"' in logs
+        )
+        assert "location = 's3a://datalake/model__dbt_tmp'" in logs