Fix column transformations for Snowflake

Author: psousa50

integration-tests/dbt_project.yml

@@ -24,7 +24,7 @@ vars:
   unit_tests_config:
     column_transformations:
       model_for_column_transformations:
-        b: upper(b)
+        column_a: round(##column##, 5)
       model_references_model_with_structs:
         a: to_json_string(##column##)
     verbose: false

integration-tests/models/model_for_column_transformations.sql

@@ -1 +1 @@
-select a / 3 as a, b from {{ dbt_unit_testing.ref ('model_a') }}
+select a / 3 as column_a, b as column_b from {{ dbt_unit_testing.ref ('model_a') }}

integration-tests/tests/unit/model_for_column_transformations.sql

@@ -1,11 +1,12 @@
 {{
     config(
-        tags=['unit-test', 'bigquery', 'snowflake', 'postgres']
+        tags=['unit-test', 'bigquery', 'postgres']
     )
 }}
 
 {% set column_transformations = {
-  "a": "round(##column##, 4)"
+  "column_a": "round(##column##, 4)",
+  "column_b": "upper(##column##)"
   }
 %}
 
@@ -14,6 +15,22 @@
     select 10.0 as a, 'lower' as b
   {% endcall %}
   {% call dbt_unit_testing.expect() %}
-    select 3.3333 as a, 'LOWER' as b
+    select 3.3333 as column_a, 'LOWER' as column_b
+  {% endcall %}
+{% endcall %}
+
+UNION ALL
+
+{% set column_transformations = {
+  "column_b": "upper(##column##)"
+  }
+%}
+
+{% call dbt_unit_testing.test('model_for_column_transformations', 'should merge config from dbt_project', options={"column_transformations": column_transformations}) %}
+  {% call dbt_unit_testing.mock_ref ('model_a') %}
+    select 10.0 as a, 'lower' as b
+  {% endcall %}
+  {% call dbt_unit_testing.expect() %}
+    select 3.33333 as column_a, 'LOWER' as column_b
   {% endcall %}
 {% endcall %}

integration-tests/tests/unit/snowflake/model_for_column_transformations_sf.sql

@@ -0,0 +1,20 @@
+{{
+    config(
+        tags=['unit-test', 'snowflake']
+    )
+}}
+
+{% set column_transformations = {
+  "COLUMN_A": "round(##column##, 4)",
+  "COLUMN_B": "upper(##column##)"
+  }
+%}
+
+{% call dbt_unit_testing.test('model_for_column_transformations', options={"column_transformations": column_transformations}) %}
+  {% call dbt_unit_testing.mock_ref ('model_a') %}
+    select 10.0 as a, 'lower' as b
+  {% endcall %}
+  {% call dbt_unit_testing.expect() %}
+    select 3.3333 as COLUMN_A, 'LOWER' as column_b
+  {% endcall %}
+{% endcall %}

run-tests-helper.sh

@@ -7,9 +7,9 @@ function run_tests() {
   if [ "$PROFILE" == "postgres" ]; then
     VERSIONS="1.3.4 1.4.6 1.5.4 1.7.4"
   elif [ "$PROFILE" == "bigquery" ]; then
-    VERSIONS="1.3.2"
+    VERSIONS="1.7.2"
   elif [ "$PROFILE" == "snowflake" ]; then
-    VERSIONS="1.3.1"
+    VERSIONS="1.7.1"
   else
     echo "Invalid profile name: $PROFILE"
     exit 1