From 6c847422df31c751b51eaf42a8bd447ae640496b Mon Sep 17 00:00:00 2001
From: Joshua Horton <joshua_horton@sil.org>
Date: Mon, 8 Sep 2025 07:46:11 -0500
Subject: [PATCH] fix(web): patch up unit tests for change to worker reversion
 pattern

---
 .../worker-thread/src/main/model-compositor.ts           | 8 ++++----
 .../worker-thread/worker-model-compositor.tests.ts       | 9 +++++----
 2 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/web/src/engine/predictive-text/worker-thread/src/main/model-compositor.ts b/web/src/engine/predictive-text/worker-thread/src/main/model-compositor.ts
index 17289eb56b..fd83e645bf 100644
--- a/web/src/engine/predictive-text/worker-thread/src/main/model-compositor.ts
+++ b/web/src/engine/predictive-text/worker-thread/src/main/model-compositor.ts
@@ -223,17 +223,17 @@ export class ModelCompositor {
    *
    * @param suggestion Suggestion selected, whether automatically or by the user.
    * @param context The context to which the suggestion should be applied.
-   * @param postTransform The original transform that is being replaced by the applied suggestion (!)
+   * @param originalInput The original transform that is being replaced by the applied suggestion (!)
    * @returns
    */
-  acceptSuggestion(suggestion: Suggestion, context: Context, postTransform?: Transform): Reversion {
+  acceptSuggestion(suggestion: Suggestion, context: Context, originalInput?: Transform): Reversion {
     // Step 1:  re-use the original input Transform as the reversion's Transform.
     // The Web engine will restore the original state of the context before accepting
     // and before reverting; all we need to do is put the original keystroke back in place.
-    let reversionTransform: Transform = postTransform;
+    let reversionTransform: Transform = originalInput ?? { insert: '', deleteLeft: 0 };
 
     // Step 2:  building the proper 'displayAs' string for the Reversion
-    const postContext = models.applyTransform(postTransform, context);
+    const postContext = originalInput ? models.applyTransform(originalInput, context) : context;
 
     let revertedPrefix: string;
     let postContextTokenization = this.tokenize(postContext);
diff --git a/web/src/test/auto/headless/engine/predictive-text/worker-thread/worker-model-compositor.tests.ts b/web/src/test/auto/headless/engine/predictive-text/worker-thread/worker-model-compositor.tests.ts
index 1134f5b0a7..7181dbd4bf 100644
--- a/web/src/test/auto/headless/engine/predictive-text/worker-thread/worker-model-compositor.tests.ts
+++ b/web/src/test/auto/headless/engine/predictive-text/worker-thread/worker-model-compositor.tests.ts
@@ -663,7 +663,7 @@ describe('ModelCompositor', function() {
       let appliedContext = models.applyTransform(baseSuggestion.transform, baseContext);
       assert.equal(appliedContext.left, "hello ");
 
-      let revertedContext = models.applyTransform(reversion.transform, appliedContext);
+      let revertedContext = models.applyTransform(reversion.transform, baseContext);
       assert.deepEqual(revertedContext, unappliedContext);
 
       // Check #2:  Are the correct display strings built, depending on the active model's punctuation?
@@ -706,7 +706,7 @@ describe('ModelCompositor', function() {
       let appliedContext = models.applyTransform(baseSuggestion.transform, baseContext);
       assert.equal(appliedContext.left, "hello world ");
 
-      let revertedContext = models.applyTransform(reversion.transform, appliedContext);
+      let revertedContext = models.applyTransform(reversion.transform, baseContext);
       assert.deepEqual(revertedContext, unappliedContext);
 
       // Check #2:  Are the correct display strings built, depending on the active model's punctuation?
@@ -742,7 +742,7 @@ describe('ModelCompositor', function() {
       let appliedContext = models.applyTransform(baseSuggestion.transform, baseContext);
       assert.equal(appliedContext.left, "hello world ");
 
-      let revertedContext = models.applyTransform(reversion.transform, appliedContext);
+      let revertedContext = models.applyTransform(reversion.transform, baseContext);
       assert.deepEqual(revertedContext, unappliedContext);
 
       // Check #2:  Are the correct display strings built, depending on the active model's punctuation?
@@ -785,7 +785,8 @@ describe('ModelCompositor', function() {
       let appliedContext = models.applyTransform(baseSuggestion.transform, baseContext);
       assert.equal(appliedContext.left, "hello ");
 
-      let revertedContext = models.applyTransform(reversion.transform, appliedContext);
+      let revertedContext = models.applyTransform(reversion.transform, baseContext);
+      assert.equal(revertedContext.left, "hi");
       assert.deepEqual(revertedContext, unappliedContext);
 
       // Check #2:  Are the correct display strings built, depending on the active model's punctuation?