Execute resolved dot payloads

1 files changed, 4 insertions, 0 deletions

diff --git a/lua/clever_f/testing/memory_host.lua b/lua/clever_f/testing/memory_host.lua
index 8b473ad..42dab71 100644
--- a/lua/clever_f/testing/memory_host.lua
+++ b/lua/clever_f/testing/memory_host.lua
@@ -952,6 +952,8 @@ function MemoryHost:register_dot_repeat(payload, callback)
   self._dot_repeat = {
     payload = payload,
     callback = callback,
+    operator = self._pending_operator,
+    mode = self._mode,
   }
   self:_record("register_dot_repeat", { payload = payload:to_table() })
   return payload
@@ -965,6 +967,8 @@ function MemoryHost:replay_dot(count)
   if self._dot_repeat == nil or self._dot_repeat.callback == nil then
     error("dot repeat is not executable", 2)
   end
+  self._pending_operator = self._dot_repeat.operator
+  self._mode = self._dot_repeat.mode
   return self._dot_repeat.callback(self._dot_repeat.payload, domain.Count.new(count))
 end