test(replay): replay full session reload path

Author: cameronr

tests/helpers.lua

@@ -21,6 +21,8 @@ function M.replay_setup()
   renderer._cleanup_subscriptions()
   renderer.reset()
 
+  M.mock_time_ago()
+
   if not config.config then
     config.config = vim.deepcopy(config.defaults)
   end
@@ -143,6 +145,56 @@ function M.load_test_data(filename)
   return vim.json.decode(content)
 end
 
+function M.load_session_from_events(events)
+  local session_data = {}
+
+  for _, event in ipairs(events) do
+    local properties = event.properties
+
+    if event.type == 'message.updated' and properties.info then
+      local msg = properties.info
+      local existing_msg = nil
+      for _, m in ipairs(session_data) do
+        if m.info.id == msg.id then
+          existing_msg = m
+          break
+        end
+      end
+
+      if existing_msg then
+        existing_msg.info = vim.deepcopy(msg)
+      else
+        table.insert(session_data, {
+          info = vim.deepcopy(msg),
+          parts = {},
+        })
+      end
+    elseif event.type == 'message.part.updated' and properties.part then
+      local part = properties.part
+      for _, msg in ipairs(session_data) do
+        if msg.info.id == part.messageID then
+          local existing_part = nil
+          for i, p in ipairs(msg.parts) do
+            if p.id == part.id then
+              existing_part = i
+              break
+            end
+          end
+
+          if existing_part then
+            msg.parts[existing_part] = vim.deepcopy(part)
+          else
+            table.insert(msg.parts, vim.deepcopy(part))
+          end
+          break
+        end
+      end
+    end
+  end
+
+  return session_data
+end
+
 function M.get_session_from_events(events)
   -- renderer needs a valid session id
   for _, event in ipairs(events) do
@@ -185,7 +237,8 @@ end
 
 function M.normalize_namespace_ids(extmarks)
   local normalized = vim.deepcopy(extmarks)
-  for _, mark in ipairs(normalized) do
+  for i, mark in ipairs(normalized) do
+    mark[1] = i
     if mark[4] and mark[4].ns_id then
       mark[4].ns_id = 3
     end

tests/manual/renderer_replay.lua

@@ -1,7 +1,5 @@
 local state = require('opencode.state')
 local renderer = require('opencode.ui.renderer')
-local ui = require('opencode.ui.ui')
-local config_file = require('opencode.config_file')
 local helpers = require('tests.helpers')
 local output_window = require('opencode.ui.output_window')
 
@@ -12,7 +10,6 @@ M.current_index = 0
 M.timer = nil
 M.last_loaded_file = nil
 M.headless_mode = false
-M.restore_time_ago = nil
 
 function M.load_events(file_path)
   file_path = file_path or 'tests/data/simple-session.json'
@@ -106,6 +103,7 @@ function M.replay_all(delay_ms)
   delay_ms = delay_ms or 50
 
   if M.timer then
+    ---@diagnostic disable-next-line: undefined-field
     M.timer:stop()
     M.timer = nil
   end
@@ -116,12 +114,14 @@ function M.replay_all(delay_ms)
   end
 
   M.timer = vim.loop.new_timer()
+  ---@diagnostic disable-next-line: undefined-field
   M.timer:start(
     0,
     delay_ms,
     vim.schedule_wrap(function()
       if M.current_index >= #M.events then
         if M.timer then
+          ---@diagnostic disable-next-line: undefined-field
           M.timer:stop()
           M.timer = nil
         end
@@ -139,6 +139,7 @@ end
 
 function M.replay_stop()
   if M.timer then
+    ---@diagnostic disable-next-line: undefined-field
     M.timer:stop()
     M.timer = nil
     vim.notify('Replay stopped at event ' .. M.current_index .. '/' .. #M.events, vim.log.levels.INFO)
@@ -210,6 +211,21 @@ function M.save_output(filename)
   return snapshot
 end
 
+function M.replay_full_session()
+  if #M.events == 0 then
+    vim.notify('No events loaded. Use :ReplayLoad first.', vim.log.levels.WARN)
+    return false
+  end
+
+  local session_data = helpers.load_session_from_events(M.events)
+
+  renderer.reset()
+  renderer._render_full_session_data(session_data)
+
+  vim.notify('Rendered full session from loaded events', vim.log.levels.INFO)
+  return true
+end
+
 function M.dump_buffer_and_quit()
   vim.schedule(function()
     if not state.windows or not state.windows.output_buf then
@@ -269,23 +285,29 @@ function M.start(opts)
     'Streaming Renderer Replay',
     '',
     'Use :ReplayLoad [file] to load event data',
+    'Use :ReplayFullSession to render loaded events using full session mode',
     '',
     'Commands:',
-    '  :ReplayLoad [file]     - Load events (default: tests/data/simple-session.json)',
-    "  :ReplayNext [step]     - Replay next [step] event(s) (default 1) (<leader>n or '>' )",
-    '  :ReplayAll [ms]        - Replay all events with delay (default 50ms) (<leader>a)',
-    '  :ReplayStop            - Stop auto-replay (<leader>s)',
-    '  :ReplayReset           - Reset to beginning (<leader>r)',
-    '  :ReplayClear           - Clear output buffer (<leader>c)',
-    '  :ReplaySave [file]     - Save snapshot (auto-derives from loaded file)',
-    '  :ReplayStatus          - Show status',
+    '  :ReplayLoad [file]        - Load events (default: tests/data/simple-session.json)',
+    '  :ReplayFullSession        - Render loaded events using full session mode',
+    "  :ReplayNext [step]        - Replay next [step] event(s) (default 1) (<leader>n or '>' )",
+    '  :ReplayAll [ms]           - Replay all events with delay (default 50ms) (<leader>a)',
+    '  :ReplayStop               - Stop auto-replay (<leader>s)',
+    '  :ReplayReset              - Reset to beginning (<leader>r)',
+    '  :ReplayClear              - Clear output buffer (<leader>c)',
+    '  :ReplaySave [file]        - Save snapshot (auto-derives from loaded file)',
+    '  :ReplayStatus             - Show status',
   })
 
   vim.api.nvim_create_user_command('ReplayLoad', function(cmd_opts)
     local file = cmd_opts.args ~= '' and cmd_opts.args or nil
     M.load_events(file)
   end, { nargs = '?', desc = 'Load event data file', complete = 'file' })
 
+  vim.api.nvim_create_user_command('ReplayFullSession', function()
+    M.replay_full_session()
+  end, { desc = 'Render loaded events using full session mode' })
+
   vim.api.nvim_create_user_command('ReplayNext', function(cmd_opts)
     local steps = cmd_opts.args ~= '' and cmd_opts.args or nil
     M.replay_next(steps)

tests/unit/renderer_spec.lua

@@ -8,17 +8,12 @@ describe('renderer', function()
 
   before_each(function()
     helpers.replay_setup()
-    restore_time_ago = helpers.mock_time_ago()
   end)
 
   after_each(function()
     if state.windows then
       ui.close_windows(state.windows)
     end
-
-    if restore_time_ago then
-      restore_time_ago()
-    end
   end)
 
   local json_files = vim.fn.glob('tests/data/*.json', false, true)
@@ -30,7 +25,7 @@ describe('renderer', function()
       local expected_path = 'tests/data/' .. name .. '.expected.json'
 
       if vim.fn.filereadable(expected_path) == 1 then
-        it('replays ' .. name .. ' correctly', function()
+        it('replays ' .. name .. ' correctly (event-by-event)', function()
           local events = helpers.load_test_data(filepath)
           state.active_session = helpers.get_session_from_events(events)
           local expected = helpers.load_test_data(expected_path)
@@ -43,6 +38,22 @@ describe('renderer', function()
           assert.are.same(expected.lines, actual.lines)
           assert.are.same(expected.extmarks, helpers.normalize_namespace_ids(actual.extmarks))
         end)
+
+        it('replays ' .. name .. ' correctly (full session load)', function()
+          local renderer = require('opencode.ui.renderer')
+          local events = helpers.load_test_data(filepath)
+          state.active_session = helpers.get_session_from_events(events)
+          local expected = helpers.load_test_data(expected_path)
+
+          local session_data = helpers.load_session_from_events(events)
+          renderer._render_full_session_data(session_data)
+          vim.wait(200)
+
+          local actual = helpers.capture_output(state.windows.output_buf, output_window.namespace)
+
+          assert.are.same(expected.lines, actual.lines)
+          assert.are.same(expected.extmarks, helpers.normalize_namespace_ids(actual.extmarks))
+        end)
       end
     end
   end