diff --git a/src/Common/dataAccess/dataTransfers.ts b/src/Common/dataAccess/dataTransfers.ts index e639f9965..721b000a2 100644 --- a/src/Common/dataAccess/dataTransfers.ts +++ b/src/Common/dataAccess/dataTransfers.ts @@ -35,24 +35,30 @@ export const getDataTransferJobs = async ( accountName: string, ): Promise => { let dataTransferJobs: DataTransferJobGetResults[] = []; - let dataTransferFeeds: DataTransferJobFeedResults = await listByDatabaseAccount( - subscriptionId, - resourceGroup, - accountName, - ); - 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 { + let dataTransferFeeds: DataTransferJobFeedResults = await listByDatabaseAccount( + subscriptionId, + resourceGroup, + accountName, + ); + 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; + } } + } 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(); });