@atlaskit/editor-plugin-table 5.3.22 → 5.3.24

package/src/__tests__/unit/layout.ts

@@ -9,6 +9,7 @@ import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
 import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import type { PluginKey } from '@atlaskit/editor-prosemirror/state';
 import { findTable } from '@atlaskit/editor-tables/utils';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -24,7 +25,6 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../plugins/table-plugin';
 import { toggleTableLayout } from '../../plugins/table/commands';
 import { getPluginState } from '../../plugins/table/pm-plugins/plugin-factory';
 import { pluginKey as tablePluginKey } from '../../plugins/table/pm-plugins/plugin-key';
@@ -32,6 +32,7 @@ import type {
   PermittedLayoutsDescriptor,
   TablePluginState,
 } from '../../plugins/table/types';
+import tablePlugin from '../../plugins/table-plugin';
 
 describe('table toolbar', () => {
   const tableOptions = {

package/src/__tests__/unit/nodeviews/TableComponent.tsx

@@ -27,7 +27,6 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import * as commands from '../../../plugins/table/commands';
 import {
   hoverTable,
@@ -37,6 +36,7 @@ import TableComponent from '../../../plugins/table/nodeviews/TableComponent';
 import { pluginKey } from '../../../plugins/table/pm-plugins/plugin-key';
 import type { TablePluginState } from '../../../plugins/table/types';
 import { TableCssClassName as ClassName } from '../../../plugins/table/types';
+import tablePlugin from '../../../plugins/table-plugin';
 
 jest.mock('../../../plugins/table/utils/nodes', () =>
   Object.assign({}, jest.requireActual('../../../plugins/table/utils/nodes'), {

package/src/__tests__/unit/nodeviews/TableContainer.tsx

@@ -23,7 +23,6 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import {
   ResizableTableContainer,
   TableContainer,
@@ -31,6 +30,7 @@ import {
 import { pluginKey } from '../../../plugins/table/pm-plugins/plugin-key';
 import { pluginKey as tableResizingPluginKey } from '../../../plugins/table/pm-plugins/table-width';
 import type { TablePluginState } from '../../../plugins/table/types';
+import tablePlugin from '../../../plugins/table-plugin';
 
 const mockStartMeasure = jest.fn();
 const mockEndMeasure = jest.fn(() => {

package/src/__tests__/unit/nodeviews/cell.ts

@@ -15,6 +15,7 @@ import {
   findCellClosestToPos,
   setCellAttrs,
 } from '@atlaskit/editor-tables/utils';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -31,10 +32,10 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import TableCell from '../../../plugins/table/nodeviews/TableCell';
 import { pluginKey } from '../../../plugins/table/pm-plugins/plugin-key';
 import type { PluginConfig } from '../../../plugins/table/types';
+import tablePlugin from '../../../plugins/table-plugin';
 
 jest.mock('@atlaskit/editor-common/utils', () => ({
   ...jest.requireActual<Object>('@atlaskit/editor-common/utils'),

package/src/__tests__/unit/nodeviews/table.ts

@@ -9,6 +9,7 @@ import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
 import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import type { Node as PMNode } from '@atlaskit/editor-prosemirror/model';
 import type { EditorView } from '@atlaskit/editor-prosemirror/view';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {

package/src/__tests__/unit/pm-plugins/decorations/column-resizing.ts

@@ -11,6 +11,7 @@ import type { ContentNodeWithPos } from '@atlaskit/editor-prosemirror/utils';
 import type { EditorView } from '@atlaskit/editor-prosemirror/view';
 import { DecorationSet } from '@atlaskit/editor-prosemirror/view';
 import { getCellsInColumn } from '@atlaskit/editor-tables/utils';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {

package/src/__tests__/unit/pm-plugins/decorations/plugin.ts

@@ -9,6 +9,7 @@ import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import { Selection } from '@atlaskit/editor-prosemirror/state';
 import { DecorationSet } from '@atlaskit/editor-prosemirror/view';
 import { addColumnAt } from '@atlaskit/editor-tables/utils';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {

package/src/__tests__/unit/pm-plugins/main.ts

@@ -10,6 +10,7 @@ import { undo } from '@atlaskit/editor-prosemirror/history';
 import type { NodeType } from '@atlaskit/editor-prosemirror/model';
 import type { Selection } from '@atlaskit/editor-prosemirror/state';
 import * as pmUtils from '@atlaskit/editor-prosemirror/utils';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {

package/src/__tests__/unit/pm-plugins/safari-delete-composition-text-issue-workaround.ts

@@ -6,6 +6,7 @@ import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
 import { guidelinePlugin } from '@atlaskit/editor-plugin-guideline';
 import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
 import { widthPlugin } from '@atlaskit/editor-plugin-width';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -21,8 +22,8 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import { pluginKey } from '../../../plugins/table/pm-plugins/plugin-key';
+import tablePlugin from '../../../plugins/table-plugin';
 
 describe('table/safari-delete-composition-text-issue-workaround', () => {
   let editor: any;

package/src/__tests__/unit/pm-plugins/sticky-headers/tableRow.tsx

@@ -16,6 +16,7 @@ import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
 import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import type { Node as ProseMirrorNode } from '@atlaskit/editor-prosemirror/model';
 import type { EditorView } from '@atlaskit/editor-prosemirror/view';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {

package/src/__tests__/unit/pm-plugins/table-resizing/event-handlers.ts

@@ -19,6 +19,7 @@ import {
   akEditorDefaultLayoutWidth,
   akEditorFullPageMaxWidth,
 } from '@atlaskit/editor-shared-styles/consts';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {

package/src/__tests__/unit/pm-plugins/table-width.ts

@@ -8,6 +8,7 @@ import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
 import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import { undo } from '@atlaskit/editor-prosemirror/history';
 import { TextSelection } from '@atlaskit/editor-prosemirror/state';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -24,8 +25,8 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import { pluginKey as tablePluginKey } from '../../../plugins/table/pm-plugins/plugin-key';
+import tablePlugin from '../../../plugins/table-plugin';
 
 const createDoc = (attrs: any) =>
   doc(

package/src/__tests__/unit/sort-column.ts

@@ -9,6 +9,7 @@ import { guidelinePlugin } from '@atlaskit/editor-plugin-guideline';
 import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
 import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import type { PluginKey } from '@atlaskit/editor-prosemirror/state';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -25,13 +26,13 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../plugins/table-plugin';
 import { sortByColumn } from '../../plugins/table/commands';
 import { pluginKey as tablePluginKey } from '../../plugins/table/pm-plugins/plugin-key';
 import type {
   PermittedLayoutsDescriptor,
   TablePluginState,
 } from '../../plugins/table/types';
+import tablePlugin from '../../plugins/table-plugin';
 
 const TABLE_LOCAL_ID = 'test-table-local-id';
 

package/src/__tests__/unit/toolbar.ts

@@ -12,6 +12,7 @@ import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
 import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import { Rect } from '@atlaskit/editor-tables/table-map';
 import { splitCell } from '@atlaskit/editor-tables/utils';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {

package/src/__tests__/unit/transforms/delete-columns.ts

@@ -10,6 +10,7 @@ import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import type { PluginKey } from '@atlaskit/editor-prosemirror/state';
 import type { Rect } from '@atlaskit/editor-tables/table-map';
 import { getSelectionRect } from '@atlaskit/editor-tables/utils';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -28,10 +29,10 @@ import {
 } from '@atlaskit/editor-test-helpers/doc-builder';
 import { ffTest } from '@atlassian/feature-flags-test-utils';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import { pluginKey } from '../../../plugins/table/pm-plugins/plugin-key';
 import { deleteColumns } from '../../../plugins/table/transforms';
 import type { TablePluginState } from '../../../plugins/table/types';
+import tablePlugin from '../../../plugins/table-plugin';
 
 const colsToRect = (cols: Array<number>, noOfRows: number): Rect => ({
   left: Math.min(...cols),

package/src/__tests__/unit/transforms/delete-rows.ts

@@ -10,6 +10,7 @@ import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import type { PluginKey } from '@atlaskit/editor-prosemirror/state';
 import type { Rect } from '@atlaskit/editor-tables/table-map';
 import { getSelectionRect } from '@atlaskit/editor-tables/utils';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -28,10 +29,10 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import { pluginKey } from '../../../plugins/table/pm-plugins/plugin-key';
 import { deleteRows } from '../../../plugins/table/transforms';
 import type { TablePluginState } from '../../../plugins/table/types';
+import tablePlugin from '../../../plugins/table-plugin';
 
 const rowsToRect = (rows: Array<number>, noOfColumns: number): Rect => ({
   left: 0,

package/src/__tests__/unit/transforms/merging.ts

@@ -8,6 +8,7 @@ import { guidelinePlugin } from '@atlaskit/editor-plugin-guideline';
 import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
 import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import type { PluginKey } from '@atlaskit/editor-prosemirror/state';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -24,10 +25,10 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import { pluginKey } from '../../../plugins/table/pm-plugins/plugin-key';
 import { mergeCells } from '../../../plugins/table/transforms';
 import type { TablePluginState } from '../../../plugins/table/types';
+import tablePlugin from '../../../plugins/table-plugin';
 
 const TABLE_LOCAL_ID = 'test-table-local-id';
 

package/src/__tests__/unit/ui/ContextualMenu.tsx

@@ -11,6 +11,7 @@ import { guidelinePlugin } from '@atlaskit/editor-plugin-guideline';
 import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
 import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import type { EditorView } from '@atlaskit/editor-prosemirror/view';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -28,8 +29,8 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import { ContextualMenu } from '../../../plugins/table/ui/FloatingContextualMenu/ContextualMenu';
+import tablePlugin from '../../../plugins/table-plugin';
 
 describe('ContextualMenu', () => {
   const getEditorContainerWidth = () => ({ width: 500 });

package/src/__tests__/unit/ui/CornerControls.tsx

@@ -12,6 +12,7 @@ import { guidelinePlugin } from '@atlaskit/editor-plugin-guideline';
 import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
 import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import { isTableSelected } from '@atlaskit/editor-tables/utils';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -26,10 +27,10 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import { getPluginState } from '../../../plugins/table/pm-plugins/plugin-factory';
 import { pluginKey } from '../../../plugins/table/pm-plugins/plugin-key';
 import { CornerControls } from '../../../plugins/table/ui/TableFloatingControls/CornerControls';
+import tablePlugin from '../../../plugins/table-plugin';
 
 describe('CornerControls', () => {
   const createEditor = createProsemirrorEditorFactory();

package/src/__tests__/unit/ui/FloatingContextualButton.tsx

@@ -15,6 +15,7 @@ import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import type { ContentNodeWithPos } from '@atlaskit/editor-prosemirror/utils';
 import * as prosemirrorUtils from '@atlaskit/editor-prosemirror/utils';
 import type { EditorView } from '@atlaskit/editor-prosemirror/view';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -31,9 +32,9 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import type { Props as FloatingContextualButtonProps } from '../../../plugins/table/ui/FloatingContextualButton';
 import FloatingContextualButton from '../../../plugins/table/ui/FloatingContextualButton';
+import tablePlugin from '../../../plugins/table-plugin';
 
 jest.mock('@atlaskit/editor-prosemirror/utils', () => {
   // Unblock prosemirror bump:

package/src/__tests__/unit/ui/FloatingContextualMenu.tsx

@@ -11,6 +11,7 @@ import { guidelinePlugin } from '@atlaskit/editor-plugin-guideline';
 import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
 import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import type { EditorView } from '@atlaskit/editor-prosemirror/view';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -28,9 +29,9 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import { getPluginState } from '../../../plugins/table/pm-plugins/plugin-factory';
 import FloatingContextualMenu from '../../../plugins/table/ui/FloatingContextualMenu';
+import tablePlugin from '../../../plugins/table-plugin';
 
 describe('FloatingContextualMenu', () => {
   const createEditor = createProsemirrorEditorFactory();

package/src/__tests__/unit/ui/FloatingDeleteButton.tsx

@@ -20,13 +20,13 @@ import {
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import { selectColumns, selectRows } from '@atlaskit/editor-test-helpers/table';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import { pluginKey } from '../../../plugins/table/pm-plugins/plugin-key';
 import type { TablePluginState } from '../../../plugins/table/types';
 import { TableCssClassName } from '../../../plugins/table/types';
 import type { Props as FloatingDeleteButtonProps } from '../../../plugins/table/ui/FloatingDeleteButton';
 import FloatingDeleteButton from '../../../plugins/table/ui/FloatingDeleteButton';
 import * as tableColumnControlsUtils from '../../../plugins/table/utils/column-controls';
+import tablePlugin from '../../../plugins/table-plugin';
 
 describe('Floating Delete Button', () => {
   const createEditor = createEditorFactory<TablePluginState>();

package/src/__tests__/unit/ui/FloatingDragMenu.tsx

@@ -0,0 +1,195 @@
+import React from 'react';
+
+import { render, screen } from '@testing-library/react';
+
+import { ReactEditorViewContext } from '@atlaskit/editor-common/ui-react';
+// eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
+import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
+import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
+import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
+import { guidelinePlugin } from '@atlaskit/editor-plugin-guideline';
+import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
+import { widthPlugin } from '@atlaskit/editor-plugin-width';
+import * as prosemirrorUtils from '@atlaskit/editor-prosemirror/utils';
+import type { ContentNodeWithPos } from '@atlaskit/editor-prosemirror/utils';
+import type { EditorView } from '@atlaskit/editor-prosemirror/view';
+// eslint-disable-next-line import/no-extraneous-dependencies
+import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
+// eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
+import {
+  createProsemirrorEditorFactory,
+  Preset,
+} from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
+// eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
+import {
+  doc,
+  p,
+  table,
+  tdCursor,
+  tdEmpty,
+  th,
+  tr,
+} from '@atlaskit/editor-test-helpers/doc-builder';
+import { ffTest } from '@atlassian/feature-flags-test-utils';
+
+import FloatingDragMenu from '../../../plugins/table/ui/FloatingDragMenu';
+import tablePlugin from '../../../plugins/table-plugin';
+
+describe('FloatingDragMenu', () => {
+  const createEditor = createProsemirrorEditorFactory();
+  let editorView: EditorView;
+  let tableNode: ContentNodeWithPos | undefined;
+  beforeEach(() => {
+    ({ editorView } = createEditor({
+      doc: doc(table()(tr(th()(p('')), th()(p(''))), tr(tdCursor, tdEmpty))),
+      preset: new Preset<LightEditorPlugin>()
+        .add([featureFlagsPlugin, {}])
+        .add([analyticsPlugin, {}])
+        .add(contentInsertionPlugin)
+        .add(widthPlugin)
+        .add(guidelinePlugin)
+        .add(selectionPlugin)
+        .add(tablePlugin),
+    }));
+
+    tableNode = prosemirrorUtils.findParentNodeOfTypeClosestToPos(
+      editorView.state.selection.$from,
+      editorView.state.schema.nodes.table,
+    );
+  });
+  describe('should group the menu items correctly', () => {
+    ffTest(
+      'platform.editor.menu.group-items',
+      () => {
+        // Run test case when FF platform.editor..menu.group-items is true
+        const tableRef = editorView.dom.querySelector('table');
+        const editorRef = {
+          current: document.createElement('div'),
+        };
+        render(
+          <ReactEditorViewContext.Provider
+            value={{
+              editorRef,
+            }}
+          >
+            <FloatingDragMenu
+              editorView={editorView}
+              isOpen
+              getEditorContainerWidth={jest.fn()}
+              tableRef={tableRef as HTMLTableElement}
+              tableNode={tableNode?.node}
+              direction="row"
+              targetCellPosition={1}
+            />
+          </ReactEditorViewContext.Provider>,
+        );
+        expect(screen.getAllByRole('menu')).toHaveLength(1);
+        expect(screen.getAllByRole('group')).toHaveLength(2);
+      },
+      () => {
+        // Run test case when FF platform.editor..menu.group-items is false
+        const tableRef = editorView.dom.querySelector('table');
+        const editorRef = {
+          current: document.createElement('div'),
+        };
+        render(
+          <ReactEditorViewContext.Provider
+            value={{
+              editorRef,
+            }}
+          >
+            <FloatingDragMenu
+              editorView={editorView}
+              isOpen
+              getEditorContainerWidth={jest.fn()}
+              tableRef={tableRef as HTMLTableElement}
+              tableNode={tableNode?.node}
+              direction="row"
+              targetCellPosition={1}
+            />
+          </ReactEditorViewContext.Provider>,
+        );
+        expect(screen.getAllByRole('menu')).toHaveLength(2);
+        expect(screen.queryByRole('group')).toBeNull();
+      },
+    );
+  });
+  describe('menu items', () => {
+    it('should render the right menu items for row', () => {
+      // Run test case when FF platform.editor..menu.group-items is true
+      const tableRef = editorView.dom.querySelector('table');
+      const editorRef = {
+        current: document.createElement('div'),
+      };
+      render(
+        <ReactEditorViewContext.Provider
+          value={{
+            editorRef,
+          }}
+        >
+          <FloatingDragMenu
+            editorView={editorView}
+            isOpen
+            getEditorContainerWidth={jest.fn()}
+            tableRef={tableRef as HTMLTableElement}
+            tableNode={tableNode?.node}
+            direction="row"
+            targetCellPosition={1}
+          />
+        </ReactEditorViewContext.Provider>,
+      );
+
+      const items = screen.getAllByRole('menuitem');
+      const menuitemTextArray = items.map((item) => item.textContent);
+      expect(menuitemTextArray).toMatchInlineSnapshot(`
+          Array [
+            "Add row aboveCtrl+Alt+↑",
+            "Add row belowCtrl+Alt+↓",
+            "Clear cells⌫",
+            "Delete row",
+            "Move row up",
+            "Move row down",
+          ]
+        `);
+    });
+    it('should render the right menu items for column', () => {
+      const tableRef = editorView.dom.querySelector('table');
+      const editorRef = {
+        current: document.createElement('div'),
+      };
+      render(
+        <ReactEditorViewContext.Provider
+          value={{
+            editorRef,
+          }}
+        >
+          <FloatingDragMenu
+            editorView={editorView}
+            isOpen
+            getEditorContainerWidth={jest.fn()}
+            tableRef={tableRef as HTMLTableElement}
+            tableNode={tableNode?.node}
+            direction="column"
+            targetCellPosition={1}
+          />
+        </ReactEditorViewContext.Provider>,
+      );
+
+      const items = screen.getAllByRole('menuitem');
+      const menuitemTextArray = items.map((item) => item.textContent);
+      expect(menuitemTextArray).toMatchInlineSnapshot(`
+          Array [
+            "Add column leftCtrl+Alt+←",
+            "Add column rightCtrl+Alt+→",
+            "Distribute columns",
+            "Clear cells⌫",
+            "Delete column",
+            "Move column left",
+            "Move column right",
+            "Sort increasing",
+            "Sort decreasing",
+          ]
+        `);
+    });
+  });
+});

package/src/__tests__/unit/ui/FloatingInsertButton.tsx

@@ -32,13 +32,13 @@ import {
   selectRows,
 } from '@atlaskit/editor-test-helpers/table';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import { setEditorFocus, setTableRef } from '../../../plugins/table/commands';
 import { getPluginState } from '../../../plugins/table/pm-plugins/plugin-factory';
 import { pluginKey } from '../../../plugins/table/pm-plugins/plugin-key';
 import type { TablePluginState } from '../../../plugins/table/types';
 import type { Props as FloatingInsertButtonProps } from '../../../plugins/table/ui/FloatingInsertButton';
 import { FloatingInsertButton } from '../../../plugins/table/ui/FloatingInsertButton';
+import tablePlugin from '../../../plugins/table-plugin';
 
 jest.mock('@atlaskit/editor-prosemirror/utils', () => {
   // Unblock prosemirror bump:

package/src/__tests__/unit/ui/NumberColumn.tsx

@@ -11,6 +11,7 @@ import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
 import { guidelinePlugin } from '@atlaskit/editor-plugin-guideline';
 import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
 import { widthPlugin } from '@atlaskit/editor-plugin-width';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -26,10 +27,10 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import { pluginKey } from '../../../plugins/table/pm-plugins/plugin-key';
 import { TableCssClassName as ClassName } from '../../../plugins/table/types';
 import NumberColumn from '../../../plugins/table/ui/TableFloatingControls/NumberColumn';
+import tablePlugin from '../../../plugins/table-plugin';
 
 describe('NumberColumn', () => {
   const createEditor = createProsemirrorEditorFactory();

package/src/__tests__/unit/ui/RowControls.tsx

@@ -13,6 +13,7 @@ import { guidelinePlugin } from '@atlaskit/editor-plugin-guideline';
 import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
 import { widthPlugin } from '@atlaskit/editor-plugin-width';
 import { getSelectionRect, selectRow } from '@atlaskit/editor-tables/utils';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -33,11 +34,11 @@ import {
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import { selectRows } from '@atlaskit/editor-test-helpers/table';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import { hoverRows } from '../../../plugins/table/commands';
 import { pluginKey } from '../../../plugins/table/pm-plugins/plugin-key';
 import TableFloatingControls from '../../../plugins/table/ui/TableFloatingControls';
 import { RowControls } from '../../../plugins/table/ui/TableFloatingControls/RowControls';
+import tablePlugin from '../../../plugins/table-plugin';
 
 describe('RowControls', () => {
   const createEditor = createProsemirrorEditorFactory();

package/src/__tests__/unit/ui/RowDragControls.tsx

@@ -11,6 +11,7 @@ import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
 import { guidelinePlugin } from '@atlaskit/editor-plugin-guideline';
 import { selectionPlugin } from '@atlaskit/editor-plugin-selection';
 import { widthPlugin } from '@atlaskit/editor-plugin-width';
+// eslint-disable-next-line import/no-extraneous-dependencies
 import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 // eslint-disable-next-line import/no-extraneous-dependencies -- Removed import for fixing circular dependencies
 import {
@@ -27,10 +28,10 @@ import {
   tr,
 } from '@atlaskit/editor-test-helpers/doc-builder';
 
-import tablePlugin from '../../../plugins/table-plugin';
 import { pluginKey } from '../../../plugins/table/pm-plugins/plugin-key';
 // import { TableCssClassName as ClassName } from '../../../plugins/table/types';
 import { DragControls } from '../../../plugins/table/ui/TableFloatingControls/RowControls';
+import tablePlugin from '../../../plugins/table-plugin';
 
 describe('NumberColumn', () => {
   const createEditor = createProsemirrorEditorFactory();