From 9f819d9043787dba4858f6af54f2cbfaa43f23f4 Mon Sep 17 00:00:00 2001
From: Bikram Choudhury <bchoudhury@microsoft.com>
Date: Sun, 4 Jan 2026 02:06:15 +0530
Subject: [PATCH] added try-catch block

---
 src/Common/dataAccess/dataTransfers.ts        | 26 ++++++++++++-------
 .../changePartitionKey.spec.ts                |  2 +-
 2 files changed, 17 insertions(+), 11 deletions(-)

diff --git a/src/Common/dataAccess/dataTransfers.ts b/src/Common/dataAccess/dataTransfers.ts
index e639f9965..0e39c2725 100644
--- a/src/Common/dataAccess/dataTransfers.ts
+++ b/src/Common/dataAccess/dataTransfers.ts
@@ -42,16 +42,22 @@ export const getDataTransferJobs = async (
   );
   dataTransferJobs = [...dataTransferJobs, ...(dataTransferFeeds?.value || [])];
   while (dataTransferFeeds?.nextLink) {
-    const nextResponse = await window.fetch(dataTransferFeeds.nextLink, {
-      headers: {
-        Authorization: userContext.authorizationToken,
-      },
-    });
-    if (nextResponse.ok) {
-      dataTransferFeeds = await nextResponse.json();
-      dataTransferJobs = [...dataTransferJobs, ...(dataTransferFeeds?.value || [])];
-    } else {
-      break;
+    try {
+      const nextResponse = await window.fetch(dataTransferFeeds.nextLink, {
+        headers: {
+          Authorization: userContext.authorizationToken,
+        },
+      });
+      if (nextResponse.ok) {
+        dataTransferFeeds = await nextResponse.json();
+        dataTransferJobs = [...dataTransferJobs, ...(dataTransferFeeds?.value || [])];
+      } else {
+        break;
+      }
+    } catch (error) {
+      // Handle network errors gracefully (e.g., in test environment)
+      console.warn("Failed to fetch next page of data transfer jobs:", error);
+      throw new Error(error.message || JSON.stringify(error));
     }
   }
   return dataTransferJobs;
diff --git a/test/sql/scaleAndSettings/changePartitionKey.spec.ts b/test/sql/scaleAndSettings/changePartitionKey.spec.ts
index 8cc40db26..f6e73dd3e 100644
--- a/test/sql/scaleAndSettings/changePartitionKey.spec.ts
+++ b/test/sql/scaleAndSettings/changePartitionKey.spec.ts
@@ -20,7 +20,7 @@ test.describe("Change Partition Key", () => {
     // Click Scale & Settings and open Partition Key tab
     await explorer.openScaleAndSettings(context);
     const PartitionKeyTab = explorer.frame.getByTestId("settings-tab-header/PartitionKeyTab");
-    
+
     await expect(PartitionKeyTab).toBeVisible();
     await PartitionKeyTab.click();
   });