From 1c0acee7b9975ad5c32df1a763ab0dc19a088f4b Mon Sep 17 00:00:00 2001
From: Pedro Sousa <pedronsousa@gmail.com>
Date: Fri, 19 Apr 2024 10:32:33 +0100
Subject: [PATCH] Fix bigquery tests

---
 .../biquery/model_with_quoted_columns_1_bq.sql  |  6 ++++++
 .../biquery/model_with_quoted_columns_2_bq.sql  | 17 +++++++++++++++++
 .../bigquery/model_with_quoted_columns_2_bq.sql | 14 ++++++++++++++
 .../tests/unit/model_with_quoted_columns_2.sql  |  2 +-
 package-lock.yml                                |  4 ++++
 5 files changed, 42 insertions(+), 1 deletion(-)
 create mode 100644 integration-tests/models/biquery/model_with_quoted_columns_1_bq.sql
 create mode 100644 integration-tests/models/biquery/model_with_quoted_columns_2_bq.sql
 create mode 100644 integration-tests/tests/unit/bigquery/model_with_quoted_columns_2_bq.sql
 create mode 100644 package-lock.yml

diff --git a/integration-tests/models/biquery/model_with_quoted_columns_1_bq.sql b/integration-tests/models/biquery/model_with_quoted_columns_1_bq.sql
new file mode 100644
index 0000000..d887810
--- /dev/null
+++ b/integration-tests/models/biquery/model_with_quoted_columns_1_bq.sql
@@ -0,0 +1,6 @@
+select  10 as lower,
+        20 as UPPER,
+        30 as MiXeD,
+        40 as `lower quoted`,
+        50 as `UPPER QUOTED`,
+        60 as `MiXeD Quoted`
diff --git a/integration-tests/models/biquery/model_with_quoted_columns_2_bq.sql b/integration-tests/models/biquery/model_with_quoted_columns_2_bq.sql
new file mode 100644
index 0000000..96986b5
--- /dev/null
+++ b/integration-tests/models/biquery/model_with_quoted_columns_2_bq.sql
@@ -0,0 +1,17 @@
+select 
+    lower,
+    UPPER,
+    MiXeD,
+    `lower quoted`,
+    `UPPER QUOTED`,
+    `MiXeD Quoted`
+from {{ dbt_unit_testing.ref('model_with_quoted_columns_1_bq') }}
+where
+    (lower = 10 or lower is null)
+and (UPPER = 20 or UPPER is null)
+and (MiXeD = 30 or MiXeD is null)
+and (`lower quoted` = 40 or `lower quoted` is null)
+and (`UPPER QUOTED` = 50 or `UPPER QUOTED` is null)
+and (`MiXeD Quoted` = 60 or `MiXeD Quoted` is null)
+
+
diff --git a/integration-tests/tests/unit/bigquery/model_with_quoted_columns_2_bq.sql b/integration-tests/tests/unit/bigquery/model_with_quoted_columns_2_bq.sql
new file mode 100644
index 0000000..4c2ea05
--- /dev/null
+++ b/integration-tests/tests/unit/bigquery/model_with_quoted_columns_2_bq.sql
@@ -0,0 +1,14 @@
+{{
+    config(
+        tags=['unit-test', 'bigquery']
+    )
+}}
+
+{% call dbt_unit_testing.test('model_with_quoted_columns_2_bq') %}
+  {% call dbt_unit_testing.mock_ref ('model_with_quoted_columns_1_bq', options={"include_missing_columns": true}) %}
+    select 10 as lower
+  {% endcall %}
+  {% call dbt_unit_testing.expect() %}
+    select 10 as lower
+  {% endcall %}
+{% endcall %}
diff --git a/integration-tests/tests/unit/model_with_quoted_columns_2.sql b/integration-tests/tests/unit/model_with_quoted_columns_2.sql
index 655c706..819a108 100644
--- a/integration-tests/tests/unit/model_with_quoted_columns_2.sql
+++ b/integration-tests/tests/unit/model_with_quoted_columns_2.sql
@@ -1,6 +1,6 @@
 {{
     config(
-        tags=['unit-test', 'bigquery', 'snowflake', 'postgres']
+        tags=['unit-test', 'snowflake', 'postgres']
     )
 }}
 
diff --git a/package-lock.yml b/package-lock.yml
new file mode 100644
index 0000000..6f7f80b
--- /dev/null
+++ b/package-lock.yml
@@ -0,0 +1,4 @@
+packages:
+- package: dbt-labs/dbt_utils
+  version: 1.1.1
+sha1_hash: 6c40b29313579fa0ad53f0b3f6b4f4779ea77de2