@atlaskit/editor-plugin-table 2.12.6 → 2.13.1

package/src/__tests__/unit/commands.ts

@@ -1,6 +1,7 @@
 import { uuid } from '@atlaskit/adf-schema';
 import { defaultSchema } from '@atlaskit/adf-schema/schema-default';
 import type { EditorAnalyticsAPI } from '@atlaskit/editor-common/analytics';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { decorationsPlugin } from '@atlaskit/editor-plugin-decorations';
@@ -75,6 +76,7 @@ describe('table plugin: actions', () => {
         .add(widthPlugin)
         .add(guidelinePlugin)
         .add(gridPlugin)
+        .add(selectionPlugin)
         .add(tablePlugin),
       pluginKey,
     });

package/src/__tests__/unit/copy-paste.ts

@@ -2,6 +2,7 @@
 
 import { uuid } from '@atlaskit/adf-schema';
 import { transformSliceToRemoveOpenExpand } from '@atlaskit/editor-common/transforms';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -119,6 +120,7 @@ describe('table plugin', () => {
     .add(contentInsertionPlugin)
     .add(widthPlugin)
     .add(guidelinePlugin)
+    .add(selectionPlugin)
     .add([tablePlugin, { tableOptions }]);
 
   const editor = (doc: DocBuilder) => {

package/src/__tests__/unit/event-handlers/index.ts

@@ -1,3 +1,4 @@
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -37,6 +38,7 @@ describe('event-handlers', () => {
           .add(contentInsertionPlugin)
           .add(widthPlugin)
           .add(guidelinePlugin)
+          .add(selectionPlugin)
           .add([tablePlugin, { tableOptions: { allowColumnResizing: false } }]);
         editor = (doc: DocBuilder) =>
           createEditor<TablePluginState, PluginKey, typeof preset>({
@@ -70,6 +72,7 @@ describe('event-handlers', () => {
           .add(contentInsertionPlugin)
           .add(widthPlugin)
           .add(guidelinePlugin)
+          .add(selectionPlugin)
           .add([tablePlugin, { tableOptions: { allowColumnResizing: false } }]);
         editor = (doc: DocBuilder) =>
           createEditor<TablePluginState, PluginKey, typeof preset>({

package/src/__tests__/unit/event-handlers.ts

@@ -1,3 +1,4 @@
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { decorationsPlugin } from '@atlaskit/editor-plugin-decorations';
@@ -47,6 +48,7 @@ describe('table plugin: decorations', () => {
         .add(contentInsertionPlugin)
         .add(widthPlugin)
         .add(guidelinePlugin)
+        .add(selectionPlugin)
         .add(tablePlugin),
       pluginKey,
     });
@@ -166,6 +168,7 @@ describe('table event handlers', () => {
         .add(widthPlugin)
         .add(guidelinePlugin)
         .add(gridPlugin)
+        .add(selectionPlugin)
         .add(tablePlugin),
       pluginKey,
     });

package/src/__tests__/unit/fix-tables.ts

@@ -1,3 +1,4 @@
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -47,6 +48,7 @@ describe('fix tables', () => {
         .add(contentInsertionPlugin)
         .add(widthPlugin)
         .add(guidelinePlugin)
+        .add(selectionPlugin)
         .add([tablePlugin, { tableOptions }]),
       pluginKey: tablePluginKey,
     });

package/src/__tests__/unit/get-toolbar-config.ts

@@ -4,6 +4,7 @@ import type {
   FloatingToolbarItem,
   GetEditorFeatureFlags,
 } from '@atlaskit/editor-common/types';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { decorationsPlugin } from '@atlaskit/editor-plugin-decorations';
@@ -80,6 +81,7 @@ describe('getToolbarConfig', () => {
         .add(decorationsPlugin)
         .add(widthPlugin)
         .add(guidelinePlugin)
+        .add(selectionPlugin)
         .add(tablePlugin),
     });
 

package/src/__tests__/unit/handlers.ts

@@ -1,3 +1,4 @@
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -37,6 +38,7 @@ describe('table action handlers', () => {
       .add(contentInsertionPlugin)
       .add(widthPlugin)
       .add(guidelinePlugin)
+      .add(selectionPlugin)
       .add(tablePlugin);
 
     editor = (doc: DocBuilder) =>

package/src/__tests__/unit/hover-selection.ts

@@ -1,3 +1,4 @@
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -46,6 +47,7 @@ describe('table hover selection plugin', () => {
     .add(contentInsertionPlugin)
     .add(widthPlugin)
     .add(guidelinePlugin)
+    .add(selectionPlugin)
     .add(tablePlugin);
 
   const editor = (doc: DocBuilder) =>

package/src/__tests__/unit/index.ts

@@ -1,6 +1,7 @@
 import { uuid } from '@atlaskit/adf-schema';
 import type { GetEditorContainerWidth } from '@atlaskit/editor-common/types';
 import { setNodeSelection } from '@atlaskit/editor-common/utils';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -90,6 +91,7 @@ describe('table plugin', () => {
       .add(guidelinePlugin)
       .add(gridPlugin)
       .add(contentInsertionPlugin)
+      .add(selectionPlugin)
       .add([tablePlugin, { tableOptions }]);
 
     return createEditor({

package/src/__tests__/unit/layout.ts

@@ -1,4 +1,5 @@
 import type { TableLayout } from '@atlaskit/adf-schema';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { decorationsPlugin } from '@atlaskit/editor-plugin-decorations';
@@ -46,6 +47,7 @@ describe('table toolbar', () => {
     .add(decorationsPlugin)
     .add(widthPlugin)
     .add(guidelinePlugin)
+    .add(selectionPlugin)
     .add([tablePlugin, { tableOptions }]);
 
   const editor = (doc: DocBuilder) => {

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

@@ -280,4 +280,62 @@ describe('table -> nodeviews -> TableContainer.tsx', () => {
       expect(mockEndMeasure).toHaveBeenCalled();
     });
   });
+
+  describe('selection', () => {
+    const selectionActionMock = jest.fn().mockReturnValue(() => {});
+
+    const buildContainer = (attrs: TableAttributes) => {
+      const { table, editorView } = createNode(attrs);
+
+      const { container } = render(
+        <ResizableTableContainer
+          containerWidth={1800}
+          lineLength={720}
+          node={table}
+          className={''}
+          editorView={editorView}
+          getPos={() => 0}
+          tableRef={
+            {
+              querySelector: () => null,
+              insertBefore: () => {},
+              style: {},
+            } as any
+          }
+          pluginInjectionApi={
+            {
+              selection: {
+                commands: { displayGapCursor: selectionActionMock },
+              },
+              // mock core so the selection command will execute
+              core: { actions: { execute: jest.fn() } },
+            } as any
+          }
+        />,
+      );
+
+      return { container, selectionActionMock };
+    };
+
+    afterEach(() => {
+      selectionActionMock.mockRestore();
+    });
+
+    it('fires selection action to hide gap cursor when present', () => {
+      const { container, selectionActionMock } = buildContainer({
+        layout: 'wide',
+      });
+
+      fireEvent.mouseDown(container.querySelector('.resizer-handle.right')!);
+      fireEvent.mouseMove(container.querySelector('.resizer-handle.right')!);
+      fireEvent.mouseMove(container.querySelector('.resizer-handle.right')!);
+      fireEvent.mouseMove(container.querySelector('.resizer-handle.right')!);
+      fireEvent.mouseUp(container.querySelector('.resizer-handle.right')!);
+
+      // when resize starts - call to hide gap cursor
+      expect(selectionActionMock).toHaveBeenNthCalledWith(1, false);
+      // when resize finishes - call to display gap cursor
+      expect(selectionActionMock).toHaveBeenNthCalledWith(2, true);
+    });
+  });
 });

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

@@ -3,6 +3,7 @@ import {
   tableBackgroundColorNames,
   uuid,
 } from '@atlaskit/adf-schema';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -53,6 +54,7 @@ describe('table -> nodeviews -> tableCell.tsx', () => {
         .add(contentInsertionPlugin)
         .add(widthPlugin)
         .add(guidelinePlugin)
+        .add(selectionPlugin)
         .add([tablePlugin, { tableOptions: { advanced: true, ...props } }]),
       pluginKey,
     });

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

@@ -1,4 +1,5 @@
 import type { TableAttributes } from '@atlaskit/adf-schema';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -45,6 +46,7 @@ describe('table -> nodeviews -> table.tsx', () => {
           .add(contentInsertionPlugin)
           .add(widthPlugin)
           .add(guidelinePlugin)
+          .add(selectionPlugin)
           .add([
             tablePlugin,
             {

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

@@ -1,3 +1,4 @@
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -44,6 +45,7 @@ describe('tables: column resizing decorations', () => {
     .add(contentInsertionPlugin)
     .add(widthPlugin)
     .add(guidelinePlugin)
+    .add(selectionPlugin)
     .add([
       tablePlugin,
       {

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

@@ -1,3 +1,4 @@
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -42,6 +43,7 @@ describe('decorations plugin', () => {
         .add(contentInsertionPlugin)
         .add(widthPlugin)
         .add(guidelinePlugin)
+        .add(selectionPlugin)
         .add(tablePlugin),
       pluginKey,
     });
@@ -166,6 +168,7 @@ describe('decorations plugin', () => {
           .add(contentInsertionPlugin)
           .add(widthPlugin)
           .add(guidelinePlugin)
+          .add(selectionPlugin)
           .add([
             tablePlugin,
             {

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

@@ -1,3 +1,4 @@
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -57,6 +58,7 @@ describe('tables: main plugin', () => {
         .add(contentInsertionPlugin)
         .add(widthPlugin)
         .add(guidelinePlugin)
+        .add(selectionPlugin)
         .add(tablePlugin),
       pluginKey,
     });

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

@@ -1,3 +1,4 @@
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -33,6 +34,7 @@ describe('table/safari-delete-composition-text-issue-workaround', () => {
           .add(contentInsertionPlugin)
           .add(widthPlugin)
           .add(guidelinePlugin)
+          .add(selectionPlugin)
           .add([tablePlugin, { tableOptions: { allowColumnResizing: true } }]),
         pluginKey,
       });

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

@@ -6,6 +6,7 @@ import createStub from 'raf-stub';
 
 import type { EventDispatcher } from '@atlaskit/editor-common/event-dispatcher';
 import { findOverflowScrollParent } from '@atlaskit/editor-common/ui';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -67,6 +68,7 @@ describe('TableRowNodeView', () => {
         .add(contentInsertionPlugin)
         .add(widthPlugin)
         .add(guidelinePlugin)
+        .add(selectionPlugin)
         .add(tablePlugin),
       pluginKey,
       attachTo: document.body,
@@ -100,6 +102,7 @@ describe('TableRowNodeView', () => {
             .add(contentInsertionPlugin)
             .add(widthPlugin)
             .add(guidelinePlugin)
+            .add(selectionPlugin)
             .add(tablePlugin),
           pluginKey,
         });

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

@@ -3,6 +3,7 @@ import {
   EVENT_TYPE,
   TABLE_ACTION,
 } from '@atlaskit/editor-common/analytics';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import type { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { decorationsPlugin } from '@atlaskit/editor-plugin-decorations';
@@ -59,6 +60,7 @@ describe('table-resizing/event-handlers', () => {
           .add(decorationsPlugin)
           .add(widthPlugin)
           .add(guidelinePlugin)
+          .add(selectionPlugin)
           .add([
             tablePlugin,
             {

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

@@ -0,0 +1,99 @@
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
+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 { widthPlugin } from '@atlaskit/editor-plugin-width';
+import { TextSelection } from '@atlaskit/editor-prosemirror/state';
+import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
+import {
+  createProsemirrorEditorFactory,
+  Preset,
+} from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
+import type { DocBuilder } from '@atlaskit/editor-test-helpers/doc-builder';
+import {
+  doc,
+  p,
+  table,
+  td,
+  tdEmpty,
+  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';
+
+describe('table width', () => {
+  const createEditor = createProsemirrorEditorFactory();
+
+  beforeAll(() => {
+    // @ts-ignore
+    global['fetch'] = jest.fn();
+  });
+
+  const editor = (doc: DocBuilder) => {
+    return createEditor({
+      doc,
+      attachTo: document.body,
+      preset: new Preset<LightEditorPlugin>()
+        .add([featureFlagsPlugin, {}])
+        .add([analyticsPlugin, {}])
+        .add(contentInsertionPlugin)
+        .add(widthPlugin)
+        .add(guidelinePlugin)
+        .add(selectionPlugin)
+        .add([
+          tablePlugin,
+          {
+            tableResizingEnabled: true, // so table-width plugin is enabled
+            tableOptions: {
+              advanced: true,
+            },
+          },
+        ]),
+      pluginKey: tablePluginKey,
+    });
+  };
+
+  describe('move selection from outside table to inside  table should fire analytics', () => {
+    const TABLE_LOCAL_ID = 'test-table-1';
+    it('should fire v3 analytics', () => {
+      // create editor with a table and a paragraph below
+      // mouse cursor is at the end of the parapgraph
+      const {
+        dispatchAnalyticsEvent,
+        editorView,
+        refs: { inTablePos },
+      } = editor(
+        doc(
+          table({ localId: TABLE_LOCAL_ID })(
+            tr(td({})(p('{inTablePos}in cell')), tdEmpty, tdEmpty),
+            tr(tdEmpty, tdEmpty, tdEmpty),
+          ),
+          p('hello cursor is here initially{<>}'),
+        ),
+      );
+
+      // move cursor to where ths inTablePos is
+      const $inTablePos = editorView.state.doc.resolve(inTablePos);
+      editorView.dispatch(
+        editorView.state.tr.setSelection(
+          new TextSelection($inTablePos, $inTablePos),
+        ),
+      );
+
+      // analytics event should be fired
+      expect(dispatchAnalyticsEvent).toHaveBeenCalledWith(
+        expect.objectContaining({
+          action: 'selected',
+          actionSubject: 'document',
+          actionSubjectId: 'table',
+          attributes: expect.objectContaining({
+            localId: TABLE_LOCAL_ID,
+          }),
+          eventType: 'track',
+        }),
+      );
+    });
+  });
+});

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

@@ -1,5 +1,6 @@
 import { uuid } from '@atlaskit/adf-schema';
 import { TableSortOrder as SortOrder } from '@atlaskit/adf-schema/steps';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -55,6 +56,7 @@ describe('table plugin', () => {
     .add(contentInsertionPlugin)
     .add(widthPlugin)
     .add(guidelinePlugin)
+    .add(selectionPlugin)
     .add([tablePlugin, { tableOptions }]);
 
   const editor = (doc: DocBuilder) => {

package/src/__tests__/unit/toolbar.ts

@@ -3,6 +3,7 @@ import type {
   DropdownOptionT,
   FloatingToolbarDropdown,
 } from '@atlaskit/editor-common/types';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -166,6 +167,7 @@ describe('getToolbarCellOptionsConfig', () => {
       .add(contentInsertionPlugin)
       .add(widthPlugin)
       .add(guidelinePlugin)
+      .add(selectionPlugin)
       .add([
         tablePlugin,
         {

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

@@ -1,4 +1,5 @@
 import { uuid } from '@atlaskit/adf-schema';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -53,6 +54,7 @@ describe('table plugin -> transforms -> delete columns', () => {
     .add(contentInsertionPlugin)
     .add(widthPlugin)
     .add(guidelinePlugin)
+    .add(selectionPlugin)
     .add(tablePlugin);
 
   const editor = (doc: DocBuilder) =>

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

@@ -1,4 +1,5 @@
 import { uuid } from '@atlaskit/adf-schema';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -53,6 +54,7 @@ describe('table plugin -> transforms -> delete rows', () => {
     .add(contentInsertionPlugin)
     .add(widthPlugin)
     .add(guidelinePlugin)
+    .add(selectionPlugin)
     .add(tablePlugin);
 
   const editor = (doc: DocBuilder) =>

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

@@ -1,4 +1,5 @@
 import { uuid } from '@atlaskit/adf-schema';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -43,6 +44,7 @@ describe('table plugin -> transforms -> merge cells', () => {
     .add(contentInsertionPlugin)
     .add(widthPlugin)
     .add(guidelinePlugin)
+    .add(selectionPlugin)
     .add(tablePlugin);
 
   const editor = (doc: DocBuilder) =>

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

@@ -3,6 +3,7 @@ import React from 'react';
 import { render, screen } from '@testing-library/react';
 import { createIntl } from 'react-intl-next';
 
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -42,6 +43,7 @@ describe('ContextualMenu', () => {
           .add(contentInsertionPlugin)
           .add(widthPlugin)
           .add(guidelinePlugin)
+          .add(selectionPlugin)
           .add([tablePlugin, { tableOptions: { advanced: true } }]),
       }));
     });

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

@@ -3,6 +3,7 @@ import React from 'react';
 import { fireEvent, render, screen } from '@testing-library/react';
 import { IntlProvider } from 'react-intl-next';
 
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -39,6 +40,7 @@ describe('CornerControls', () => {
         .add(contentInsertionPlugin)
         .add(widthPlugin)
         .add(guidelinePlugin)
+        .add(selectionPlugin)
         .add(tablePlugin),
       pluginKey,
     });

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

@@ -4,6 +4,7 @@ import { render, screen } from '@testing-library/react';
 import { IntlProvider } from 'react-intl-next';
 
 import type { UIAnalyticsEvent } from '@atlaskit/analytics-next';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -55,6 +56,7 @@ const editor = (doc: DocBuilder) =>
       .add(contentInsertionPlugin)
       .add(widthPlugin)
       .add(guidelinePlugin)
+      .add(selectionPlugin)
       .add(tablePlugin),
   });
 

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

@@ -3,15 +3,16 @@ import React from 'react';
 import { render, screen } from '@testing-library/react';
 import { IntlProvider } from 'react-intl-next';
 
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 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 { widthPlugin } from '@atlaskit/editor-plugin-width';
-import { EditorView } from '@atlaskit/editor-prosemirror/view';
+import type { EditorView } from '@atlaskit/editor-prosemirror/view';
+import type { LightEditorPlugin } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 import {
   createProsemirrorEditorFactory,
-  LightEditorPlugin,
   Preset,
 } from '@atlaskit/editor-test-helpers/create-prosemirror-editor';
 import {
@@ -41,6 +42,7 @@ describe('FloatingContextualMenu', () => {
           .add(contentInsertionPlugin)
           .add(widthPlugin)
           .add(guidelinePlugin)
+          .add(selectionPlugin)
           .add(tablePlugin),
       }));
     });

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

@@ -4,6 +4,7 @@ import { fireEvent, render, screen } from '@testing-library/react';
 import { IntlProvider } from 'react-intl-next';
 
 import { setTextSelection } from '@atlaskit/editor-common/utils';
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -67,6 +68,7 @@ describe('RowControls', () => {
         .add(contentInsertionPlugin)
         .add(widthPlugin)
         .add(guidelinePlugin)
+        .add(selectionPlugin)
         .add(tablePlugin),
       pluginKey,
     });

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

@@ -3,6 +3,7 @@ import React from 'react';
 import { render, screen } from '@testing-library/react';
 import { IntlProvider } from 'react-intl-next';
 
+import selectionPlugin from '@atlaskit/editor-core/src/plugins/selection';
 import { analyticsPlugin } from '@atlaskit/editor-plugin-analytics';
 import { contentInsertionPlugin } from '@atlaskit/editor-plugin-content-insertion';
 import { featureFlagsPlugin } from '@atlaskit/editor-plugin-feature-flags';
@@ -40,6 +41,7 @@ describe('TableFloatingControls', () => {
     .add(contentInsertionPlugin)
     .add(widthPlugin)
     .add(guidelinePlugin)
+    .add(selectionPlugin)
     .add(tablePlugin);
 
   const editor = (doc: DocBuilder) =>