@atlaskit/adf-schema 28.1.7 → 28.1.9

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@atlaskit/adf-schema",
-  "version": "28.1.7",
+  "version": "28.1.9",
   "description": "Shared package that contains the ADF-schema (json) and ProseMirror node/mark specs",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/"
@@ -27,7 +27,7 @@
     "build:types": "NODE_ENV=production tsc --project ./build/tsconfig.json --emitDeclarationOnly --outDir ./dist/types --rootDir ./src && yarn build:jsontypes",
     "build:jsontypes": "mkdir -p dist/types/json-schema && cp src/json-schema/index.ts dist/types/json-schema/index.d.ts",
     "build:all": "yarn build:cjs && yarn build:esm && yarn build:es2019 && yarn build:types",
-    "build": "yarn run build:all && yarn run generate:json-schema && yarn run build:json-schema",
+    "build": "yarn workspace @atlaskit/editor-prosemirror build && yarn run build:all && yarn run generate:json-schema && yarn run build:json-schema",
     "clean": "../../scripts/build-clean.sh"
   },
   "dependencies": {

package/src/__tests__/unit/schema/nodes/confluence-unsupported-block.ts

@@ -1,8 +1,4 @@
-import {
-  schema,
-  toDOM,
-  fromHTML,
-} from 'adf-test-helpers/src/adf-schema';
+import { schema, toDOM, fromHTML } from 'adf-test-helpers/src/adf-schema';
 
 const packageName = process.env._PACKAGE_NAME_ as string;
 

package/src/__tests__/unit/schema/nodes/confluence-unsupported-inline.ts

@@ -1,6 +1,6 @@
 import { toDOM, fromHTML } from 'adf-test-helpers/src/adf-schema/html-helpers';
 
-import { schema } from 'adf-test-helpers/src/adf-schema'
+import { schema } from 'adf-test-helpers/src/adf-schema';
 
 const packageName = process.env._PACKAGE_NAME_ as string;
 

package/src/__tests__/unit/schema/nodes/date.ts

@@ -1,7 +1,7 @@
 import { toDOM, fromHTML } from 'adf-test-helpers/src/adf-schema/html-helpers';
 import { date } from '../../../..';
 
-import { schema } from 'adf-test-helpers/src/adf-schema'
+import { schema } from 'adf-test-helpers/src/adf-schema';
 
 const packageName = process.env._PACKAGE_NAME_ as string;
 

package/src/__tests__/unit/schema/nodes/expand.ts

@@ -1,6 +1,6 @@
 import { toDOM, fromHTML } from 'adf-test-helpers/src/adf-schema/html-helpers';
 import { expand } from '../../../..';
-import { schema } from 'adf-test-helpers/src/adf-schema'
+import { schema } from 'adf-test-helpers/src/adf-schema';
 const packageName = process.env._PACKAGE_NAME_ as string;
 
 describe(`${packageName}/schema expand node`, () => {

package/src/__tests__/unit/schema/nodes/inline-extension.ts

@@ -1,6 +1,6 @@
 import { toDOM, fromHTML } from 'adf-test-helpers/src/adf-schema/html-helpers';
 import { inlineExtension } from '../../../..';
-import { schema } from 'adf-test-helpers/src/adf-schema'
+import { schema } from 'adf-test-helpers/src/adf-schema';
 
 const packageName = process.env._PACKAGE_NAME_ as string;
 

package/src/__tests__/unit/schema/nodes/list-item.ts

@@ -2,7 +2,7 @@ import { toHTML, fromHTML } from 'adf-test-helpers/src/adf-schema/html-helpers';
 
 import { doc, ol, li, p } from 'adf-test-helpers/src/adf-schema/schema-builder';
 
-import { schema } from 'adf-test-helpers/src/adf-schema'
+import { schema } from 'adf-test-helpers/src/adf-schema';
 const packageName = process.env._PACKAGE_NAME_ as string;
 
 describe(`${packageName}/schema listItem node`, () => {

package/src/__tests__/unit/schema/nodes/media-group.ts

@@ -1,6 +1,6 @@
 import { fromHTML, toDOM } from 'adf-test-helpers/src/adf-schema/html-helpers';
 
-import { schema } from 'adf-test-helpers/src/adf-schema'
+import { schema } from 'adf-test-helpers/src/adf-schema';
 
 const packageName = process.env._PACKAGE_NAME_ as string;
 

package/src/__tests__/unit/schema/nodes/media-single.ts

@@ -1,6 +1,6 @@
 import { toDOM, fromHTML } from 'adf-test-helpers/src/adf-schema/html-helpers';
 
-import { schema } from 'adf-test-helpers/src/adf-schema'
+import { schema } from 'adf-test-helpers/src/adf-schema';
 
 const packageName = process.env._PACKAGE_NAME_ as string;
 

package/src/__tests__/unit/schema/nodes/media.ts

@@ -6,11 +6,7 @@ import {
   createMediaSpec,
 } from '../../../../schema/nodes/media';
 import { image as ImageNodeSpec } from '../../../../schema/nodes/image';
-import {
-  fromHTML,
-  toDOM,
-  schema,
-} from 'adf-test-helpers/src/adf-schema';
+import { fromHTML, toDOM, schema } from 'adf-test-helpers/src/adf-schema';
 
 // Note: We can't use dom.dataset in jest until it's upgraded to use latest version
 //       of jsdom. In the meantime we can use this helper-method.

package/src/__tests__/unit/schema/nodes/nestedExpand.ts

@@ -1,7 +1,7 @@
 import { toDOM, fromHTML } from 'adf-test-helpers/src/adf-schema/html-helpers';
 import { nestedExpand } from '../../../..';
 import { Schema, Node } from '@atlaskit/editor-prosemirror/model';
-import { schema } from 'adf-test-helpers/src/adf-schema'
+import { schema } from 'adf-test-helpers/src/adf-schema';
 
 const findNestedExpand = (doc: Node, schema: Schema) => {
   let expand: Node | null = null;

package/src/__tests__/unit/schema/nodes/ordered-list.ts

@@ -1,5 +1,5 @@
 import { toHTML } from 'adf-test-helpers/src/adf-schema/html-helpers';
-import { schema } from 'adf-test-helpers/src/adf-schema'
+import { schema } from 'adf-test-helpers/src/adf-schema';
 const packageName = process.env._PACKAGE_NAME_ as string;
 
 describe(`${packageName}/schema orderedList node`, () => {

package/src/__tests__/unit/schema/nodes/placeholder.ts

@@ -1,6 +1,6 @@
 import { toDOM, fromHTML } from 'adf-test-helpers/src/adf-schema/html-helpers';
 import { placeholder } from '../../../..';
-import { schema } from 'adf-test-helpers/src/adf-schema'
+import { schema } from 'adf-test-helpers/src/adf-schema';
 
 const packageName = process.env._PACKAGE_NAME_ as string;
 

package/src/__tests__/unit/schema/nodes/status.ts

@@ -1,7 +1,7 @@
 import { toDOM, fromHTML } from 'adf-test-helpers/src/adf-schema/html-helpers';
 import { status } from '../../../..';
 import { StatusLocalIdRegex } from 'adf-test-helpers/src/constants';
-import { schema } from 'adf-test-helpers/src/adf-schema'
+import { schema } from 'adf-test-helpers/src/adf-schema';
 const packageName = process.env._PACKAGE_NAME_ as string;
 
 describe(`${packageName}/schema status node`, () => {

package/src/__tests__/unit/schema/nodes/unknown-block.ts

@@ -1,7 +1,7 @@
 import { fromHTML, toHTML } from 'adf-test-helpers/src/adf-schema/html-helpers';
 import { inlineNodes } from '../../../../schema/inline-nodes';
 
-import { schema } from 'adf-test-helpers/src/adf-schema'
+import { schema } from 'adf-test-helpers/src/adf-schema';
 
 const packageName = process.env._PACKAGE_NAME_ as string;
 

package/src/__tests__/unit/schema/nodes/unsupported-block.ts

@@ -1,6 +1,6 @@
 import { toDOM, fromHTML } from 'adf-test-helpers/src/adf-schema/html-helpers';
 
-import { schema } from 'adf-test-helpers/src/adf-schema'
+import { schema } from 'adf-test-helpers/src/adf-schema';
 
 const packageName = process.env._PACKAGE_NAME_ as string;
 

package/src/__tests__/unit/schema/nodes/unsupported-inline.ts

@@ -1,6 +1,6 @@
 import { toDOM, fromHTML } from 'adf-test-helpers/src/adf-schema/html-helpers';
 
-import { schema } from 'adf-test-helpers/src/adf-schema'
+import { schema } from 'adf-test-helpers/src/adf-schema';
 
 const packageName = process.env._PACKAGE_NAME_ as string;
 

package/src/schema/nodes/media-single.ts

@@ -57,10 +57,10 @@ export type MediaSingleFullDefinition = MediaSingleBaseDefinition &
 
 export const defaultAttrs: {
   width: {
-      default: null;
+    default: null;
   };
   layout: {
-      default: string;
+    default: string;
   };
 } = {
   width: { default: null }, // null makes small images to have original size by default

package/src/schema/nodes/media.ts

@@ -201,7 +201,11 @@ export const media: NodeSpec = createMediaSpec(defaultAttrs);
 export const camelCaseToKebabCase = (str: string) =>
   str.replace(/([^A-Z]+)([A-Z])/g, (_, x, y) => `${x}-${y.toLowerCase()}`);
 
-export const copyPrivateAttributes: (from: Record<string, any>, to: Record<string, any>, map?: ((str: string) => string) | undefined) => void = (
+export const copyPrivateAttributes: (
+  from: Record<string, any>,
+  to: Record<string, any>,
+  map?: ((str: string) => string) | undefined,
+) => void = (
   from: Record<string, any>,
   to: Record<string, any>,
   map?: (str: string) => string,

package/src/schema/nodes/tableNodes.ts

@@ -65,16 +65,14 @@ export const tableCellContentDomSelector = `${tablePrefixSelector}-cell-nodeview
 const DEFAULT_TABLE_HEADER_CELL_BACKGROUND = N20.toLocaleLowerCase();
 
 export const getCellAttrs: (
-  dom: HTMLElement, defaultValues?: CellAttributes
+  dom: HTMLElement,
+  defaultValues?: CellAttributes,
 ) => {
   colspan: number;
   rowspan: number;
   colwidth: number[] | null;
   background: string | null;
-} = (
-  dom: HTMLElement,
-  defaultValues: CellAttributes = {},
-) => {
+} = (dom: HTMLElement, defaultValues: CellAttributes = {}) => {
   const widthAttr = dom.getAttribute('data-colwidth');
   const width =
     widthAttr && /^\d+(,\d+)*$/.test(widthAttr)

package/src/steps/__tests__/unit/override-document-step.ts

@@ -23,8 +23,8 @@ describe('Override Document Step', () => {
         }),
       );
 
-      expect(tr.doc).toEqualDocument(nextDocument);
-      expect(tr.docs[0]).toEqualDocument(currentDocument(defaultSchema));
+      expect(tr.doc).toEqual(nextDocument);
+      expect(tr.docs[0]).toEqual(currentDocument(defaultSchema));
     });
 
     describe('when the inverted step is applied', () => {
@@ -40,9 +40,9 @@ describe('Override Document Step', () => {
         tr.step(step);
         tr.step(stepInverted);
 
-        expect(tr.doc).toEqualDocument(currentDocument(defaultSchema));
-        expect(tr.docs[1]).toEqualDocument(nextDocument);
-        expect(tr.docs[0]).toEqualDocument(currentDocument(defaultSchema));
+        expect(tr.doc).toEqual(currentDocument(defaultSchema));
+        expect(tr.docs[1]).toEqual(nextDocument);
+        expect(tr.docs[0]).toEqual(currentDocument(defaultSchema));
       });
     });
 
@@ -166,7 +166,7 @@ describe('Override Document Step', () => {
         currentDocument(defaultSchema),
       );
 
-      expect(stepAppliedResult.doc).toEqualDocument(
+      expect(stepAppliedResult.doc).toMatchObject(
         deserializedStepAppliedResult.doc,
       );
     });

package/src/steps/table/__tests__/_utils.ts

@@ -1,12 +1,11 @@
 import { Node as ProseMirrorNode } from '@atlaskit/editor-prosemirror/model';
-import { EditorState, Plugin, Transaction } from '@atlaskit/editor-prosemirror/state';
-import { history, undo } from '@atlaskit/editor-prosemirror/history';
 import {
-  DocBuilder,
-  p,
-  Refs,
-  td,
-} from 'adf-test-helpers/src/doc-builder';
+  EditorState,
+  Plugin,
+  Transaction,
+} from '@atlaskit/editor-prosemirror/state';
+import { history, undo } from '@atlaskit/editor-prosemirror/history';
+import { DocBuilder, p, Refs, td } from 'adf-test-helpers/src/doc-builder';
 import { setSelectionTransform } from 'adf-test-helpers/src/set-selection-transform';
 
 import { defaultSchema } from '../../../schema/default-schema';

package/src/steps/table/__tests__/add-column-merge-columns.ts

@@ -147,7 +147,9 @@ describe('AddColumnStep', () => {
             addColumnAtFactory('table', column)(editorState, refs),
           );
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
     });
@@ -165,7 +167,9 @@ describe('AddColumnStep', () => {
             removeColumnAtFactory('table', column)(editorState, refs),
           );
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
     });
@@ -195,7 +199,9 @@ describe('AddColumnStep', () => {
           transaction,
           editorState,
         );
-        expect(editorState.doc).toEqualDocument(originalDoc);
+        expect(editorState.doc.toJSON()).toEqual(
+          originalDoc(defaultSchema).toJSON(),
+        );
       });
     });
 
@@ -285,7 +291,9 @@ describe('AddColumnStep', () => {
 
           editorState = editorState.apply(editorState.tr.step(addColumnStep!));
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
     });
@@ -359,7 +367,9 @@ describe('AddColumnStep', () => {
               historyTransactions,
             );
 
-            expect(editorState.doc).toEqualDocument(expectDoc);
+            expect(editorState.doc.toJSON()).toEqual(
+              expectDoc(defaultSchema).toJSON(),
+            );
           });
         },
       );
@@ -380,7 +390,9 @@ describe('AddColumnStep', () => {
           const addColumnStep = Step.fromJSON(defaultSchema, jsonStep);
           editorState = editorState.apply(editorState.tr.step(addColumnStep));
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
 
@@ -398,7 +410,9 @@ describe('AddColumnStep', () => {
           const addColumnStep = Step.fromJSON(defaultSchema, jsonStep);
           editorState = editorState.apply(editorState.tr.step(addColumnStep));
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
 
@@ -413,7 +427,9 @@ describe('AddColumnStep', () => {
           const addColumnStep = Step.fromJSON(defaultSchema, jsonStep);
           editorState = editorState.apply(editorState.tr.step(addColumnStep));
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
 
@@ -430,7 +446,9 @@ describe('AddColumnStep', () => {
           const addColumnStep = Step.fromJSON(defaultSchema, jsonStep);
           editorState = editorState.apply(editorState.tr.step(addColumnStep));
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
     });

package/src/steps/table/__tests__/add-column-merge-rows.ts

@@ -166,7 +166,9 @@ describe('AddColumnStep', () => {
             addColumnAtFactory('table', column)(editorState, refs),
           );
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
     });
@@ -184,7 +186,9 @@ describe('AddColumnStep', () => {
             removeColumnAtFactory('table', column)(editorState, refs),
           );
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
     });
@@ -214,7 +218,9 @@ describe('AddColumnStep', () => {
           transaction,
           editorState,
         );
-        expect(editorState.doc).toEqualDocument(originalDoc);
+        expect(editorState.doc.toJSON()).toEqual(
+          originalDoc(defaultSchema).toJSON(),
+        );
       });
     });
 
@@ -323,7 +329,9 @@ describe('AddColumnStep', () => {
 
           editorState = editorState.apply(editorState.tr.step(addColumnStep!));
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
     });
@@ -343,7 +351,9 @@ describe('AddColumnStep', () => {
           const addColumnStep = Step.fromJSON(defaultSchema, jsonStep);
           editorState = editorState.apply(editorState.tr.step(addColumnStep));
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
 
@@ -361,7 +371,9 @@ describe('AddColumnStep', () => {
           const addColumnStep = Step.fromJSON(defaultSchema, jsonStep);
           editorState = editorState.apply(editorState.tr.step(addColumnStep));
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
 
@@ -376,7 +388,9 @@ describe('AddColumnStep', () => {
           const addColumnStep = Step.fromJSON(defaultSchema, jsonStep);
           editorState = editorState.apply(editorState.tr.step(addColumnStep));
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
 
@@ -393,7 +407,9 @@ describe('AddColumnStep', () => {
           const addColumnStep = Step.fromJSON(defaultSchema, jsonStep);
           editorState = editorState.apply(editorState.tr.step(addColumnStep));
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
     });

package/src/steps/table/__tests__/add-column.ts

@@ -119,7 +119,9 @@ describe('AddColumnStep', () => {
         removeColumnAtFactory('table', 0)(editorState, refs),
       );
 
-      expect(editorState.doc).toEqualDocument(doc(p('')));
+      expect(editorState.doc.toJSON()).toEqual(
+        doc(p(''))(defaultSchema).toJSON(),
+      );
     });
 
     it('should add the table after been removed it for a remove column', () => {
@@ -127,7 +129,9 @@ describe('AddColumnStep', () => {
         removeColumnAtFactory('table', 0)(editorState, refs),
         editorState,
       );
-      expect(editorState.doc).toEqualDocument(threeByOne);
+      expect(editorState.doc.toJSON()).toEqual(
+        threeByOne(defaultSchema).toJSON(),
+      );
     });
 
     it('should remove the table after undoing the last column ', function () {
@@ -137,7 +141,9 @@ describe('AddColumnStep', () => {
         [removeColumnAtFactory('table', 1)],
       );
 
-      expect(editorState.doc).toEqualDocument(doc(p('')));
+      expect(editorState.doc.toJSON()).toEqual(
+        doc(p(''))(defaultSchema).toJSON(),
+      );
     });
 
     it('should drop add column step when I remove the column', function () {
@@ -176,10 +182,14 @@ describe('AddColumnStep', () => {
 
       editorState = editorState.apply(editorState.tr.step(addColumnStep));
 
+      const expectedDoc = doc(
+        p(''),
+        '{table}',
+        table()(tr(tdEmpty), tr(tdEmpty), tr(tdEmpty)),
+      )(defaultSchema).toJSON();
+
       // Should keep the new created single column and remove the old one
-      expect(editorState.doc).toEqualDocument(
-        doc(p(''), '{table}', table()(tr(tdEmpty), tr(tdEmpty), tr(tdEmpty))),
-      );
+      expect(editorState.doc.toJSON()).toEqual(expectedDoc);
     });
 
     it('should parse a json that we just generate', function () {
@@ -216,7 +226,9 @@ describe('AddColumnStep', () => {
         editorState.tr.step(removeLastRowInverted),
       );
 
-      expect(editorState.doc).toEqualDocument(threeByOne);
+      expect(editorState.doc.toJSON()).toEqual(
+        threeByOne(defaultSchema).toJSON(),
+      );
     });
 
     it('should remove the table when I received a JSON', () => {
@@ -229,7 +241,9 @@ describe('AddColumnStep', () => {
         editorState.tr.step(removeLastRowInverted),
       );
 
-      expect(editorState.doc).toEqualDocument(doc(p('')));
+      expect(editorState.doc.toJSON()).toEqual(
+        doc(p(''))(defaultSchema).toJSON(),
+      );
     });
   });
 
@@ -313,7 +327,9 @@ describe('AddColumnStep', () => {
             addColumnAtFactory('table', column)(editorState, refs),
           );
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
     });
@@ -331,7 +347,9 @@ describe('AddColumnStep', () => {
             removeColumnAtFactory('table', column)(editorState, refs),
           );
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
     });
@@ -363,7 +381,9 @@ describe('AddColumnStep', () => {
             addColumnAtFactory('table', column)(editorState, refs),
           );
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
     });
@@ -381,7 +401,9 @@ describe('AddColumnStep', () => {
             removeColumnAtFactory('table', column)(editorState, refs),
           );
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
     });
@@ -432,7 +454,7 @@ describe('AddColumnStep', () => {
           transaction,
           editorState,
         );
-        expect(editorState.doc).toEqualDocument(originalDoc);
+        expect(editorState.doc.toJSON()).toEqual(originalDoc.toJSON());
       });
     });
 
@@ -496,7 +518,9 @@ describe('AddColumnStep', () => {
 
           editorState = editorState.apply(editorState.tr.step(addColumnStep!));
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
     });
@@ -509,13 +533,13 @@ describe('AddColumnStep', () => {
           [addColumnAtFactory('table', 1), addColumnAtFactory('table', 1)],
         );
 
-        expect(editorState.doc).toEqualDocument(
+        expect(editorState.doc.toJSON()).toEqual(
           doc(
             table()(
               tr(tdColorA, tdEmpty, tdEmpty, tdColorB, tdColorC),
               tr(tdColorA, tdEmpty, tdEmpty, tdColorB, tdColorC),
             ),
-          ),
+          )(defaultSchema).toJSON(),
         );
       });
     });
@@ -533,7 +557,9 @@ describe('AddColumnStep', () => {
           const addColumnStep = Step.fromJSON(defaultSchema, jsonStep);
           editorState = editorState.apply(editorState.tr.step(addColumnStep));
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
 
@@ -548,7 +574,9 @@ describe('AddColumnStep', () => {
           const addColumnStep = Step.fromJSON(defaultSchema, jsonStep);
           editorState = editorState.apply(editorState.tr.step(addColumnStep));
 
-          expect(editorState.doc).toEqualDocument(expectedDoc);
+          expect(editorState.doc.toJSON()).toEqual(
+            expectedDoc(defaultSchema).toJSON(),
+          );
         },
       );
     });

package/src/steps/table/add-column.ts

@@ -1,4 +1,7 @@
-import { Node as ProseMirrorNode, Schema } from '@atlaskit/editor-prosemirror/model';
+import {
+  Node as ProseMirrorNode,
+  Schema,
+} from '@atlaskit/editor-prosemirror/model';
 import {
   Mappable,
   Step,

package/src/steps/table/utils/cells-at-column.ts

@@ -21,7 +21,10 @@ export interface Cell {
  * @param rect
  * @param col
  */
-export function* cellsAtColumn(rect: TableRect, col: number): Generator<Cell, void, number | undefined> {
+export function* cellsAtColumn(
+  rect: TableRect,
+  col: number,
+): Generator<Cell, void, number | undefined> {
   const { map, tableStart, table } = rect;
 
   let refColumn: number | null = col > 0 ? -1 : 0;