add selection assertions to addMark tests

test/transforms/fixtures/add-mark/across-inlines/index.js

@@ -1,18 +1,25 @@
 
+import assert from 'assert'
+
 export default function (state) {
   const { document, selection } = state
   const texts = document.getTexts()
   const first = texts.first()
   const second = texts.last()
+  const range = selection.merge({
+    anchorKey: first.key,
+    anchorOffset: 2,
+    focusKey: second.key,
+    focusOffset: 2
+  })
 
-  return state
+  const next = state
     .transform()
-    .moveTo({
-      anchorKey: first.key,
-      anchorOffset: 2,
-      focusKey: second.key,
-      focusOffset: 2
-    })
+    .moveTo(range)
     .addMark('bold')
     .apply()
+
+  assert.deepEqual(next.selection.toJS(), range.toJS())
+
+  return next
 }

test/transforms/fixtures/add-mark/collapsed-selection-off-again/index.js

@@ -1,20 +1,28 @@
 
+import assert from 'assert'
+
 export default function (state) {
   const { document, selection } = state
   const texts = document.getTexts()
   const first = texts.first()
+  const range = selection.merge({
+    anchorKey: first.key,
+    anchorOffset: 0,
+    focusKey: first.key,
+    focusOffset: 0
+  })
 
-  return state
+  const next = state
     .transform()
     .moveTo({
-      anchorKey: first.key,
-      anchorOffset: 0,
-      focusKey: first.key,
-      focusOffset: 0
     })
     .addMark('bold')
     .insertText('a')
     .removeMark('bold')
     .insertText('b')
     .apply()
+
+  assert.deepEqual(next.selection.toJS(), range.moveForward().moveForward().toJS())
+
+  return next
 }

test/transforms/fixtures/add-mark/collapsed-selection/index.js

@@ -1,18 +1,25 @@
 
+import assert from 'assert'
+
 export default function (state) {
   const { document, selection } = state
   const texts = document.getTexts()
   const first = texts.first()
+  const range = selection.merge({
+    anchorKey: first.key,
+    anchorOffset: 0,
+    focusKey: first.key,
+    focusOffset: 0
+  })
 
-  return state
+  const next = state
     .transform()
-    .moveTo({
-      anchorKey: first.key,
-      anchorOffset: 0,
-      focusKey: first.key,
-      focusOffset: 0
-    })
+    .moveTo(range)
     .addMark('bold')
     .insertText('a')
     .apply()
+
+  assert.deepEqual(next.selection.toJS(), range.moveForward().toJS())
+
+  return next
 }

test/transforms/fixtures/add-mark/existing-marks/index.js

@@ -1,17 +1,24 @@
 
+import assert from 'assert'
+
 export default function (state) {
   const { document, selection } = state
   const texts = document.getTexts()
   const first = texts.first()
+  const range = selection.merge({
+    anchorKey: first.key,
+    anchorOffset: 0,
+    focusKey: first.key,
+    focusOffset: 2
+  })
 
-  return state
+  const next = state
     .transform()
-    .moveTo({
-      anchorKey: first.key,
-      anchorOffset: 0,
-      focusKey: first.key,
-      focusOffset: 2
-    })
+    .moveTo(range)
     .addMark('bold')
     .apply()
+
+  assert.deepEqual(next.selection.toJS(), range.toJS())
+
+  return next
 }

test/transforms/fixtures/add-mark/first-character/index.js

@@ -1,17 +1,24 @@
 
+import assert from 'assert'
+
 export default function (state) {
   const { document, selection } = state
   const texts = document.getTexts()
   const first = texts.first()
+  const range = selection.merge({
+    anchorKey: first.key,
+    anchorOffset: 0,
+    focusKey: first.key,
+    focusOffset: 1
+  })
 
-  return state
+  const next = state
     .transform()
-    .moveTo({
-      anchorKey: first.key,
-      anchorOffset: 0,
-      focusKey: first.key,
-      focusOffset: 1
-    })
+    .moveTo(range)
     .addMark('bold')
     .apply()
+
+  assert.deepEqual(next.selection.toJS(), range.toJS())
+
+  return next
 }

test/transforms/fixtures/add-mark/last-character/index.js

@@ -1,17 +1,24 @@
 
+import assert from 'assert'
+
 export default function (state) {
   const { document, selection } = state
   const texts = document.getTexts()
   const first = texts.first()
+  const range = selection.merge({
+    anchorKey: first.key,
+    anchorOffset: first.length - 1,
+    focusKey: first.key,
+    focusOffset: first.length
+  })
 
-  return state
+  const next = state
     .transform()
-    .moveTo({
-      anchorKey: first.key,
-      anchorOffset: first.length - 1,
-      focusKey: first.key,
-      focusOffset: first.length
-    })
+    .moveTo(range)
     .addMark('bold')
     .apply()
+
+  assert.deepEqual(next.selection.toJS(), range.toJS())
+
+  return next
 }

test/transforms/fixtures/add-mark/middle-character/index.js

@@ -1,17 +1,24 @@
 
+import assert from 'assert'
+
 export default function (state) {
   const { document, selection } = state
   const texts = document.getTexts()
   const first = texts.first()
+  const range = selection.merge({
+    anchorKey: first.key,
+    anchorOffset: 1,
+    focusKey: first.key,
+    focusOffset: 2
+  })
 
-  return state
+  const next = state
     .transform()
-    .moveTo({
-      anchorKey: first.key,
-      anchorOffset: 1,
-      focusKey: first.key,
-      focusOffset: 2
-    })
+    .moveTo(range)
     .addMark('bold')
     .apply()
+
+  assert.deepEqual(next.selection.toJS(), range.toJS())
+
+  return next
 }

test/transforms/fixtures/add-mark/whole-word/index.js

@@ -1,17 +1,24 @@
 
+import assert from 'assert'
+
 export default function (state) {
   const { document, selection } = state
   const texts = document.getTexts()
   const first = texts.first()
+  const range = selection.merge({
+    anchorKey: first.key,
+    anchorOffset: 0,
+    focusKey: first.key,
+    focusOffset: first.length
+  })
 
-  return state
+  const next = state
     .transform()
-    .moveTo({
-      anchorKey: first.key,
-      anchorOffset: 0,
-      focusKey: first.key,
-      focusOffset: first.length
-    })
+    .moveTo(range)
     .addMark('bold')
     .apply()
+
+  assert.deepEqual(next.selection.toJS(), range.toJS())
+
+  return next
 }

test/transforms/fixtures/add-mark/with-mark-object/index.js

@@ -1,22 +1,28 @@
 
 import { Mark } from '../../../../..'
+import assert from 'assert'
 
 export default function (state) {
   const { document, selection } = state
   const texts = document.getTexts()
   const first = texts.first()
+  const range = selection.merge({
+    anchorKey: first.key,
+    anchorOffset: 0,
+    focusKey: first.key,
+    focusOffset: 1
+  })
 
-  return state
+  const next = state
     .transform()
-    .moveTo({
-      anchorKey: first.key,
-      anchorOffset: 0,
-      focusKey: first.key,
-      focusOffset: 1
-    })
+    .moveTo(range)
     .addMark(Mark.create({
       type: 'bold',
       data: { key: 'value' }
     }))
     .apply()
+
+  assert.deepEqual(next.selection.toJS(), range.toJS())
+
+  return next
 }

test/transforms/fixtures/add-mark/with-plain-object/index.js

@@ -1,20 +1,27 @@
 
+import assert from 'assert'
+
 export default function (state) {
   const { document, selection } = state
   const texts = document.getTexts()
   const first = texts.first()
+  const range = selection.merge({
+    anchorKey: first.key,
+    anchorOffset: 0,
+    focusKey: first.key,
+    focusOffset: 1
+  })
 
-  return state
+  const next = state
     .transform()
-    .moveTo({
-      anchorKey: first.key,
-      anchorOffset: 0,
-      focusKey: first.key,
-      focusOffset: 1
-    })
+    .moveTo(range)
     .addMark({
       type: 'bold',
       data: { key: 'value' }
     })
     .apply()
+
+  assert.deepEqual(next.selection.toJS(), range.toJS())
+
+  return next
 }