test(repeatable_move): add operator-pending mode test cases

Author: Illia Shkroba

tests/repeatable_move/common.lua

@@ -20,7 +20,7 @@ function M.run_builtin_find_test(file, spec)
   for col = 0, num_cols - 1 do
     for _, cmd in pairs({ 'f', 'F', 't', 'T' }) do
       for _, repeat_cmd in pairs({ ';', ',' }) do
-        -- Get ground truth using vim's built-in search and repeat
+        -- Get ground truth using vim's built-in search and repeat (normal mode)
         vim.api.nvim_win_set_cursor(0, { spec.row, col })
         local gt_cols = {}
         vim.cmd([[normal! ]] .. cmd .. spec.char)
@@ -36,7 +36,7 @@ function M.run_builtin_find_test(file, spec)
         vim.cmd([[normal! 2]] .. cmd .. spec.char)
         gt_cols[#gt_cols + 1] = vim.fn.col('.')
 
-        -- test using tstextobj repeatable_move.lua
+        -- test using tstextobj repeatable_move.lua (normal mode)
         vim.api.nvim_win_set_cursor(0, { spec.row, col })
         local ts_cols = {}
         vim.cmd([[normal ]] .. cmd .. spec.char)
@@ -61,7 +61,82 @@ function M.run_builtin_find_test(file, spec)
         assert.are.same(
           gt_cols,
           ts_cols,
-          string.format("Command %s works differently than vim's built-in find, col: %d", cmd, col)
+          string.format(
+            "Command %s with repeat %s works differently than vim's built-in find, col: %d",
+            cmd,
+            repeat_cmd,
+            col
+          )
+        )
+
+        -- Get ground truth using vim's built-in search and repeat (operator-pending mode)
+        vim.api.nvim_win_set_cursor(0, { spec.row, col })
+        local gt_regs = {}
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal! y]] .. cmd .. spec.char)
+        gt_regs[#gt_regs + 1] = vim.fn.getreg('0')
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal! y]] .. repeat_cmd)
+        gt_regs[#gt_regs + 1] = vim.fn.getreg('0')
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal! y2]] .. repeat_cmd)
+        gt_regs[#gt_regs + 1] = vim.fn.getreg('0')
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal! l]] .. repeat_cmd)
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal! y2]] .. repeat_cmd)
+        gt_regs[#gt_regs + 1] = vim.fn.getreg('0')
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal! h]] .. repeat_cmd)
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal! y2]] .. repeat_cmd)
+        gt_regs[#gt_regs + 1] = vim.fn.getreg('0')
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal! 2y]] .. cmd .. spec.char)
+        gt_regs[#gt_regs + 1] = vim.fn.getreg('0')
+
+        -- test using tstextobj repeatable_move.lua (operator-pending mode)
+        vim.api.nvim_win_set_cursor(0, { spec.row, col })
+        local ts_regs = {}
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal y]] .. cmd .. spec.char)
+        ts_regs[#ts_regs + 1] = vim.fn.getreg('0')
+        assert.are.same(spec.row, vim.fn.line('.'), "Command shouldn't move cursor over rows")
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal y]] .. repeat_cmd)
+        ts_regs[#ts_regs + 1] = vim.fn.getreg('0')
+        assert.are.same(spec.row, vim.fn.line('.'), "Command shouldn't move cursor over rows")
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal y2]] .. repeat_cmd)
+        ts_regs[#ts_regs + 1] = vim.fn.getreg('0')
+        assert.are.same(spec.row, vim.fn.line('.'), "Command shouldn't move cursor over rows")
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal l]] .. repeat_cmd)
+        assert.are.same(spec.row, vim.fn.line('.'), "Command shouldn't move cursor over rows")
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal y2]] .. repeat_cmd)
+        ts_regs[#ts_regs + 1] = vim.fn.getreg('0')
+        assert.are.same(spec.row, vim.fn.line('.'), "Command shouldn't move cursor over rows")
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal h]] .. repeat_cmd)
+        assert.are.same(spec.row, vim.fn.line('.'), "Command shouldn't move cursor over rows")
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal y2]] .. repeat_cmd)
+        ts_regs[#ts_regs + 1] = vim.fn.getreg('0')
+        assert.are.same(spec.row, vim.fn.line('.'), "Command shouldn't move cursor over rows")
+        vim.fn.setreg('0', '')
+        vim.cmd([[normal 2y]] .. cmd .. spec.char)
+        ts_regs[#ts_regs + 1] = vim.fn.getreg('0')
+
+        assert.are.same(
+          gt_regs,
+          ts_regs,
+          string.format(
+            "Command %s with repeat %s works differently than vim's built-in find, col: %d",
+            cmd,
+            repeat_cmd,
+            col
+          )
         )
       end
     end