Fixed stale doc in replaceStep.ts, added test.

Author: VV-EE

src/__tests__/withSuggestChanges.test.ts

@@ -0,0 +1,78 @@
+ 
+/* eslint-disable @typescript-eslint/no-non-null-assertion */
+import { EditorState } from "prosemirror-state";
+import { Fragment, Slice } from "prosemirror-model";
+import { type ReplaceStep, replaceStep } from "prosemirror-transform";
+import { assert, describe, expect, it } from "vitest";
+
+import { type TaggedNode, testBuilders } from "../testing/testBuilders.js";
+import { suggestChanges } from "../plugin.js";
+import { transformToSuggestionTransaction } from "../withSuggestChanges.js";
+
+describe("withSuggestChanges", () => {
+  it("should wrap an insertion in a mark", () => {
+    // Create document with all tags at once
+    const doc = testBuilders.doc(
+      testBuilders.paragraph({ id: null }, "This is a <a>test<b> paragraph."),
+      testBuilders.paragraph(
+        { id: null },
+        "This is a <c>test<d> <e>test<f> paragraph.",
+      ),
+    ) as TaggedNode;
+
+    const editorState = EditorState.create({
+      doc,
+      plugins: [suggestChanges()],
+    });
+
+    const originalTransaction = editorState.tr;
+
+    // Step 1: Replace "test" with "WORK" in first paragraph
+    const step1 = replaceStep(
+      doc,
+      doc.tag["a"]!,
+      doc.tag["b"],
+      new Slice(Fragment.from(testBuilders.schema.text("WORK")), 0, 0),
+    ) as ReplaceStep | null;
+    assert(step1, "Could not create step1");
+    originalTransaction.step(step1);
+
+    // Step 2: Replace first "test" with "WORK" in second paragraph
+    // Use the updated document from the transaction
+    const step2 = replaceStep(
+      originalTransaction.doc,
+      doc.tag["c"]!,
+      doc.tag["d"],
+      new Slice(Fragment.from(testBuilders.schema.text("WORK")), 0, 0),
+    ) as ReplaceStep | null;
+    assert(step2, "Could not create step2");
+    originalTransaction.step(step2);
+
+    // Step 3: Replace second "test" with "WORK" in second paragraph
+    // Use the updated document from the transaction
+    const step3 = replaceStep(
+      originalTransaction.doc,
+      doc.tag["e"]!,
+      doc.tag["f"],
+      new Slice(Fragment.from(testBuilders.schema.text("WORK")), 0, 0),
+    ) as ReplaceStep | null;
+    assert(step3, "Could not create step3");
+    originalTransaction.step(step3);
+
+    // Verify the transaction produces expected result
+    const originalTrState = editorState.apply(originalTransaction);
+    expect(originalTrState.doc.textContent).toEqual(
+      "This is a WORK paragraph.This is a WORK WORK paragraph.",
+    );
+
+    // Transform to suggestion transaction
+    const suggestedTr = transformToSuggestionTransaction(
+      originalTransaction,
+      editorState,
+    );
+    const newState = editorState.apply(suggestedTr);
+
+    // Should pass without error - document should be modified
+    expect(newState.doc.toJSON()).not.toEqual(doc.toJSON());
+  });
+});

src/replaceStep.ts

@@ -84,7 +84,9 @@ export function suggestReplaceStep(
   let stepTo = rebasePos(step.to, prevSteps, trackedTransaction.steps);
 
   if (state.selection.empty && stepFrom !== stepTo) {
-    trackedTransaction.setSelection(TextSelection.near(doc.resolve(stepFrom)));
+    trackedTransaction.setSelection(
+      TextSelection.near(trackedTransaction.doc.resolve(stepFrom)),
+    );
   }
 
   // Make a list of any existing insertions that fall within the