From 445b2ba54a8f25b714e6dab2ce76af889478f1c0 Mon Sep 17 00:00:00 2001
From: Darius Jahandarie <djahandarie@gmail.com>
Date: Thu, 13 Nov 2025 13:34:39 +0900
Subject: [PATCH] Add support for Materialize

---
 macros/utils.sql | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/macros/utils.sql b/macros/utils.sql
index 2df31af..e8f2e86 100644
--- a/macros/utils.sql
+++ b/macros/utils.sql
@@ -1,4 +1,8 @@
 {% macro run_query(query) %}
+  {{ return(adapter.dispatch('run_query','dbt_unit_testing')(query)) }}
+{% endmacro %}
+
+{% macro default__run_query(query) %}
   {% set start_time = modules.datetime.datetime.now() %}
   {{ dbt_unit_testing.verbose('Running query => ' ~ dbt_unit_testing.sanitize(query)) }}
   {% set results = dbt.run_query(query) %}
@@ -8,6 +12,16 @@
   {{ return (results) }}
 {% endmacro %}
 
+{% macro materialize__run_query(query) %}
+  {% set cluster = target.get('cluster', none) %}
+  {% if cluster %}
+    {% call statement(auto_begin=True) %}
+      SET CLUSTER = {{ cluster }}
+    {% endcall %}
+  {% endif %}
+  {{ return(dbt_unit_testing.default__run_query(query)) }}
+{% endmacro %}
+
 {% macro sanitize(s) %}
   {{ return (" ".join(s.split())) }}
 {% endmacro %}