From 8379a3881adde2bcf90bcc101ee956689ce6882c Mon Sep 17 00:00:00 2001
From: yashmehta10 <yash_mehta27@ymail.com>
Date: Tue, 24 Oct 2023 09:17:30 +1100
Subject: [PATCH] adding dispatch for bigquery diff macro

---
 dbt_unit_test/templates/diff_macro.sql | 35 ++++++++++++++++++++------
 1 file changed, 27 insertions(+), 8 deletions(-)

diff --git a/dbt_unit_test/templates/diff_macro.sql b/dbt_unit_test/templates/diff_macro.sql
index b737a1a..9418b0b 100644
--- a/dbt_unit_test/templates/diff_macro.sql
+++ b/dbt_unit_test/templates/diff_macro.sql
@@ -1,10 +1,29 @@
 {% macro test_diff(model, test) %}
-with extra_rows as (
-    SELECT * FROM {{ model }} EXCEPT SELECT * FROM {{ test }}
-), 
-missing_rows as (
-    SELECT * FROM {{ test }} EXCEPT SELECT * FROM {{ model }}
-)
-SELECT count(*)
-FROM (SELECT * FROM extra_rows UNION ALL SELECT * FROM missing_rows) a
+    {{ return(adapter.dispatch('test_diff')(model, test)) }}
+{% endmacro %}
+
+{% macro default__test_diff(model, test) %}
+    with extra_rows as (
+        SELECT * FROM {{ model }} EXCEPT SELECT * FROM {{ test }}
+    ), 
+    missing_rows as (
+        SELECT * FROM {{ test }} EXCEPT SELECT * FROM {{ model }}
+    )
+    SELECT count(*)
+    FROM (SELECT * FROM extra_rows UNION ALL SELECT * FROM missing_rows) a
+{% endmacro %}
+
+{% macro bigquery__test_diff(model, test) %}
+    WITH extra_rows AS (
+    SELECT * FROM {{ model }} EXCEPT DISTINCT SELECT * FROM {{ test }}
+    ), 
+    missing_rows AS (
+        SELECT * FROM {{ test }} EXCEPT DISTINCT SELECT * FROM {{ model }}
+    ),
+    all_missing as (
+        SELECT COUNT(*) AS missing_count
+        FROM (SELECT * FROM extra_rows UNION ALL SELECT * FROM missing_rows) a
+    )
+    SELECT missing_count FROM all_missing
+    WHERE missing_count > 0
 {% endmacro %}