Merge branch 'master' of https://github.com/Azure/cosmos-explorer into users/aisayas/default-throughput-bucket

test/sql/scaleAndSettings/changePartitionKey.spec.ts

@@ -23,12 +23,9 @@ test.describe("Change Partition Key", () => {
     await PartitionKeyTab.click();
   });
 
-  // Delete database only if not running in CI
-  if (!process.env.CI) {
-    test.afterEach("Delete Test Database", async () => {
-      await context?.dispose();
-    });
-  }
+  test.afterEach("Delete Test Database", async () => {
+    await context?.dispose();
+  });
 
   test("Change partition key path", async ({ page }) => {
     await expect(explorer.frame.getByText("/partitionKey")).toBeVisible();

test/sql/scaleAndSettings/containerPolicies/vectorPolicy.spec.ts

@@ -0,0 +1,186 @@
+import { expect, test } from "@playwright/test";
+import { CommandBarButton, DataExplorer, ONE_MINUTE_MS, TestAccount } from "../../../fx";
+import { createTestSQLContainer, TestContainerContext } from "../../../testData";
+
+test.describe("Vector Policy under Scale & Settings", () => {
+  let context: TestContainerContext = null!;
+  let explorer: DataExplorer = null!;
+
+  test.beforeAll("Create Test Database", async () => {
+    context = await createTestSQLContainer();
+  });
+
+  test.beforeEach("Open Container Policy tab under Scale & Settings", async ({ page }) => {
+    explorer = await DataExplorer.open(page, TestAccount.SQL);
+    await explorer.waitForContainerNode(context.database.id, context.container.id);
+
+    // Click Scale & Settings and open Container Policy tab
+    await explorer.openScaleAndSettings(context);
+    const containerPolicyTab = explorer.frame.getByTestId("settings-tab-header/ContainerVectorPolicyTab");
+    await containerPolicyTab.click();
+
+    // Click on Vector Policy tab
+    const vectorPolicyTab = explorer.frame.getByRole("tab", { name: "Vector Policy" });
+    await vectorPolicyTab.click();
+  });
+
+  test.afterAll("Delete Test Database", async () => {
+    await context?.dispose();
+  });
+
+  test("Add new vector embedding policy", async () => {
+    // Click Add vector embedding button
+    const addButton = explorer.frame.locator("#add-vector-policy");
+    await addButton.click();
+
+    // Fill in path
+    const pathInput = explorer.frame.locator("#vector-policy-path-1");
+    await pathInput.fill("/embedding");
+
+    // Fill in dimensions
+    const dimensionsInput = explorer.frame.locator("#vector-policy-dimension-1");
+    await dimensionsInput.fill("1500");
+
+    // Save changes
+    const saveButton = explorer.commandBarButton(CommandBarButton.Save);
+    await expect(saveButton).toBeEnabled();
+    await saveButton.click();
+    await expect(explorer.getConsoleHeaderStatus()).toContainText(
+      `Successfully updated container ${context.container.id}`,
+      {
+        timeout: 2 * ONE_MINUTE_MS,
+      },
+    );
+  });
+
+  test("Existing vector embedding policy fields are disabled", async () => {
+    // First add a vector embedding policy
+    const addButton = explorer.frame.locator("#add-vector-policy");
+    await addButton.click();
+
+    const pathInput = explorer.frame.locator("#vector-policy-path-1");
+    await pathInput.fill("/existingEmbedding");
+
+    const dimensionsInput = explorer.frame.locator("#vector-policy-dimension-1");
+    await dimensionsInput.fill("700");
+
+    // Save the policy
+    const saveButton = explorer.commandBarButton(CommandBarButton.Save);
+    await saveButton.click();
+    await expect(explorer.getConsoleHeaderStatus()).toContainText(
+      `Successfully updated container ${context.container.id}`,
+      {
+        timeout: 2 * ONE_MINUTE_MS,
+      },
+    );
+
+    // Verify the path field is disabled for the existing policy
+    const existingPathInput = explorer.frame.locator("#vector-policy-path-1");
+    await expect(existingPathInput).toBeDisabled();
+
+    // Verify the dimensions field is disabled for the existing policy
+    const existingDimensionsInput = explorer.frame.locator("#vector-policy-dimension-1");
+    await expect(existingDimensionsInput).toBeDisabled();
+  });
+
+  test("New vector embedding policy fields are enabled while existing are disabled", async () => {
+    // First, create an existing policy
+    const addButton = explorer.frame.locator("#add-vector-policy");
+    await addButton.click();
+
+    const firstPathInput = explorer.frame.locator("#vector-policy-path-1");
+    await firstPathInput.fill("/existingPolicy");
+
+    const firstDimensionsInput = explorer.frame.locator("#vector-policy-dimension-1");
+    await firstDimensionsInput.fill("500");
+
+    // Save the policy to make it "existing"
+    const saveButton = explorer.commandBarButton(CommandBarButton.Save);
+    await saveButton.click();
+    await expect(explorer.getConsoleHeaderStatus()).toContainText(
+      `Successfully updated container ${context.container.id}`,
+      {
+        timeout: 2 * ONE_MINUTE_MS,
+      },
+    );
+
+    // Now add a new policy
+    await addButton.click();
+
+    // Verify the existing policy fields are disabled
+    const existingPathInput = explorer.frame.locator("#vector-policy-path-1");
+    const existingDimensionsInput = explorer.frame.locator("#vector-policy-dimension-1");
+    await expect(existingPathInput).toBeDisabled();
+    await expect(existingDimensionsInput).toBeDisabled();
+
+    // Verify the new policy fields are enabled
+    const newPathInput = explorer.frame.locator("#vector-policy-path-2");
+    const newDimensionsInput = explorer.frame.locator("#vector-policy-dimension-2");
+    await expect(newPathInput).toBeEnabled();
+    await expect(newDimensionsInput).toBeEnabled();
+  });
+
+  test("Delete existing vector embedding policy", async () => {
+    // First add a vector embedding policy
+    const addButton = explorer.frame.locator("#add-vector-policy");
+    await addButton.click();
+
+    const pathInput = explorer.frame.locator("#vector-policy-path-1");
+    await pathInput.fill("/toBeDeleted");
+
+    const dimensionsInput = explorer.frame.locator("#vector-policy-dimension-1");
+    await dimensionsInput.fill("256");
+
+    // Save the policy
+    const saveButton = explorer.commandBarButton(CommandBarButton.Save);
+    await saveButton.click();
+    await expect(explorer.getConsoleHeaderStatus()).toContainText(
+      `Successfully updated container ${context.container.id}`,
+      {
+        timeout: 2 * ONE_MINUTE_MS,
+      },
+    );
+
+    // Verify the policy exists
+    await expect(pathInput).toBeVisible();
+
+    // Click the delete (trash) button for the vector embedding
+    const deleteButton = explorer.frame.locator("#delete-Vector-embedding-1");
+    await expect(deleteButton).toBeEnabled();
+    await deleteButton.click();
+
+    // Verify the policy fields are removed
+    await expect(explorer.frame.locator("#vector-policy-path-1")).not.toBeVisible();
+    await expect(explorer.frame.locator("#vector-policy-dimension-1")).not.toBeVisible();
+
+    // Save the deletion
+    await expect(saveButton).toBeEnabled();
+    await saveButton.click();
+    await expect(explorer.getConsoleHeaderStatus()).toContainText(
+      `Successfully updated container ${context.container.id}`,
+      {
+        timeout: 2 * ONE_MINUTE_MS,
+      },
+    );
+
+    // Verify the policy is still gone after save
+    await expect(explorer.frame.locator("#vector-policy-path-1")).not.toBeVisible();
+  });
+
+  test("Validation error for empty path", async () => {
+    const addButton = explorer.frame.locator("#add-vector-policy");
+    await addButton.click();
+
+    // Leave path empty, just fill dimensions
+    const dimensionsInput = explorer.frame.locator("#vector-policy-dimension-1");
+    await dimensionsInput.fill("512");
+
+    // Check for validation error on path
+    const pathError = explorer.frame.locator("text=Path should not be empty");
+    await expect(pathError).toBeVisible();
+
+    // Verify save button is disabled due to validation error
+    const saveButton = explorer.commandBarButton(CommandBarButton.Save);
+    await expect(saveButton).toBeDisabled();
+  });
+});

test/sql/scaleAndSettings/dataMasking.spec.ts

@@ -0,0 +1,127 @@
+import { expect, test, type Page } from "@playwright/test";
+import { DataExplorer, TestAccount } from "../../fx";
+import { createTestSQLContainer, TestContainerContext } from "../../testData";
+
+/**
+ * Tests for Dynamic Data Masking (DDM) feature.
+ *
+ * Prerequisites:
+ * - Test account must have the EnableDynamicDataMasking capability enabled
+ * - If the capability is not enabled, the DataMaskingTab will not be visible and tests will be skipped
+ *
+ * Important Notes:
+ * - Tests focus on enabling DDM and modifying the masking policy configuration
+ */
+
+let testContainer: TestContainerContext;
+let DATABASE_ID: string;
+let CONTAINER_ID: string;
+
+test.beforeAll(async () => {
+  testContainer = await createTestSQLContainer();
+  DATABASE_ID = testContainer.database.id;
+  CONTAINER_ID = testContainer.container.id;
+});
+
+// Clean up test database after all tests
+test.afterAll(async () => {
+  if (testContainer) {
+    await testContainer.dispose();
+  }
+});
+
+// Helper function to navigate to Data Masking tab
+async function navigateToDataMaskingTab(page: Page, explorer: DataExplorer): Promise<boolean> {
+  // Refresh the tree to see the newly created database
+  const refreshButton = explorer.frame.getByTestId("Sidebar/RefreshButton");
+  await refreshButton.click();
+  await page.waitForTimeout(3000);
+
+  // Expand database and container nodes
+  const databaseNode = await explorer.waitForNode(DATABASE_ID);
+  await databaseNode.expand();
+  await page.waitForTimeout(2000);
+
+  const containerNode = await explorer.waitForNode(`${DATABASE_ID}/${CONTAINER_ID}`);
+  await containerNode.expand();
+  await page.waitForTimeout(1000);
+
+  // Click Scale & Settings or Settings (depending on container type)
+  let settingsNode = explorer.frame.getByTestId(`TreeNode:${DATABASE_ID}/${CONTAINER_ID}/Scale & Settings`);
+  const isScaleAndSettings = await settingsNode.isVisible().catch(() => false);
+
+  if (!isScaleAndSettings) {
+    settingsNode = explorer.frame.getByTestId(`TreeNode:${DATABASE_ID}/${CONTAINER_ID}/Settings`);
+  }
+
+  await settingsNode.click();
+  await page.waitForTimeout(2000);
+
+  // Check if Data Masking tab is available
+  const dataMaskingTab = explorer.frame.getByTestId("settings-tab-header/DataMaskingTab");
+  const isTabVisible = await dataMaskingTab.isVisible().catch(() => false);
+
+  if (!isTabVisible) {
+    return false;
+  }
+
+  await dataMaskingTab.click();
+  await page.waitForTimeout(1000);
+  return true;
+}
+
+test.describe("Data Masking under Scale & Settings", () => {
+  test("Data Masking tab should be visible and show JSON editor", async ({ page }) => {
+    const explorer = await DataExplorer.open(page, TestAccount.SQL);
+    const isTabAvailable = await navigateToDataMaskingTab(page, explorer);
+
+    if (!isTabAvailable) {
+      test.skip(
+        true,
+        "Data Masking tab is not available. Test account may not have EnableDynamicDataMasking capability.",
+      );
+    }
+
+    // Verify the Data Masking editor is visible
+    const dataMaskingEditor = explorer.frame.locator(".settingsV2Editor");
+    await expect(dataMaskingEditor).toBeVisible();
+  });
+
+  test("Data Masking editor should contain default policy structure", async ({ page }) => {
+    const explorer = await DataExplorer.open(page, TestAccount.SQL);
+    const isTabAvailable = await navigateToDataMaskingTab(page, explorer);
+
+    if (!isTabAvailable) {
+      test.skip(
+        true,
+        "Data Masking tab is not available. Test account may not have EnableDynamicDataMasking capability.",
+      );
+    }
+
+    // Verify the editor contains the expected JSON structure fields
+    const editorContent = explorer.frame.locator(".settingsV2Editor");
+    await expect(editorContent).toBeVisible();
+
+    // Check that the editor contains key policy fields (default policy has empty arrays)
+    await expect(editorContent).toContainText("includedPaths");
+    await expect(editorContent).toContainText("excludedPaths");
+  });
+
+  test("Data Masking editor should have correct default policy values", async ({ page }) => {
+    const explorer = await DataExplorer.open(page, TestAccount.SQL);
+    const isTabAvailable = await navigateToDataMaskingTab(page, explorer);
+
+    if (!isTabAvailable) {
+      test.skip(
+        true,
+        "Data Masking tab is not available. Test account may not have EnableDynamicDataMasking capability.",
+      );
+    }
+
+    const editorContent = explorer.frame.locator(".settingsV2Editor");
+    await expect(editorContent).toBeVisible();
+
+    // Default policy should have empty includedPaths and excludedPaths arrays
+    await expect(editorContent).toContainText("[]");
+  });
+});

test/sql/scaleAndSettings/scale.spec.ts

@@ -118,7 +118,5 @@ async function openScaleTab(browser: Browser): Promise<SetupResult> {
 }
 
 async function cleanup({ context }: Partial<SetupResult>) {
-  if (!process.env.CI) {
-    await context?.dispose();
-  }
+  await context?.dispose();
 }

test/sql/scaleAndSettings/settings.spec.ts

@@ -17,12 +17,9 @@ test.describe("Settings under Scale & Settings", () => {
     await settingsTab.click();
   });
 
-  // Delete database only if not running in CI
-  if (!process.env.CI) {
-    test.afterAll("Delete Test Database", async () => {
-      await context?.dispose();
-    });
-  }
+  test.afterAll("Delete Test Database", async () => {
+    await context?.dispose();
+  });
 
   test("Update TTL to On (no default)", async () => {
     const ttlOnNoDefaultRadioButton = explorer.frame.getByRole("radio", { name: "ttl-on-no-default-option" });

test/sql/scaleAndSettings/sharedThroughput.spec.ts

@@ -0,0 +1,229 @@
+import { Locator, expect, test } from "@playwright/test";
+import {
+  CommandBarButton,
+  DataExplorer,
+  ONE_MINUTE_MS,
+  TEST_AUTOSCALE_MAX_THROUGHPUT_RU_4K,
+  TEST_MANUAL_THROUGHPUT_RU,
+  TestAccount,
+} from "../../fx";
+import { TestDatabaseContext, createTestDB } from "../../testData";
+
+test.describe("Database with Shared Throughput", () => {
+  let dbContext: TestDatabaseContext = null!;
+  let explorer: DataExplorer = null!;
+  const containerId = "sharedcontainer";
+
+  // Helper methods
+  const getThroughputInput = (type: "manual" | "autopilot"): Locator => {
+    return explorer.frame.getByTestId(`${type}-throughput-input`);
+  };
+
+  test.afterEach("Delete Test Database", async () => {
+    await dbContext?.dispose();
+  });
+
+  test.describe("Manual Throughput Tests", () => {
+    test.beforeEach(async ({ page }) => {
+      explorer = await DataExplorer.open(page, TestAccount.SQL);
+    });
+
+    test("Create database with shared manual throughput and verify Scale node in UI", async () => {
+      test.setTimeout(120000); // 2 minutes timeout
+      // Create database with shared manual throughput (400 RU/s)
+      dbContext = await createTestDB({ throughput: 400 });
+
+      // Verify database node appears in the tree
+      const databaseNode = await explorer.waitForNode(dbContext.database.id);
+      expect(databaseNode).toBeDefined();
+
+      // Expand the database node to see child nodes
+      await databaseNode.expand();
+
+      // Verify that "Scale" node appears under the database
+      const scaleNode = await explorer.waitForNode(`${dbContext.database.id}/Scale`);
+      expect(scaleNode).toBeDefined();
+      await expect(scaleNode.element).toBeVisible();
+    });
+
+    test("Add container to shared database without dedicated throughput", async () => {
+      // Create database with shared manual throughput
+      dbContext = await createTestDB({ throughput: 400 });
+
+      // Wait for the database to appear in the tree
+      await explorer.waitForNode(dbContext.database.id);
+
+      // Add a container to the shared database via UI
+      const newContainerButton = await explorer.globalCommandButton("New Container");
+      await newContainerButton.click();
+
+      await explorer.whilePanelOpen(
+        "New Container",
+        async (panel, okButton) => {
+          // Select "Use existing" database
+          const useExistingRadio = panel.getByRole("radio", { name: /Use existing/i });
+          await useExistingRadio.click();
+
+          // Select the database from dropdown using the new data-testid
+          const databaseDropdown = panel.getByRole("combobox", { name: "Choose an existing database" });
+          await databaseDropdown.click();
+
+          await explorer.frame.getByRole("option", { name: dbContext.database.id }).click();
+          // Now you can target the specific database option by its data-testid
+          //await panel.getByTestId(`database-option-${dbContext.database.id}`).click();
+          // Fill container id
+          await panel.getByRole("textbox", { name: "Container id, Example Container1" }).fill(containerId);
+
+          // Fill partition key
+          await panel.getByRole("textbox", { name: "Partition key" }).fill("/pk");
+
+          // Ensure "Provision dedicated throughput" is NOT checked
+          const dedicatedThroughputCheckbox = panel.getByRole("checkbox", {
+            name: /Provision dedicated throughput for this container/i,
+          });
+
+          if (await dedicatedThroughputCheckbox.isVisible()) {
+            const isChecked = await dedicatedThroughputCheckbox.isChecked();
+            if (isChecked) {
+              await dedicatedThroughputCheckbox.uncheck();
+            }
+          }
+
+          await okButton.click();
+        },
+        { closeTimeout: 5 * ONE_MINUTE_MS },
+      );
+
+      // Verify container was created under the database
+      const containerNode = await explorer.waitForContainerNode(dbContext.database.id, containerId);
+      expect(containerNode).toBeDefined();
+    });
+
+    test("Scale shared database manual throughput", async () => {
+      // Create database with shared manual throughput (400 RU/s)
+      dbContext = await createTestDB({ throughput: 400 });
+
+      // Navigate to the scale settings by clicking the "Scale" node in the tree
+      const databaseNode = await explorer.waitForNode(dbContext.database.id);
+      await databaseNode.expand();
+      const scaleNode = await explorer.waitForNode(`${dbContext.database.id}/Scale`);
+      await scaleNode.element.click();
+
+      // Update manual throughput from 400 to 800
+      await getThroughputInput("manual").fill(TEST_MANUAL_THROUGHPUT_RU.toString());
+
+      // Save changes
+      await explorer.commandBarButton(CommandBarButton.Save).click();
+
+      // Verify success message
+      await expect(explorer.getConsoleHeaderStatus()).toContainText(
+        `Successfully updated offer for database ${dbContext.database.id}`,
+        {
+          timeout: 2 * ONE_MINUTE_MS,
+        },
+      );
+    });
+
+    test("Scale shared database from manual to autoscale", async () => {
+      // Create database with shared manual throughput (400 RU/s)
+      dbContext = await createTestDB({ throughput: 400 });
+
+      // Open database settings by clicking the "Scale" node
+      const databaseNode = await explorer.waitForNode(dbContext.database.id);
+      await databaseNode.expand();
+      const scaleNode = await explorer.waitForNode(`${dbContext.database.id}/Scale`);
+      await scaleNode.element.click();
+
+      // Switch to Autoscale
+      const autoscaleRadio = explorer.frame.getByText("Autoscale", { exact: true });
+      await autoscaleRadio.click();
+
+      // Set autoscale max throughput to 1000
+      //await getThroughputInput("autopilot").fill(TEST_AUTOSCALE_THROUGHPUT_RU.toString());
+
+      // Save changes
+      await explorer.commandBarButton(CommandBarButton.Save).click();
+
+      await expect(explorer.getConsoleHeaderStatus()).toContainText(
+        `Successfully updated offer for database ${dbContext.database.id}`,
+        {
+          timeout: 2 * ONE_MINUTE_MS,
+        },
+      );
+    });
+  });
+
+  test.describe("Autoscale Throughput Tests", () => {
+    test.beforeEach(async ({ page }) => {
+      explorer = await DataExplorer.open(page, TestAccount.SQL);
+    });
+
+    test("Create database with shared autoscale throughput and verify Scale node in UI", async () => {
+      test.setTimeout(120000); // 2 minutes timeout
+
+      // Create database with shared autoscale throughput (max 1000 RU/s)
+      dbContext = await createTestDB({ maxThroughput: 1000 });
+
+      // Verify database node appears
+      const databaseNode = await explorer.waitForNode(dbContext.database.id);
+      expect(databaseNode).toBeDefined();
+
+      // Expand the database node to see child nodes
+      await databaseNode.expand();
+
+      // Verify that "Scale" node appears under the database
+      const scaleNode = await explorer.waitForNode(`${dbContext.database.id}/Scale`);
+      expect(scaleNode).toBeDefined();
+      await expect(scaleNode.element).toBeVisible();
+    });
+
+    test("Scale shared database autoscale throughput", async () => {
+      // Create database with shared autoscale throughput (max 1000 RU/s)
+      dbContext = await createTestDB({ maxThroughput: 1000 });
+
+      // Open database settings
+      const databaseNode = await explorer.waitForNode(dbContext.database.id);
+      await databaseNode.expand();
+      const scaleNode = await explorer.waitForNode(`${dbContext.database.id}/Scale`);
+      await scaleNode.element.click();
+
+      // Update autoscale max throughput from 1000 to 4000
+      await getThroughputInput("autopilot").fill(TEST_AUTOSCALE_MAX_THROUGHPUT_RU_4K.toString());
+
+      // Save changes
+      await explorer.commandBarButton(CommandBarButton.Save).click();
+
+      // Verify success message
+      await expect(explorer.getConsoleHeaderStatus()).toContainText(
+        `Successfully updated offer for database ${dbContext.database.id}`,
+        {
+          timeout: 2 * ONE_MINUTE_MS,
+        },
+      );
+    });
+
+    test("Scale shared database from autoscale to manual", async () => {
+      // Create database with shared autoscale throughput (max 1000 RU/s)
+      dbContext = await createTestDB({ maxThroughput: 1000 });
+
+      // Open database settings
+      const databaseNode = await explorer.waitForNode(dbContext.database.id);
+      await databaseNode.expand();
+      const scaleNode = await explorer.waitForNode(`${dbContext.database.id}/Scale`);
+      await scaleNode.element.click();
+
+      // Switch to Manual
+      const manualRadio = explorer.frame.getByText("Manual", { exact: true });
+      await manualRadio.click();
+
+      // Save changes
+      await explorer.commandBarButton(CommandBarButton.Save).click();
+
+      // Verify success message
+      await expect(explorer.getConsoleHeaderStatus()).toContainText(
+        `Successfully updated offer for database ${dbContext.database.id}`,
+        { timeout: 2 * ONE_MINUTE_MS },
+      );
+    });
+  });
+});