extending and using dbt_unit_testing's quote_identifier macro

EqualExperts · Aug 4, 2023 · 6d76b45 · 6d76b45
1 parent 279e6ce
commit 6d76b45
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/macros/mock_builders.sql b/macros/mock_builders.sql
@@ -69,7 +69,7 @@
         left join (select 
         {%- for column in missing_columns %}
           {%- if options.quoted_columns %}
-            "{{ column }}"
+            {{ dbt_unit_testing.quote_identifier(column, true) }}
           {%- else %}
             {{ column }}
           {%- endif %}

diff --git a/macros/utils.sql b/macros/utils.sql
@@ -124,29 +124,31 @@
   {{ return (dbt_unit_testing.merge_jsons([unit_tests_config] + configs)) }}
 {% endmacro %}
 
-{% macro quote_identifier(identifier) %}
-    {{ return(adapter.dispatch('quote_identifier','dbt_unit_testing')(identifier)) }}
+{% macro quote_identifier(identifier, preserve_case=false) %}
+    {{ return(adapter.dispatch('quote_identifier','dbt_unit_testing')(identifier, preserve_case)) }}
 {% endmacro %}
 
-{% macro default__quote_identifier(identifier) -%}
+{% macro default__quote_identifier(identifier, preserve_case) -%}
     {% if identifier.startswith('"') %}
       {{ return(identifier) }}
     {% else %}
       {{ return('"' ~ identifier ~ '"') }}
     {% endif %}
 {%- endmacro %}
 
-{% macro bigquery__quote_identifier(identifier) %}
+{% macro bigquery__quote_identifier(identifier, preserve_case) %}
     {% if identifier.startswith('`') %}
       {{ return(identifier) }}
     {% else %}
       {{ return('`' ~ identifier ~ '`') }}
     {% endif %}
 {% endmacro %}
 
-{% macro snowflake__quote_identifier(identifier) %}
+{% macro snowflake__quote_identifier(identifier, preserve_case) %}
     {% if identifier.startswith('"') %}
       {{ return(identifier) }}
+    {% elif preserve_case %}
+      {{ return('"' ~ identifier ~ '"') }}
     {% else %}
       {{ return('"' ~ identifier | upper ~ '"') }}
     {% endif %}