diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index adfec59b7..dac26c32d 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -213,6 +213,8 @@ jobs: # MONGO_READONLY_TESTACCOUNT_TOKEN=$(az account get-access-token --scope "https://github-e2etests-mongo-readonly.documents.azure.com/.default" -o tsv --query accessToken) # echo "::add-mask::$MONGO_READONLY_TESTACCOUNT_TOKEN" # echo MONGO_READONLY_TESTACCOUNT_TOKEN=$MONGO_READONLY_TESTACCOUNT_TOKEN >> $GITHUB_ENV + - name: List test files for shard ${{ matrix['shardIndex'] }} of ${{ matrix['shardTotal']}} + run: npx playwright test --shard=${{ matrix.shardIndex }}/${{ matrix.shardTotal }} --list - name: Run test shard ${{ matrix['shardIndex'] }} of ${{ matrix['shardTotal']}} run: npx playwright test --shard=${{ matrix.shardIndex }}/${{ matrix.shardTotal }} --workers=3 - name: Upload blob report to GitHub Actions Artifacts diff --git a/test/fx.ts b/test/fx.ts index 6aa545407..c1c2b6a47 100644 --- a/test/fx.ts +++ b/test/fx.ts @@ -602,30 +602,46 @@ export async function waitForApiResponse( // eslint-disable-next-line @typescript-eslint/no-explicit-any payloadValidator?: (payload: any) => boolean, ) { - return page.waitForResponse(async (response) => { - const request = response.request(); - - if (!request.url().includes(urlPattern)) { - return false; + try { + // Check if page is still valid before waiting + if (page.isClosed()) { + throw new Error(`Page is closed, cannot wait for API response: ${urlPattern}`); } - if (method && request.method() !== method) { - return false; - } + return page.waitForResponse( + async (response) => { + const request = response.request(); - if (payloadValidator && (request.method() === "POST" || request.method() === "PUT")) { - const postData = request.postData(); - if (postData) { - try { - const payload = JSON.parse(postData); - return payloadValidator(payload); - } catch { + if (!request.url().includes(urlPattern)) { return false; } - } + + if (method && request.method() !== method) { + return false; + } + + if (payloadValidator && (request.method() === "POST" || request.method() === "PUT")) { + const postData = request.postData(); + if (postData) { + try { + const payload = JSON.parse(postData); + return payloadValidator(payload); + } catch { + return false; + } + } + } + return true; + }, + { timeout: 60 * 1000 }, + ); + } catch (error) { + if (error instanceof Error && error.message.includes("Target page, context or browser has been closed")) { + console.warn("Page was closed while waiting for API response:", urlPattern); + throw new Error(`Page closed while waiting for API response: ${urlPattern}`); } - return true; - }); + throw error; + } } export async function interceptAndInspectApiRequest( page: Page, diff --git a/test/sql/containercopy.spec.ts b/test/sql/containercopy.spec.ts index a7b05b07e..00742c515 100644 --- a/test/sql/containercopy.spec.ts +++ b/test/sql/containercopy.spec.ts @@ -1,7 +1,6 @@ /* eslint-disable @typescript-eslint/no-explicit-any */ import { expect, Frame, Locator, Page, test } from "@playwright/test"; import { set } from "lodash"; -import { DatabaseAccount, Subscription } from "../../src/Contracts/DataModels"; import { truncateName } from "../../src/Explorer/ContainerCopy/CopyJobUtils"; import { ContainerCopy, @@ -11,20 +10,21 @@ import { TestAccount, waitForApiResponse, } from "../fx"; +import { createMultipleTestContainers } from "../testData"; -test.describe.configure({ mode: "serial" }); let page: Page; let wrapper: Locator = null!; let panel: Locator = null!; let frame: Frame = null!; let expectedCopyJobNameInitial: string = null!; -let expectedSourceSubscription: any = null!; -let expectedSourceAccount: DatabaseAccount = null!; let expectedJobName: string = ""; let targetAccountName: string = ""; let expectedSourceAccountName: string = ""; +let expectedSubscriptionName: string = ""; test.beforeAll("Container Copy - Before All", async ({ browser }) => { + await createMultipleTestContainers({ accountType: TestAccount.SQLContainerCopyOnly, containerCount: 3 }); + page = await browser.newPage(); ({ wrapper, frame } = await ContainerCopy.open(page, TestAccount.SQLContainerCopyOnly)); expectedJobName = `test_job_${Date.now()}`; @@ -42,15 +42,380 @@ test("Loading and verifying the content of the page", async () => { await expect(wrapper.getByTestId("CommandBar/Button:Feedback")).toBeVisible(); }); -test("Opening the Create Copy Job panel", async () => { +test("Successfully create a copy job for offline migration", async () => { + expect(wrapper).not.toBeNull(); + // Loading and verifying subscription & account dropdown + + const createCopyJobButton = wrapper.getByTestId("CommandBar/Button:Create Copy Job"); + await createCopyJobButton.click(); + panel = frame.getByTestId("Panel:Create copy job"); + await expect(panel).toBeVisible(); + + await page.waitForTimeout(10 * 1000); + + const subscriptionDropdown = panel.getByTestId("subscription-dropdown"); + + const expectedAccountName = targetAccountName; + expectedSubscriptionName = await subscriptionDropdown.locator("span.ms-Dropdown-title").innerText(); + + await subscriptionDropdown.click(); + const subscriptionItem = await getDropdownItemByNameOrPosition( + frame, + { name: expectedSubscriptionName }, + { ariaLabel: "Subscription" }, + ); + await subscriptionItem.click(); + + // Load account dropdown based on selected subscription + + const accountDropdown = panel.getByTestId("account-dropdown"); + await expect(accountDropdown).toHaveText(new RegExp(expectedAccountName)); + await accountDropdown.click(); + + const accountItem = await getDropdownItemByNameOrPosition( + frame, + { name: expectedAccountName }, + { ariaLabel: "Account" }, + ); + await accountItem.click(); + + // Verifying online or offline checkbox functionality + /** + * This test verifies the functionality of the migration type checkbox that toggles between + * online and offline container copy modes. It ensures that: + * 1. When online mode is selected, the user is directed to a permissions screen + * 2. When offline mode is selected, the user bypasses the permissions screen + * 3. The UI correctly reflects the selected migration type throughout the workflow + */ + const fluentUiCheckboxContainer = panel.getByTestId("migration-type-checkbox").locator("div.ms-Checkbox"); + await fluentUiCheckboxContainer.click(); + await panel.getByRole("button", { name: "Next" }).click(); + await expect(panel.getByTestId("Panel:AssignPermissionsContainer")).toBeVisible(); + await expect(panel.getByText("Online container copy", { exact: true })).toBeVisible(); + await panel.getByRole("button", { name: "Previous" }).click(); + await fluentUiCheckboxContainer.click(); + await panel.getByRole("button", { name: "Next" }).click(); + await expect(panel.getByTestId("Panel:SelectSourceAndTargetContainers")).toBeVisible(); + await expect(panel.getByTestId("Panel:AssignPermissionsContainer")).not.toBeVisible(); + + // Verifying source and target container selection + + const sourceContainerDropdown = panel.getByTestId("source-containerDropdown"); + expect(sourceContainerDropdown).toBeVisible(); + await expect(sourceContainerDropdown).toHaveClass(/(^|\s)is-disabled(\s|$)/); + + const sourceDatabaseDropdown = panel.getByTestId("source-databaseDropdown"); + await sourceDatabaseDropdown.click(); + + const sourceDbDropdownItem = await getDropdownItemByNameOrPosition(frame, { position: 0 }, { ariaLabel: "Database" }); + await sourceDbDropdownItem.click(); + + await expect(sourceContainerDropdown).not.toHaveClass(/(^|\s)is-disabled(\s|$)/); + await sourceContainerDropdown.click(); + const sourceContainerDropdownItem = await getDropdownItemByNameOrPosition( + frame, + { position: 0 }, + { ariaLabel: "Container" }, + ); + await sourceContainerDropdownItem.click(); + + const targetContainerDropdown = panel.getByTestId("target-containerDropdown"); + expect(targetContainerDropdown).toBeVisible(); + await expect(targetContainerDropdown).toHaveClass(/(^|\s)is-disabled(\s|$)/); + + const targetDatabaseDropdown = panel.getByTestId("target-databaseDropdown"); + await targetDatabaseDropdown.click(); + const targetDbDropdownItem = await getDropdownItemByNameOrPosition(frame, { position: 0 }, { ariaLabel: "Database" }); + await targetDbDropdownItem.click(); + + await expect(targetContainerDropdown).not.toHaveClass(/(^|\s)is-disabled(\s|$)/); + await targetContainerDropdown.click(); + const targetContainerDropdownItem1 = await getDropdownItemByNameOrPosition( + frame, + { position: 0 }, + { ariaLabel: "Container" }, + ); + await targetContainerDropdownItem1.click(); + + await panel.getByRole("button", { name: "Next" }).click(); + + const errorContainer = panel.getByTestId("Panel:ErrorContainer"); + await expect(errorContainer).toBeVisible(); + await expect(errorContainer).toHaveText(/Source and destination containers cannot be the same/i); + + // Reselect target container to be different from source container + await targetContainerDropdown.click(); + const targetContainerDropdownItem2 = await getDropdownItemByNameOrPosition( + frame, + { position: 1 }, + { ariaLabel: "Container" }, + ); + await targetContainerDropdownItem2.click(); + + const selectedSourceDatabase = await sourceDatabaseDropdown.innerText(); + const selectedSourceContainer = await sourceContainerDropdown.innerText(); + const selectedTargetDatabase = await targetDatabaseDropdown.innerText(); + const selectedTargetContainer = await targetContainerDropdown.innerText(); + expectedCopyJobNameInitial = `${truncateName(selectedSourceDatabase)}.${truncateName( + selectedSourceContainer, + )}_${truncateName(selectedTargetDatabase)}.${truncateName(selectedTargetContainer)}`; + + await panel.getByRole("button", { name: "Next" }).click(); + + await expect(errorContainer).not.toBeVisible(); + await expect(panel.getByTestId("Panel:PreviewCopyJob")).toBeVisible(); + + // Verifying the preview of the copy job + const previewContainer = panel.getByTestId("Panel:PreviewCopyJob"); + await expect(previewContainer).toBeVisible(); + await expect(previewContainer.getByTestId("source-subscription-name")).toHaveText(expectedSubscriptionName); + await expect(previewContainer.getByTestId("source-account-name")).toHaveText(expectedAccountName); + const jobNameInput = previewContainer.getByTestId("job-name-textfield"); + await expect(jobNameInput).toHaveValue(new RegExp(expectedCopyJobNameInitial)); + const primaryBtn = panel.getByRole("button", { name: "Copy", exact: true }); + await expect(primaryBtn).not.toHaveClass(/(^|\s)is-disabled(\s|$)/); + + await jobNameInput.fill("test job name"); + await expect(primaryBtn).toHaveClass(/(^|\s)is-disabled(\s|$)/); + + // Testing API request interception with duplicate job name + const duplicateJobName = "test-job-name-1"; + await jobNameInput.fill(duplicateJobName); + + const copyButton = panel.getByRole("button", { name: "Copy", exact: true }); + const expectedErrorMessage = `Duplicate job name '${duplicateJobName}'`; + await interceptAndInspectApiRequest( + page, + `${expectedAccountName}/dataTransferJobs/${duplicateJobName}`, + "PUT", + new Error(expectedErrorMessage), + (url?: string) => url?.includes(duplicateJobName) ?? false, + ); + + let errorThrown = false; + try { + await copyButton.click(); + await page.waitForTimeout(2000); + } catch (error: any) { + errorThrown = true; + expect(error.message).toContain("not allowed"); + } + if (!errorThrown) { + const errorContainer = panel.getByTestId("Panel:ErrorContainer"); + await expect(errorContainer).toBeVisible(); + await expect(errorContainer).toHaveText(new RegExp(expectedErrorMessage, "i")); + } + + await expect(panel).toBeVisible(); + + // Testing API request success with valid job name and verifying copy job creation + + const validJobName = expectedJobName; + + const copyJobCreationPromise = waitForApiResponse( + page, + `${expectedAccountName}/dataTransferJobs/${validJobName}`, + "PUT", + ); + + await jobNameInput.fill(validJobName); + await expect(copyButton).not.toHaveClass(/(^|\s)is-disabled(\s|$)/); + + await copyButton.click(); + + const response = await copyJobCreationPromise; + expect(response.ok()).toBe(true); + + await expect(panel).not.toBeVisible({ timeout: 10000 }); + + const jobsListContainer = wrapper.locator(".CopyJobListContainer .ms-DetailsList-contentWrapper .ms-List-page"); + await jobsListContainer.waitFor({ state: "visible" }); + + const jobItem = jobsListContainer.getByText(validJobName); + await jobItem.waitFor({ state: "visible" }); + await expect(jobItem).toBeVisible(); + + /* // Cancel the created job to clean up + + // Rapid polling to catch the job in running state + let attempts = 0; + const maxAttempts = 50; // Try for ~5 seconds + let jobCancelled = false; + + while (attempts < maxAttempts && !jobCancelled) { + try { + // Look for the job row + const jobRow = jobsListContainer.locator(".ms-DetailsRow", { hasText: validJobName }); + + if (await jobRow.isVisible({ timeout: 100 })) { + const statusCell = jobRow.locator("[data-automationid='DetailsRowCell'][data-automation-key='CopyJobStatus']"); + const statusText = await statusCell.textContent({ timeout: 100 }); + + // If job is still running/queued, try to cancel it + if (statusText && /running|queued|pending/i.test(statusText)) { + const actionMenuButton = wrapper.getByTestId(`CopyJobActionMenu/Button:${validJobName}`); + await actionMenuButton.click({ timeout: 1000 }); + + const cancelAction = frame.locator(".ms-ContextualMenu-list button:has-text('Cancel')"); + if (await cancelAction.isVisible({ timeout: 1000 })) { + await cancelAction.click(); + + // Verify cancellation + await expect(statusCell).toContainText(/cancelled|canceled|failed/i, { timeout: 5000 }); + jobCancelled = true; + break; + } + } else if (statusText && /completed|succeeded|finished/i.test(statusText)) { + // Job completed too fast, skip the test + // console.log(`Job ${validJobName} completed too quickly to test cancellation`); + test.skip(true, "Job completed too quickly for cancellation test"); + return; + } + } + + // Refresh the job list + const refreshButton = wrapper.getByTestId("CommandBar/Button:Refresh"); + if (await refreshButton.isVisible({ timeout: 100 })) { + await refreshButton.click(); + await page.waitForTimeout(100); // Small delay between attempts + } + } catch (error) { + // Continue trying if there's any error + } + + attempts++; + } + + if (!jobCancelled) { + // If we couldn't cancel in time, at least verify the job was created + const jobRow = jobsListContainer.locator(".ms-DetailsRow", { hasText: validJobName }); + await expect(jobRow).toBeVisible({ timeout: 5000 }); + test.skip(true, "Could not catch job in running state for cancellation test"); + } */ +}); + +/* test.skip("Pause a running copy job", async () => { + const jobsListContainer = wrapper.locator(".CopyJobListContainer .ms-DetailsList-contentWrapper .ms-List-page"); + await jobsListContainer.waitFor({ state: "visible" }); + + const firstJobRow = jobsListContainer.locator(".ms-DetailsRow", { hasText: expectedJobName }); + await firstJobRow.waitFor({ state: "visible" }); + + const actionMenuButton = wrapper.getByTestId(`CopyJobActionMenu/Button:${expectedJobName}`); + await actionMenuButton.waitFor({ state: "visible" }); + await actionMenuButton.click(); + + const pauseAction = frame.locator(".ms-ContextualMenu-list button:has-text('Pause')"); + await pauseAction.waitFor({ state: "visible" }); + await pauseAction.click(); + + const updatedJobRow = jobsListContainer.locator(".ms-DetailsRow").filter({ hasText: expectedJobName }); + const statusCell = updatedJobRow.locator("[data-automationid='DetailsRowCell'][data-automation-key='CopyJobStatus']"); + await expect(statusCell).toContainText(/paused/i, { timeout: 10000 }); +}); + +test.skip("Resume a paused copy job", async () => { + const jobsListContainer = wrapper.locator(".CopyJobListContainer .ms-DetailsList-contentWrapper .ms-List-page"); + await jobsListContainer.waitFor({ state: "visible" }); + + const pausedJobRow = jobsListContainer.locator(".ms-DetailsRow", { hasText: expectedJobName }); + await pausedJobRow.waitFor({ state: "visible" }); + + const statusCell = pausedJobRow.locator("[data-automationid='DetailsRowCell'][data-automation-key='CopyJobStatus']"); + await expect(statusCell).toContainText(/paused/i); + + const actionMenuButton = wrapper.getByTestId(`CopyJobActionMenu/Button:${expectedJobName}`); + await actionMenuButton.waitFor({ state: "visible" }); + await actionMenuButton.click(); + + const resumeAction = frame.locator(".ms-ContextualMenu-list button:has-text('Resume')"); + await resumeAction.waitFor({ state: "visible" }); + await resumeAction.click(); + + await expect(statusCell).toContainText(/running|queued/i); +}); */ + +test("Create and Cancel a copy job", async () => { + expect(wrapper).not.toBeNull(); + // Create a new job specifically for cancellation testing + const cancelJobName = `cancel_test_job_${Date.now()}`; + + // Navigate to create job panel + const createCopyJobButton = wrapper.getByTestId("CommandBar/Button:Create Copy Job"); + await createCopyJobButton.click(); + panel = frame.getByTestId("Panel:Create copy job"); + + // Skip to container selection (offline mode for faster creation) + await panel.getByRole("button", { name: "Next" }).click(); + + // Select source containers quickly + const sourceDatabaseDropdown = panel.getByTestId("source-databaseDropdown"); + await sourceDatabaseDropdown.click(); + const sourceDatabaseDropdownItem = await getDropdownItemByNameOrPosition( + frame, + { position: 0 }, + { ariaLabel: "Database" }, + ); + await sourceDatabaseDropdownItem.click(); + + const sourceContainerDropdown = panel.getByTestId("source-containerDropdown"); + await sourceContainerDropdown.click(); + const sourceContainerDropdownItem = await getDropdownItemByNameOrPosition( + frame, + { position: 0 }, + { ariaLabel: "Container" }, + ); + await sourceContainerDropdownItem.click(); + + // Select target containers + const targetDatabaseDropdown = panel.getByTestId("target-databaseDropdown"); + await targetDatabaseDropdown.click(); + const targetDatabaseDropdownItem = await getDropdownItemByNameOrPosition( + frame, + { position: 0 }, + { ariaLabel: "Database" }, + ); + await targetDatabaseDropdownItem.click(); + + const targetContainerDropdown = panel.getByTestId("target-containerDropdown"); + await targetContainerDropdown.click(); + const targetContainerDropdownItem = await getDropdownItemByNameOrPosition( + frame, + { position: 1 }, + { ariaLabel: "Container" }, + ); + await targetContainerDropdownItem.click(); + + await panel.getByRole("button", { name: "Next" }).click(); + + // Set job name and create + const jobNameInput = panel.getByTestId("job-name-textfield"); + await jobNameInput.fill(cancelJobName); + + const copyButton = panel.getByRole("button", { name: "Copy", exact: true }); + + // Create job and immediately start polling for it + await copyButton.click(); + + // Wait for panel to close and job list to refresh + await expect(panel).not.toBeVisible({ timeout: 10000 }); + + const jobsListContainer = wrapper.locator(".CopyJobListContainer .ms-DetailsList-contentWrapper .ms-List-page"); + await jobsListContainer.waitFor({ state: "visible" }); +}); + +test("Verify Online or Offline Container Copy Permissions Panel", async () => { + expect(wrapper).not.toBeNull(); + + // Opening the Create Copy Job panel again to verify initial state const createCopyJobButton = wrapper.getByTestId("CommandBar/Button:Create Copy Job"); await createCopyJobButton.click(); panel = frame.getByTestId("Panel:Create copy job"); await expect(panel).toBeVisible(); await expect(panel.getByRole("heading", { name: "Create copy job" })).toBeVisible(); -}); -test("select different account dropdown", async () => { + // select different account dropdown + const accountDropdown = panel.getByTestId("account-dropdown"); await accountDropdown.click(); @@ -79,17 +444,17 @@ test("select different account dropdown", async () => { await fluentUiCheckboxContainer.click(); await panel.getByRole("button", { name: "Next" }).click(); -}); -test("Verifying Assign Permissions panel for online copy", async () => { + // Verifying Assign Permissions panel for online copy + const permissionScreen = panel.getByTestId("Panel:AssignPermissionsContainer"); await expect(permissionScreen).toBeVisible(); await expect(permissionScreen.getByText("Online container copy", { exact: true })).toBeVisible(); await expect(permissionScreen.getByText("Cross-account container copy", { exact: true })).toBeVisible(); -}); -test("Verify Point-in-Time Restore timer and refresh button workflow", async () => { + // Verify Point-in-Time Restore timer and refresh button workflow + await page.route(`**/Microsoft.DocumentDB/databaseAccounts/${expectedSourceAccountName}**`, async (route) => { const mockData = { identity: { @@ -124,15 +489,14 @@ test("Verify Point-in-Time Restore timer and refresh button workflow", async () } }); - const permissionScreen = panel.getByTestId("Panel:AssignPermissionsContainer"); await expect(permissionScreen).toBeVisible(); - const expandedAccordionHeader = permissionScreen + const expandedOnlineAccordionHeader = permissionScreen .getByTestId("permission-group-container-onlineConfigs") .locator("button[aria-expanded='true']"); - await expect(expandedAccordionHeader).toBeVisible(); + await expect(expandedOnlineAccordionHeader).toBeVisible(); - const accordionItem = expandedAccordionHeader + const accordionItem = expandedOnlineAccordionHeader .locator("xpath=ancestor::*[contains(@class, 'fui-AccordionItem') or contains(@data-test, 'accordion-item')]") .first(); @@ -168,9 +532,9 @@ test("Verify Point-in-Time Restore timer and refresh button workflow", async () await expect(loadingOverlay).toBeVisible(); await expect(loadingOverlay).toBeHidden({ timeout: 10 * 1000 }); -}); -test("Veify Popover & Loading Overlay on permission screen with API mocks and accordion interactions", async () => { + // Veify Popover & Loading Overlay on permission screen with API mocks and accordion interactions + await page.route( `**/Microsoft.DocumentDB/databaseAccounts/${expectedSourceAccountName}/sqlRoleAssignments*`, async (route) => { @@ -244,21 +608,22 @@ test("Veify Popover & Loading Overlay on permission screen with API mocks and ac } }); - const permissionScreen = panel.getByTestId("Panel:AssignPermissionsContainer"); await expect(permissionScreen).toBeVisible(); - const expandedAccordionHeader = permissionScreen + const expandedCrossAccordionHeader = permissionScreen .getByTestId("permission-group-container-crossAccountConfigs") .locator("button[aria-expanded='true']"); - await expect(expandedAccordionHeader).toBeVisible(); + await expect(expandedCrossAccordionHeader).toBeVisible(); - const accordionItem = expandedAccordionHeader + const crossAccordionItem = expandedCrossAccordionHeader .locator("xpath=ancestor::*[contains(@class, 'fui-AccordionItem') or contains(@data-test, 'accordion-item')]") .first(); - const accordionPanel = accordionItem.locator("[role='tabpanel'], .fui-AccordionPanel, [data-test*='panel']").first(); + const crossAccordionPanel = crossAccordionItem + .locator("[role='tabpanel'], .fui-AccordionPanel, [data-test*='panel']") + .first(); - const toggleButton = accordionPanel.getByTestId("btn-toggle"); + const toggleButton = crossAccordionPanel.getByTestId("btn-toggle"); await expect(toggleButton).toBeVisible(); await toggleButton.click(); @@ -272,7 +637,6 @@ test("Veify Popover & Loading Overlay on permission screen with API mocks and ac await yesButton.click(); - const loadingOverlay = frame.locator("[data-test='loading-overlay']"); await expect(loadingOverlay).toBeVisible(); await expect(loadingOverlay).toBeHidden({ timeout: 10 * 1000 }); @@ -281,396 +645,6 @@ test("Veify Popover & Loading Overlay on permission screen with API mocks and ac await panel.getByRole("button", { name: "Cancel" }).click(); }); -test("Loading and verifying subscription & account dropdown", async () => { - const createCopyJobButton = wrapper.getByTestId("CommandBar/Button:Create Copy Job"); - await createCopyJobButton.click(); - panel = frame.getByTestId("Panel:Create copy job"); - await expect(panel).toBeVisible(); - - const subscriptionPromise = waitForApiResponse(page, "/Microsoft.ResourceGraph/resources", "POST", (payload: any) => { - return ( - payload.query.includes("resources | where type == 'microsoft.documentdb/databaseaccounts'") && - payload.query.includes("| where type == 'microsoft.resources/subscriptions'") - ); - }); - - const accountPromise = waitForApiResponse(page, "/Microsoft.ResourceGraph/resources", "POST", (payload: any) => { - return payload.query.includes("resources | where type =~ 'microsoft.documentdb/databaseaccounts'"); - }); - - const subscriptionResponse = await subscriptionPromise; - const data = await subscriptionResponse.json(); - expect(subscriptionResponse.ok()).toBe(true); - - const accountResponse = await accountPromise; - const accountData = await accountResponse.json(); - expect(accountResponse.ok()).toBe(true); - - const selectedSubscription = data.data.find( - (item: Subscription) => item.subscriptionId === process.env.DE_TEST_SUBSCRIPTION_ID, - ); - - const subscriptionDropdown = panel.getByTestId("subscription-dropdown"); - await expect(subscriptionDropdown).toHaveText(new RegExp(selectedSubscription.subscriptionName)); - await subscriptionDropdown.click(); - - const subscriptionItem = await getDropdownItemByNameOrPosition( - frame, - { name: selectedSubscription.subscriptionName }, - { ariaLabel: "Subscription", itemCount: data.count }, - ); - await subscriptionItem.click(); - - const expectedAccountName = getAccountName(TestAccount.SQLContainerCopyOnly); - const selectedAccount = accountData.data.find((item: DatabaseAccount) => item.name === expectedAccountName); - - const accountDropdown = panel.getByTestId("account-dropdown"); - await expect(accountDropdown).toHaveText(new RegExp(expectedAccountName)); - await accountDropdown.click(); - - const accountItem = await getDropdownItemByNameOrPosition( - frame, - { name: expectedAccountName }, - { ariaLabel: "Account" }, - ); - await accountItem.click(); - - expectedSourceSubscription = selectedSubscription; - expectedSourceAccount = selectedAccount; -}); - -test("Verifying online or offline checkbox", async () => { - /** - * This test verifies the functionality of the migration type checkbox that toggles between - * online and offline container copy modes. It ensures that: - * 1. When online mode is selected, the user is directed to a permissions screen - * 2. When offline mode is selected, the user bypasses the permissions screen - * 3. The UI correctly reflects the selected migration type throughout the workflow - */ - const fluentUiCheckboxContainer = panel.getByTestId("migration-type-checkbox").locator("div.ms-Checkbox"); - await fluentUiCheckboxContainer.click(); - await panel.getByRole("button", { name: "Next" }).click(); - await expect(panel.getByTestId("Panel:AssignPermissionsContainer")).toBeVisible(); - await expect(panel.getByText("Online container copy", { exact: true })).toBeVisible(); - await panel.getByRole("button", { name: "Previous" }).click(); - await fluentUiCheckboxContainer.click(); - await panel.getByRole("button", { name: "Next" }).click(); - await expect(panel.getByTestId("Panel:SelectSourceAndTargetContainers")).toBeVisible(); - await expect(panel.getByTestId("Panel:AssignPermissionsContainer")).not.toBeVisible(); -}); - -test("Verifying source and target container selection", async () => { - const sourceContainerDropdown = panel.getByTestId("source-containerDropdown"); - expect(sourceContainerDropdown).toBeVisible(); - await expect(sourceContainerDropdown).toHaveClass(/(^|\s)is-disabled(\s|$)/); - - const sourceDatabaseDropdown = panel.getByTestId("source-databaseDropdown"); - await sourceDatabaseDropdown.click(); - - const sourceDbDropdownItem = await getDropdownItemByNameOrPosition(frame, { position: 0 }, { ariaLabel: "Database" }); - await sourceDbDropdownItem.click(); - - await expect(sourceContainerDropdown).not.toHaveClass(/(^|\s)is-disabled(\s|$)/); - await sourceContainerDropdown.click(); - const sourceContainerDropdownItem = await getDropdownItemByNameOrPosition( - frame, - { position: 0 }, - { ariaLabel: "Container" }, - ); - await sourceContainerDropdownItem.click(); - - const targetContainerDropdown = panel.getByTestId("target-containerDropdown"); - expect(targetContainerDropdown).toBeVisible(); - await expect(targetContainerDropdown).toHaveClass(/(^|\s)is-disabled(\s|$)/); - - const targetDatabaseDropdown = panel.getByTestId("target-databaseDropdown"); - await targetDatabaseDropdown.click(); - const targetDbDropdownItem = await getDropdownItemByNameOrPosition(frame, { position: 0 }, { ariaLabel: "Database" }); - await targetDbDropdownItem.click(); - - await expect(targetContainerDropdown).not.toHaveClass(/(^|\s)is-disabled(\s|$)/); - await targetContainerDropdown.click(); - const targetContainerDropdownItem1 = await getDropdownItemByNameOrPosition( - frame, - { position: 0 }, - { ariaLabel: "Container" }, - ); - await targetContainerDropdownItem1.click(); - - await panel.getByRole("button", { name: "Next" }).click(); - - const errorContainer = panel.getByTestId("Panel:ErrorContainer"); - await expect(errorContainer).toBeVisible(); - await expect(errorContainer).toHaveText(/Source and destination containers cannot be the same/i); - - // Reselect target container to be different from source container - await targetContainerDropdown.click(); - const targetContainerDropdownItem2 = await getDropdownItemByNameOrPosition( - frame, - { position: 1 }, - { ariaLabel: "Container" }, - ); - await targetContainerDropdownItem2.click(); - - const selectedSourceDatabase = await sourceDatabaseDropdown.innerText(); - const selectedSourceContainer = await sourceContainerDropdown.innerText(); - const selectedTargetDatabase = await targetDatabaseDropdown.innerText(); - const selectedTargetContainer = await targetContainerDropdown.innerText(); - expectedCopyJobNameInitial = `${truncateName(selectedSourceDatabase)}.${truncateName( - selectedSourceContainer, - )}_${truncateName(selectedTargetDatabase)}.${truncateName(selectedTargetContainer)}`; - - await panel.getByRole("button", { name: "Next" }).click(); - - await expect(errorContainer).not.toBeVisible(); - await expect(panel.getByTestId("Panel:PreviewCopyJob")).toBeVisible(); -}); - -test("Verifying the preview of the copy job", async () => { - const previewContainer = panel.getByTestId("Panel:PreviewCopyJob"); - await expect(previewContainer).toBeVisible(); - await expect(previewContainer.getByTestId("source-subscription-name")).toHaveText( - expectedSourceSubscription.subscriptionName, - ); - await expect(previewContainer.getByTestId("source-account-name")).toHaveText(expectedSourceAccount.name); - const jobNameInput = previewContainer.getByTestId("job-name-textfield"); - await expect(jobNameInput).toHaveValue(new RegExp(expectedCopyJobNameInitial)); - const primaryBtn = panel.getByRole("button", { name: "Copy", exact: true }); - await expect(primaryBtn).not.toHaveClass(/(^|\s)is-disabled(\s|$)/); - - await jobNameInput.fill("test job name"); - await expect(primaryBtn).toHaveClass(/(^|\s)is-disabled(\s|$)/); -}); - -test("Testing API request interception with duplicate job name", async () => { - const previewContainer = panel.getByTestId("Panel:PreviewCopyJob"); - const jobNameInput = previewContainer.getByTestId("job-name-textfield"); - const duplicateJobName = "test-job-name-1"; - await jobNameInput.fill(duplicateJobName); - - const copyButton = panel.getByRole("button", { name: "Copy", exact: true }); - const expectedErrorMessage = `Duplicate job name '${duplicateJobName}'`; - await interceptAndInspectApiRequest( - page, - `${expectedSourceAccount.name}/dataTransferJobs/${duplicateJobName}`, - "PUT", - new Error(expectedErrorMessage), - (url?: string) => url?.includes(duplicateJobName) ?? false, - ); - - let errorThrown = false; - try { - await copyButton.click(); - await page.waitForTimeout(2000); - } catch (error: any) { - errorThrown = true; - expect(error.message).toContain("not allowed"); - } - if (!errorThrown) { - const errorContainer = panel.getByTestId("Panel:ErrorContainer"); - await expect(errorContainer).toBeVisible(); - await expect(errorContainer).toHaveText(new RegExp(expectedErrorMessage, "i")); - } - - await expect(panel).toBeVisible(); -}); - -test("Testing API request success with valid job name and verifying copy job creation", async () => { - const previewContainer = panel.getByTestId("Panel:PreviewCopyJob"); - const jobNameInput = previewContainer.getByTestId("job-name-textfield"); - const copyButton = panel.getByRole("button", { name: "Copy", exact: true }); - - const validJobName = expectedJobName; - - const copyJobCreationPromise = waitForApiResponse( - page, - `${expectedSourceAccount.name}/dataTransferJobs/${validJobName}`, - "PUT", - ); - - await jobNameInput.fill(validJobName); - await expect(copyButton).not.toHaveClass(/(^|\s)is-disabled(\s|$)/); - - await copyButton.click(); - - const response = await copyJobCreationPromise; - expect(response.ok()).toBe(true); - - await expect(panel).not.toBeVisible({ timeout: 10000 }); - - const jobsListContainer = wrapper.locator(".CopyJobListContainer .ms-DetailsList-contentWrapper .ms-List-page"); - await jobsListContainer.waitFor({ state: "visible" }); - - const jobItem = jobsListContainer.getByText(validJobName); - await jobItem.waitFor({ state: "visible" }); - await expect(jobItem).toBeVisible(); -}); - -test.skip("Pause a running copy job", async () => { - const jobsListContainer = wrapper.locator(".CopyJobListContainer .ms-DetailsList-contentWrapper .ms-List-page"); - await jobsListContainer.waitFor({ state: "visible" }); - - const firstJobRow = jobsListContainer.locator(".ms-DetailsRow", { hasText: expectedJobName }); - await firstJobRow.waitFor({ state: "visible" }); - - const actionMenuButton = wrapper.getByTestId(`CopyJobActionMenu/Button:${expectedJobName}`); - await actionMenuButton.waitFor({ state: "visible" }); - await actionMenuButton.click(); - - const pauseAction = frame.locator(".ms-ContextualMenu-list button:has-text('Pause')"); - await pauseAction.waitFor({ state: "visible" }); - await pauseAction.click(); - - const updatedJobRow = jobsListContainer.locator(".ms-DetailsRow").filter({ hasText: expectedJobName }); - const statusCell = updatedJobRow.locator("[data-automationid='DetailsRowCell'][data-automation-key='CopyJobStatus']"); - await expect(statusCell).toContainText(/paused/i, { timeout: 10000 }); -}); - -test.skip("Resume a paused copy job", async () => { - const jobsListContainer = wrapper.locator(".CopyJobListContainer .ms-DetailsList-contentWrapper .ms-List-page"); - await jobsListContainer.waitFor({ state: "visible" }); - - const pausedJobRow = jobsListContainer.locator(".ms-DetailsRow", { hasText: expectedJobName }); - await pausedJobRow.waitFor({ state: "visible" }); - - const statusCell = pausedJobRow.locator("[data-automationid='DetailsRowCell'][data-automation-key='CopyJobStatus']"); - await expect(statusCell).toContainText(/paused/i); - - const actionMenuButton = wrapper.getByTestId(`CopyJobActionMenu/Button:${expectedJobName}`); - await actionMenuButton.waitFor({ state: "visible" }); - await actionMenuButton.click(); - - const resumeAction = frame.locator(".ms-ContextualMenu-list button:has-text('Resume')"); - await resumeAction.waitFor({ state: "visible" }); - await resumeAction.click(); - - await expect(statusCell).toContainText(/running|queued/i); -}); - -test("Cancel a copy job", async () => { - // Create a new job specifically for cancellation testing - const cancelJobName = `cancel_test_job_${Date.now()}`; - - // Navigate to create job panel - const createCopyJobButton = wrapper.getByTestId("CommandBar/Button:Create Copy Job"); - await createCopyJobButton.click(); - panel = frame.getByTestId("Panel:Create copy job"); - - // Skip to container selection (offline mode for faster creation) - await panel.getByRole("button", { name: "Next" }).click(); - - // Select source containers quickly - const sourceDatabaseDropdown = panel.getByTestId("source-databaseDropdown"); - await sourceDatabaseDropdown.click(); - const sourceDatabaseDropdownItem = await getDropdownItemByNameOrPosition( - frame, - { position: 0 }, - { ariaLabel: "Database" }, - ); - await sourceDatabaseDropdownItem.click(); - - const sourceContainerDropdown = panel.getByTestId("source-containerDropdown"); - await sourceContainerDropdown.click(); - const sourceContainerDropdownItem = await getDropdownItemByNameOrPosition( - frame, - { position: 0 }, - { ariaLabel: "Container" }, - ); - await sourceContainerDropdownItem.click(); - - // Select target containers - const targetDatabaseDropdown = panel.getByTestId("target-databaseDropdown"); - await targetDatabaseDropdown.click(); - const targetDatabaseDropdownItem = await getDropdownItemByNameOrPosition( - frame, - { position: 0 }, - { ariaLabel: "Database" }, - ); - await targetDatabaseDropdownItem.click(); - - const targetContainerDropdown = panel.getByTestId("target-containerDropdown"); - await targetContainerDropdown.click(); - const targetContainerDropdownItem = await getDropdownItemByNameOrPosition( - frame, - { position: 1 }, - { ariaLabel: "Container" }, - ); - await targetContainerDropdownItem.click(); - - await panel.getByRole("button", { name: "Next" }).click(); - - // Set job name and create - const jobNameInput = panel.getByTestId("job-name-textfield"); - await jobNameInput.fill(cancelJobName); - - const copyButton = panel.getByRole("button", { name: "Copy", exact: true }); - - // Create job and immediately start polling for it - await copyButton.click(); - - // Wait for panel to close and job list to refresh - await expect(panel).not.toBeVisible({ timeout: 10000 }); - - const jobsListContainer = wrapper.locator(".CopyJobListContainer .ms-DetailsList-contentWrapper .ms-List-page"); - await jobsListContainer.waitFor({ state: "visible" }); - - // Rapid polling to catch the job in running state - let attempts = 0; - const maxAttempts = 50; // Try for ~5 seconds - let jobCancelled = false; - - while (attempts < maxAttempts && !jobCancelled) { - try { - // Look for the job row - const jobRow = jobsListContainer.locator(".ms-DetailsRow", { hasText: cancelJobName }); - - if (await jobRow.isVisible({ timeout: 100 })) { - const statusCell = jobRow.locator("[data-automationid='DetailsRowCell'][data-automation-key='CopyJobStatus']"); - const statusText = await statusCell.textContent({ timeout: 100 }); - - // If job is still running/queued, try to cancel it - if (statusText && /running|queued|pending/i.test(statusText)) { - const actionMenuButton = wrapper.getByTestId(`CopyJobActionMenu/Button:${cancelJobName}`); - await actionMenuButton.click({ timeout: 1000 }); - - const cancelAction = frame.locator(".ms-ContextualMenu-list button:has-text('Cancel')"); - if (await cancelAction.isVisible({ timeout: 1000 })) { - await cancelAction.click(); - - // Verify cancellation - await expect(statusCell).toContainText(/cancelled|canceled|failed/i, { timeout: 5000 }); - jobCancelled = true; - break; - } - } else if (statusText && /completed|succeeded|finished/i.test(statusText)) { - // Job completed too fast, skip the test - // console.log(`Job ${cancelJobName} completed too quickly to test cancellation`); - test.skip(true, "Job completed too quickly for cancellation test"); - return; - } - } - - // Refresh the job list - const refreshButton = wrapper.getByTestId("CommandBar/Button:Refresh"); - if (await refreshButton.isVisible({ timeout: 100 })) { - await refreshButton.click(); - await page.waitForTimeout(100); // Small delay between attempts - } - } catch (error) { - // Continue trying if there's any error - } - - attempts++; - } - - if (!jobCancelled) { - // If we couldn't cancel in time, at least verify the job was created - const jobRow = jobsListContainer.locator(".ms-DetailsRow", { hasText: cancelJobName }); - await expect(jobRow).toBeVisible({ timeout: 5000 }); - test.skip(true, "Could not catch job in running state for cancellation test"); - } -}); - test.afterAll("Container Copy - After All", async () => { await page.unroute(/.*/, (route) => route.continue()); await page.close(); diff --git a/test/sql/scaleAndSettings/changePartitionKey.spec.ts b/test/sql/scaleAndSettings/changePartitionKey.spec.ts index e1a15c3f6..95f5a957a 100644 --- a/test/sql/scaleAndSettings/changePartitionKey.spec.ts +++ b/test/sql/scaleAndSettings/changePartitionKey.spec.ts @@ -83,12 +83,12 @@ // await changePkPanel.getByLabel("Use existing container").check(); // await changePkPanel.getByText("Choose an existing container").click(); - // const containerDropdownItem = await getDropdownItemByNameOrPosition( - // explorer.frame, - // { name: newContainerId }, - // { ariaLabel: "Existing Containers" }, - // ); - // await containerDropdownItem.click(); +// const containerDropdownItem = await getDropdownItemByNameOrPosition( +// explorer.frame, +// { name: newContainerId }, +// { ariaLabel: "Existing Containers" }, +// ); +// await containerDropdownItem.click(); // await changePkPanel.getByTestId("Panel/OkButton").click(); // await explorer.frame.getByRole("button", { name: "Cancel" }).click(); diff --git a/test/testData.ts b/test/testData.ts index 9729a90b4..b440f565c 100644 --- a/test/testData.ts +++ b/test/testData.ts @@ -80,6 +80,69 @@ type createTestSqlContainerConfig = { databaseName?: string; }; +type createMultipleTestSqlContainerConfig = { + containerCount?: number; + partitionKey?: string; + databaseName?: string; + accountType: TestAccount.SQLContainerCopyOnly | TestAccount.SQL; +}; + +export async function createMultipleTestContainers({ + partitionKey = "/partitionKey", + databaseName = "", + containerCount = 1, + accountType = TestAccount.SQL, +}: createMultipleTestSqlContainerConfig): Promise { + const creationPromises: Promise[] = []; + + const databaseId = databaseName ? databaseName : generateUniqueName("db"); + const credentials = getAzureCLICredentials(); + const adaptedCredentials = new AzureIdentityCredentialAdapter(credentials); + const armClient = new CosmosDBManagementClient(adaptedCredentials, subscriptionId); + const accountName = getAccountName(accountType); + const account = await armClient.databaseAccounts.get(resourceGroupName, accountName); + + const clientOptions: CosmosClientOptions = { + endpoint: account.documentEndpoint!, + }; + + const rbacToken = + accountType === TestAccount.SQL + ? process.env.NOSQL_TESTACCOUNT_TOKEN + : accountType === TestAccount.SQLContainerCopyOnly + ? process.env.NOSQL_CONTAINERCOPY_TESTACCOUNT_TOKEN + : ""; + if (rbacToken) { + clientOptions.tokenProvider = async (): Promise => { + const AUTH_PREFIX = `type=aad&ver=1.0&sig=`; + const authorizationToken = `${AUTH_PREFIX}${rbacToken}`; + return authorizationToken; + }; + } else { + const keys = await armClient.databaseAccounts.listKeys(resourceGroupName, accountName); + clientOptions.key = keys.primaryMasterKey; + } + + const client = new CosmosClient(clientOptions); + const { database } = await client.databases.createIfNotExists({ id: databaseId }); + + try { + for (let i = 0; i < containerCount; i++) { + const containerId = `testcontainer_${Date.now()}_${Math.random().toString(36).substring(6)}_${i}`; + creationPromises.push( + database.containers.createIfNotExists({ id: containerId, partitionKey }).then(({ container }) => { + return new TestContainerContext(armClient, client, database, container, new Map()); + }), + ); + } + const contexts = await Promise.all(creationPromises); + return contexts; + } catch (e) { + await database.delete(); + throw e; + } +} + export async function createTestSQLContainer({ includeTestData = false, partitionKey = "/partitionKey",