Fix cassandra tests

2026-01-06 19:19:56 +00:00 · 2025-12-10 11:19:09 -06:00
parent 945c457bd5
commit 98a21f2fb1
12 changed files with 511 additions and 845 deletions
--- a/test/cassandra/container.spec.ts
+++ b/test/cassandra/container.spec.ts
@@ -1,50 +1,114 @@
 import { expect, test } from "@playwright/test";

 import { DataExplorer, TEST_AUTOSCALE_THROUGHPUT_RU, TestAccount, generateUniqueName } from "../fx";
+import {
+  deleteContainer,
+  deleteKeyspace,
+  openAndFillCreateCassandraTablePanel,
+} from "../helpers/containerCreationHelpers";

-test("Cassandra keyspace and table CRUD", async ({ page }) => {
-  const keyspaceId = generateUniqueName("db");
-  const tableId = "testtable"; // A unique table name isn't needed because the keyspace is unique
+test("Cassandra: Keyspace and table CRUD", async ({ page }) => {
+  const keyspaceId = generateUniqueName("keyspace");
+  const tableId = generateUniqueName("table");

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

-  await explorer.globalCommandButton("New Table").click();
-  await explorer.whilePanelOpen(
-    "Add Table",
-    async (panel, okButton) => {
-      await panel.getByPlaceholder("Type a new keyspace id").fill(keyspaceId);
-      await panel.getByPlaceholder("Enter table Id").fill(tableId);
-      await panel.getByTestId("autoscaleRUInput").fill(TEST_AUTOSCALE_THROUGHPUT_RU.toString());
-      await okButton.click();
-    },
-    { closeTimeout: 5 * 60 * 1000 },
-  );
+  // Create
+  await openAndFillCreateCassandraTablePanel(explorer, {
+    keyspaceId,
+    tableId,
+    isAutoscale: true,
+    throughputValue: TEST_AUTOSCALE_THROUGHPUT_RU,
+  });
+
+  const keyspaceNode = await explorer.waitForNode(keyspaceId);
+  const tableNode = await explorer.waitForContainerNode(keyspaceId, tableId);
+  await expect(tableNode.element).toBeAttached();
+
+  // Delete table
+  await deleteContainer(explorer, keyspaceId, tableId, "Delete Table");
+  await expect(tableNode.element).not.toBeAttached();
+
+  // Delete keyspace
+  await deleteKeyspace(explorer, keyspaceId);
+  await expect(keyspaceNode.element).not.toBeAttached();
+});
+
+test("Cassandra: New keyspace shared throughput", async ({ page }) => {
+  const keyspaceId = generateUniqueName("keyspace");
+  const tableId = generateUniqueName("table");
+
+  const explorer = await DataExplorer.open(page, TestAccount.Cassandra);
+
+  await openAndFillCreateCassandraTablePanel(explorer, {
+    keyspaceId,
+    tableId,
+    useSharedThroughput: true,
+  });

  const keyspaceNode = await explorer.waitForNode(keyspaceId);
  const tableNode = await explorer.waitForContainerNode(keyspaceId, tableId);

-  await tableNode.openContextMenu();
-  await tableNode.contextMenuItem("Delete Table").click();
-  await explorer.whilePanelOpen(
-    "Delete Table",
-    async (panel, okButton) => {
-      await panel.getByRole("textbox", { name: "Confirm by typing the table id" }).fill(tableId);
-      await okButton.click();
-    },
-    { closeTimeout: 5 * 60 * 1000 },
-  );
-  await expect(tableNode.element).not.toBeAttached();
-
-  await keyspaceNode.openContextMenu();
-  await keyspaceNode.contextMenuItem("Delete Keyspace").click();
-  await explorer.whilePanelOpen(
-    "Delete Keyspace",
-    async (panel, okButton) => {
-      await panel.getByRole("textbox", { name: "Confirm by typing the Keyspace id" }).fill(keyspaceId);
-      await okButton.click();
-    },
-    { closeTimeout: 5 * 60 * 1000 },
-  );
+  await expect(tableNode.element).toBeAttached();

+  // Cleanup
+  await deleteKeyspace(explorer, keyspaceId);
+  await expect(keyspaceNode.element).not.toBeAttached();
+});
+
+test("Cassandra: Manual throughput", async ({ page }) => {
+  const keyspaceId = generateUniqueName("keyspace");
+  const tableId = generateUniqueName("table");
+  const manualThroughput = 400;
+
+  const explorer = await DataExplorer.open(page, TestAccount.Cassandra);
+
+  await openAndFillCreateCassandraTablePanel(explorer, {
+    keyspaceId,
+    tableId,
+    isAutoscale: false,
+    throughputValue: manualThroughput,
+  });
+
+  const keyspaceNode = await explorer.waitForNode(keyspaceId);
+  const tableNode = await explorer.waitForContainerNode(keyspaceId, tableId);
+
+  await expect(tableNode.element).toBeAttached();
+
+  // Cleanup
+  await deleteKeyspace(explorer, keyspaceId);
+  await expect(keyspaceNode.element).not.toBeAttached();
+});
+
+test("Cassandra: Multiple tables in keyspace", async ({ page }) => {
+  const keyspaceId = generateUniqueName("keyspace");
+  const table1Id = generateUniqueName("table");
+  const table2Id = generateUniqueName("table");
+
+  const explorer = await DataExplorer.open(page, TestAccount.Cassandra);
+
+  // Create first table
+  await openAndFillCreateCassandraTablePanel(explorer, {
+    keyspaceId,
+    tableId: table1Id,
+    isAutoscale: true,
+    throughputValue: TEST_AUTOSCALE_THROUGHPUT_RU,
+  });
+
+  const keyspaceNode = await explorer.waitForNode(keyspaceId);
+  await explorer.waitForContainerNode(keyspaceId, table1Id);
+
+  // Create second table in same keyspace
+  await openAndFillCreateCassandraTablePanel(explorer, {
+    keyspaceId,
+    tableId: table2Id,
+    isAutoscale: true,
+    throughputValue: TEST_AUTOSCALE_THROUGHPUT_RU,
+  });
+
+  await explorer.waitForContainerNode(keyspaceId, table2Id);
+
+  // Cleanup
+  await deleteKeyspace(explorer, keyspaceId);
  await expect(keyspaceNode.element).not.toBeAttached();
 });