|
4 | 4 |
|
5 | 5 | {% set expected_rows = config.get('expected_rows') %} |
6 | 6 | {% set expected_sql = config.get('expected_sql') %} |
7 | | - {% set tested_expected_column_names = expected_rows[0].keys() if (expected_rows | length ) > 0 else get_columns_in_query(sql) %} %} |
| 7 | + {% set tested_expected_column_names = expected_rows[0].keys() if (expected_rows | length ) > 0 else get_columns_in_query(sql) %} |
8 | 8 |
|
9 | 9 | {%- set target_relation = this.incorporate(type='table') -%} |
10 | 10 | {%- set temp_relation = make_temp_relation(target_relation, '__dbt_tmp')-%} |
11 | 11 | {%- set temp_relation = temp_relation.incorporate(is_temporary=True)-%} {# file overwrite to ensure this is marked as temporary #} |
12 | 12 | {% do run_query(get_create_table_as_sql(True, temp_relation, get_empty_subquery_sql(sql))) %} |
13 | 13 | {%- set columns_in_relation = adapter.get_columns_in_relation(temp_relation) -%} |
14 | 14 | {%- set column_name_to_data_types = {} -%} |
| 15 | + {%- set column_name_to_quoted = {} -%} |
15 | 16 | {%- for column in columns_in_relation -%} |
16 | 17 | {%- do column_name_to_data_types.update({column.name|lower: column.data_type}) -%} |
| 18 | + {%- do column_name_to_quoted.update({column.name|lower: column.quoted}) -%} |
| 19 | + {%- endfor -%} |
| 20 | + |
| 21 | + {%- set expected_column_names_quoted = [] -%} |
| 22 | + {%- for column_name in tested_expected_column_names -%} |
| 23 | + {%- do expected_column_names_quoted.append(column_name_to_quoted[column_name|lower]) -%} |
17 | 24 | {%- endfor -%} |
18 | 25 |
|
19 | 26 | {% if not expected_sql %} |
20 | | - {% set expected_sql = get_expected_sql(expected_rows, column_name_to_data_types) %} |
| 27 | + {% set expected_sql = get_expected_sql(expected_rows, column_name_to_data_types, column_name_to_quoted) %} |
21 | 28 | {% endif %} |
22 | | - {% set unit_test_sql = get_unit_test_sql(sql, expected_sql, tested_expected_column_names) %} |
| 29 | + {% set unit_test_sql = get_unit_test_sql(sql, expected_sql, expected_column_names_quoted) %} |
23 | 30 |
|
24 | 31 | {% call statement('main', fetch_result=True) -%} |
25 | 32 |
|
|
0 commit comments