cosmos-explorer/test/gremlin/container.spec.ts

import { expect, test } from "@playwright/test";

import { DataExplorer, TEST_AUTOSCALE_THROUGHPUT_RU, TestAccount, generateUniqueName } from "../fx";
import {
  GREMLIN_CONFIG,
  deleteContainer,
  deleteDatabase,
  openAndFillCreateContainerPanel,
} from "../helpers/containerCreationHelpers";

test("Gremlin: Database and graph CRUD", async ({ page }) => {
  const databaseId = generateUniqueName("db");
  const graphId = generateUniqueName("graph");

  const explorer = await DataExplorer.open(page, TestAccount.Gremlin);

  // Create
  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();

  // Delete graph
  await deleteContainer(explorer, databaseId, graphId, "Delete Graph");
  await expect(graphNode.element).not.toBeAttached();

  // Delete database
  await deleteDatabase(explorer, databaseId);
  await expect(databaseNode.element).not.toBeAttached();
});

test("Gremlin: New database 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("Gremlin: 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("Gremlin: No unique keys support", async ({ page }) => {
  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());

      // Verify unique key button is not present (Gremlin-specific API limitation)
      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();
});