From 1063e019baf10cf77214ed4054fc34bc1475554c Mon Sep 17 00:00:00 2001
From: Keshav Mohta <68001229+keshavmohta09@users.noreply.github.com>
Date: Thu, 17 Apr 2025 11:43:14 +0530
Subject: [PATCH] Fixes: Bigquery E2E (#20863)

---
 ingestion/tests/cli_e2e/test_cli_bigquery.py                  | 4 ++--
 ingestion/tests/cli_e2e/test_cli_bigquery_multiple_project.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/ingestion/tests/cli_e2e/test_cli_bigquery.py b/ingestion/tests/cli_e2e/test_cli_bigquery.py
index bd5f674af2b..9ad080c5cc7 100644
--- a/ingestion/tests/cli_e2e/test_cli_bigquery.py
+++ b/ingestion/tests/cli_e2e/test_cli_bigquery.py
@@ -131,7 +131,7 @@ class BigqueryCliTest(CliCommonDB.TestSuite, SQACommonMethods):
 
     @staticmethod
     def expected_filtered_table_includes() -> int:
-        return 1
+        return 2
 
     @staticmethod
     def expected_filtered_table_excludes() -> int:
@@ -165,7 +165,7 @@ class BigqueryCliTest(CliCommonDB.TestSuite, SQACommonMethods):
                     SystemProfile(
                         timestamp=Timestamp(root=0),
                         operation=DmlOperationType.INSERT,
-                        rowsAffected=2,
+                        rowsAffected=1000,
                     ),
                     SystemProfile(
                         timestamp=Timestamp(root=1),
diff --git a/ingestion/tests/cli_e2e/test_cli_bigquery_multiple_project.py b/ingestion/tests/cli_e2e/test_cli_bigquery_multiple_project.py
index 7b024cb62f1..8ad84b631ff 100644
--- a/ingestion/tests/cli_e2e/test_cli_bigquery_multiple_project.py
+++ b/ingestion/tests/cli_e2e/test_cli_bigquery_multiple_project.py
@@ -119,7 +119,7 @@ class BigqueryCliTest(CliCommonDB.TestSuite, SQACommonMethods):
 
     @staticmethod
     def expected_filtered_mix() -> int:
-        return 17
+        return 19
 
     @staticmethod
     def delete_queries() -> List[str]: