(#201) Fix for column tests not rendering on quoted columns

dbt-labs · May 31, 2023 · 2f09d3a · 2f09d3a
1 parent deb8cfa
commit 2f09d3a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/.changes/unreleased/Docs-20230531-115419.yaml b/.changes/unreleased/Docs-20230531-115419.yaml
@@ -0,0 +1,6 @@
+kind: Docs
+body: Fix for column tests not rendering on quoted columns
+time: 2023-05-31T11:54:19.687363-04:00
+custom:
+  Author: drewbanin
+  Issue: "201"
diff --git a/src/app/services/project_service.js b/src/app/services/project_service.js
@@ -228,7 +228,13 @@ angular
                     }
                     var node = project.nodes[model];
                     var column = _.find(node.columns, function(col, col_name) {
-                        return col_name.toLowerCase() == test_column.toLowerCase();
+                        // strip quotes from start and end of test column if present in both locations
+                        // this is necessary to attach a test to a column when `quote: true` is set for a column
+                        let test_column_name = test_column;
+                        if (test_column.startsWith('"') && test_column.endsWith('"')) {
+                            test_column_name = test_column.substring(1, test_column.length-1);
+                        }
+                        return col_name.toLowerCase() == test_column_name.toLowerCase();
                     });
 
                     if (column) {