From 26a3ba98d2f3341af542bec75c5a00c682104a7f Mon Sep 17 00:00:00 2001
From: Radek Stankiewicz <radoslaws@google.com>
Date: Thu, 27 Nov 2025 14:13:53 +0100
Subject: [PATCH] Align FakeDatasetService and BigQueryServicesImpl behaviour
 on missing dataset when getTable is called. BQServicesImpl on 404 is
 returning null

---
 .../apache/beam/sdk/io/gcp/testing/FakeDatasetService.java    | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/testing/FakeDatasetService.java b/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/testing/FakeDatasetService.java
index 3c0ea08c033b..e9b1e25a7afc 100644
--- a/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/testing/FakeDatasetService.java
+++ b/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/testing/FakeDatasetService.java
@@ -261,9 +261,7 @@ public Table getTableImpl(
       Map<String, TableContainer> dataset =
           tables.get(tableRef.getProjectId(), tableRef.getDatasetId());
       if (dataset == null) {
-        throwNotFound(
-            "Tried to get a dataset %s:%s, but no such dataset was set",
-            tableRef.getProjectId(), tableRef.getDatasetId());
+        return null;
       }
       TableContainer tableContainer = dataset.get(tableRef.getTableId());