Add tests for menu opening

nebari-dev · May 20, 2024 · 3291ff2 · 3291ff2
1 parent 9bcc716
commit 3291ff2
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/src/components/table.tsx b/src/components/table.tsx
@@ -315,7 +315,9 @@ export function KernelTable(props: {
         onContextMenu={(event: React.MouseEvent) => {
           event.preventDefault();
           const contextMenu = new MenuSvg({ commands: props.commands });
+          contextMenu.addClass('jp-NewLauncher-contextMenu');
           const columnsSubMenu = new MenuSvg({ commands: props.commands });
+          columnsSubMenu.addClass('jp-NewLauncher-contextMenu-visibleColumns');
           for (const column of columns) {
             columnsSubMenu.addItem({
               command: CommandIDs.toggleColumn,

diff --git a/ui-tests/tests/jupyterlab_new_launcher.spec.ts b/ui-tests/tests/jupyterlab_new_launcher.spec.ts
@@ -7,6 +7,29 @@ test.describe('Default settings', () => {
     const launcher = page.locator('.jp-LauncherBody');
     expect(await launcher.screenshot()).toMatchSnapshot('launcher.png');
   });
+
+  test('should open table context menu', async ({ page }) => {
+    const tableHead = page.locator(
+      '.jp-Launcher-launchNotebook thead > tr > th[data-id="star"]'
+    );
+    await tableHead.click({ button: 'right' });
+    const contextMenu = page.locator('.jp-NewLauncher-contextMenu');
+    await expect(contextMenu).toBeVisible();
+    expect
+      .soft(await contextMenu.screenshot())
+      .toMatchSnapshot('table-context-menu.png');
+    const visibleColumnsEntry = page.locator(
+      '.jp-NewLauncher-contextMenu li >> text="Visible Columns"'
+    );
+    await visibleColumnsEntry.hover();
+    const columnsContextMenu = page.locator(
+      '.jp-NewLauncher-contextMenu-visibleColumns'
+    );
+    await expect(columnsContextMenu).toBeVisible();
+    expect
+      .soft(await columnsContextMenu.screenshot())
+      .toMatchSnapshot('table-context-menu-visible-columns.png');
+  });
 });
 
 test.describe('With starred section', () => {