Added different container creation playwirhgt tests

2026-01-30 07:04:12 +00:00 · 2025-12-09 19:07:54 -06:00
parent 5b7d1a74af
commit 945c457bd5
22 changed files with 1088 additions and 71 deletions
--- a/test/gremlin/containerCreation.spec.ts
+++ b/test/gremlin/containerCreation.spec.ts
@@ -0,0 +1,135 @@
+import { expect, test } from "@playwright/test";
+
+import { DataExplorer, TEST_AUTOSCALE_THROUGHPUT_RU, TestAccount, generateUniqueName } from "../fx";
+import { GREMLIN_CONFIG, deleteDatabase, openAndFillCreateContainerPanel } from "../helpers/containerCreationHelpers";
+
+test.describe("Gremlin API - Graph Creation", () => {
+  test("Create graph in new database with non-shared throughput", async ({ page }) => {
+    const databaseId = generateUniqueName("db");
+    const graphId = generateUniqueName("graph");
+
+    const explorer = await DataExplorer.open(page, TestAccount.Gremlin);
+
+    await openAndFillCreateContainerPanel(explorer, GREMLIN_CONFIG, {
+      databaseId,
+      containerId: graphId,
+      partitionKey: "/pk",
+      isAutoscale: true,
+      throughputValue: TEST_AUTOSCALE_THROUGHPUT_RU,
+    });
+
+    const databaseNode = await explorer.waitForNode(databaseId);
+    const graphNode = await explorer.waitForContainerNode(databaseId, graphId);
+
+    await expect(graphNode.element).toBeAttached();
+
+    // Cleanup
+    await deleteDatabase(explorer, databaseId);
+    await expect(databaseNode.element).not.toBeAttached();
+  });
+
+  test("Create graph in new database with shared throughput", async ({ page }) => {
+    const databaseId = generateUniqueName("db");
+    const graphId = generateUniqueName("graph");
+
+    const explorer = await DataExplorer.open(page, TestAccount.Gremlin);
+
+    await openAndFillCreateContainerPanel(explorer, GREMLIN_CONFIG, {
+      databaseId,
+      containerId: graphId,
+      partitionKey: "/pk",
+      useSharedThroughput: true,
+    });
+
+    const databaseNode = await explorer.waitForNode(databaseId);
+    const graphNode = await explorer.waitForContainerNode(databaseId, graphId);
+
+    await expect(graphNode.element).toBeAttached();
+
+    // Cleanup
+    await deleteDatabase(explorer, databaseId);
+    await expect(databaseNode.element).not.toBeAttached();
+  });
+
+  test("Create graph with autoscale throughput", async ({ page }) => {
+    const databaseId = generateUniqueName("db");
+    const graphId = generateUniqueName("graph");
+
+    const explorer = await DataExplorer.open(page, TestAccount.Gremlin);
+
+    await openAndFillCreateContainerPanel(explorer, GREMLIN_CONFIG, {
+      databaseId,
+      containerId: graphId,
+      partitionKey: "/pk",
+      isAutoscale: true,
+      throughputValue: TEST_AUTOSCALE_THROUGHPUT_RU,
+    });
+
+    const databaseNode = await explorer.waitForNode(databaseId);
+    const graphNode = await explorer.waitForContainerNode(databaseId, graphId);
+
+    await expect(graphNode.element).toBeAttached();
+
+    // Cleanup
+    await deleteDatabase(explorer, databaseId);
+    await expect(databaseNode.element).not.toBeAttached();
+  });
+
+  test("Create graph with manual throughput", async ({ page }) => {
+    const databaseId = generateUniqueName("db");
+    const graphId = generateUniqueName("graph");
+    const manualThroughput = 400;
+
+    const explorer = await DataExplorer.open(page, TestAccount.Gremlin);
+
+    await openAndFillCreateContainerPanel(explorer, GREMLIN_CONFIG, {
+      databaseId,
+      containerId: graphId,
+      partitionKey: "/pk",
+      isAutoscale: false,
+      throughputValue: manualThroughput,
+    });
+
+    const databaseNode = await explorer.waitForNode(databaseId);
+    const graphNode = await explorer.waitForContainerNode(databaseId, graphId);
+
+    await expect(graphNode.element).toBeAttached();
+
+    // Cleanup
+    await deleteDatabase(explorer, databaseId);
+    await expect(databaseNode.element).not.toBeAttached();
+  });
+
+  test("Create graph - no unique keys support", async ({ page }) => {
+    // Gremlin doesn't support unique keys, verify panel doesn't show unique key UI
+    const databaseId = generateUniqueName("db");
+    const graphId = generateUniqueName("graph");
+
+    const explorer = await DataExplorer.open(page, TestAccount.Gremlin);
+
+    await explorer.globalCommandButton("New Graph").click();
+    await explorer.whilePanelOpen(
+      "New Graph",
+      async (panel, okButton) => {
+        await panel.getByTestId("AddCollectionPanel/DatabaseId").fill(databaseId);
+        await panel.getByTestId("AddCollectionPanel/CollectionId").fill(graphId);
+        await panel.getByRole("textbox", { name: "Partition key" }).fill("/pk");
+        await panel.getByTestId("ThroughputInput/AutoscaleRUInput").fill(TEST_AUTOSCALE_THROUGHPUT_RU.toString());
+
+        const uniqueKeyButton = panel.getByTestId("AddCollectionPanel/AddUniqueKeyButton");
+        await expect(uniqueKeyButton).not.toBeVisible();
+
+        await okButton.click();
+      },
+      { closeTimeout: 5 * 60 * 1000 },
+    );
+
+    const databaseNode = await explorer.waitForNode(databaseId);
+    const graphNode = await explorer.waitForContainerNode(databaseId, graphId);
+    await expect(graphNode.element).toBeAttached();
+
+    // Cleanup
+    await deleteDatabase(explorer, databaseId);
+    await expect(databaseNode.element).not.toBeAttached();
+  });
+});