From c766f3c0547d2a56edb0d7aa67f849f218e9224e Mon Sep 17 00:00:00 2001
From: dxlin <dxlin@users.noreply.github.com>
Date: Mon, 28 Jul 2025 01:35:50 -0400
Subject: [PATCH] test(SegmentationPanel): add test for jump to center of
 segments from PACS and basic add, rename, delete from panel (#5262)

---
 tests/SegmentationPanel.spec.ts | 97 +++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 tests/SegmentationPanel.spec.ts

diff --git a/tests/SegmentationPanel.spec.ts b/tests/SegmentationPanel.spec.ts
new file mode 100644
index 000000000..ca2f39145
--- /dev/null
+++ b/tests/SegmentationPanel.spec.ts
@@ -0,0 +1,97 @@
+import { test, expect } from 'playwright-test-coverage';
+import { visitStudy } from './utils';
+
+test.beforeEach(async ({ page }) => {
+  // Using same one as JumpToMeasurementMPR.spec.ts
+  const studyInstanceUID = '1.3.6.1.4.1.14519.5.2.1.256467663913010332776401703474716742458';
+  const mode = 'segmentation'; // To also test add/remove
+  await visitStudy(page, studyInstanceUID, mode, 2000);
+});
+
+test('checks basic add, rename, delete segments from panel', async ({ page }) => {
+  // Segmentation Panel should already be open
+  const segmentationPanel = page.getByTestId('panelSegmentationWithTools-btn');
+  await expect(segmentationPanel).toBeVisible();
+
+  // Add segmentation
+  const addSegmentationBtn = page.getByTestId('addSegmentation');
+  await addSegmentationBtn.click();
+
+  // Expect new segmentation and blank segment named "Segment 1"
+  await expect(page.getByTestId('data-row')).toHaveCount(1);
+  await expect(page.getByTestId('data-row')).toContainText('Segment 1');
+
+  // Rename
+  const segment1Dropdown = page
+    .getByTestId('data-row')
+    .first()
+    .getByRole('button', { name: 'Actions' });
+  await segment1Dropdown.click();
+
+  const renameButton = page.getByRole('menuitem', { name: 'Rename' });
+  await expect(renameButton).toBeVisible();
+  await renameButton.click();
+
+  const renameDialog = page.getByRole('dialog', { name: 'Edit Segment Label' });
+  await expect(renameDialog).toBeVisible();
+
+  const renameInput = page.getByRole('textbox', { name: 'Enter new label' });
+  await renameInput.fill('Segment One');
+
+  await page.getByTestId('input-dialog-save-button').click();
+
+  await expect(page.getByTestId('data-row')).toContainText('Segment One');
+  await expect(page.getByTestId('data-row')).not.toContainText('Segment 1');
+
+  // Delete
+  await segment1Dropdown.click();
+  const deleteButton = page.getByRole('menuitem', { name: 'Delete' });
+  await expect(deleteButton).toBeVisible();
+  await deleteButton.click();
+
+  await expect(page.getByTestId('data-row')).toHaveCount(0);
+});
+
+test('checks saved segmentations loads and jumps to slices', async ({ page }) => {
+  const viewportInfoBottomRight = page.getByTestId('viewport-overlay-bottom-right');
+
+  // Image loads on slice 1, confirm on slice 1
+  await expect(viewportInfoBottomRight).toContainText('1/', { timeout: 10000 });
+
+  // Add Segmentations
+  const segmentationsThumbnail = page
+    .getByTestId('study-browser-thumbnail-no-image')
+    .getByRole('button', { name: 'Segmentation' });
+  await segmentationsThumbnail.dblclick();
+
+  await page.waitForTimeout(3000);
+
+  // Confirm open segmentation
+  const viewportNotification = page.getByTestId('viewport-notification');
+  await expect(viewportNotification).toBeVisible();
+  await page.getByTestId('yes-hydrate-btn').click();
+
+  // Segmentation Panel should already be open
+  const segmentationPanel = page.getByTestId('panelSegmentationWithTools-btn');
+  await expect(segmentationPanel).toBeVisible();
+
+  // Confirm spleen jumps to slice 17
+  // First iteration repeat to account for segmentation loading delays
+  const spleenRow = page.getByTestId('data-row').filter({ hasText: 'Spleen' });
+  await expect(async () => {
+    await spleenRow.click();
+    await expect(viewportInfoBottomRight).toContainText('17/');
+  }).toPass({
+    timeout: 10000,
+  });
+
+  // Esophagus - 5
+  const esophagusRow = page.getByTestId('data-row').filter({ hasText: 'Esophagus' });
+  await esophagusRow.click();
+  await expect(viewportInfoBottomRight).toContainText('5/');
+
+  // Pancreas - 22
+  const pancreasRow = page.getByTestId('data-row').filter({ hasText: 'Pancreas' });
+  await pancreasRow.click();
+  await expect(viewportInfoBottomRight).toContainText('22/');
+});