@machina.ai/cell-cli 1.11.0-rc1 → 1.13.0-rc1

package/dist/src/ui/hooks/vim.test.js

@@ -4,7 +4,9 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { renderHook, act } from '@testing-library/react';
+import { act } from 'react';
+import { renderHook } from '../../test-utils/render.js';
+import { waitFor } from '../../test-utils/async.js';
 import { useVim } from './vim.js';
 import { textBufferReducer } from '../components/shared/text-buffer.js';
 // Mock the VimModeContext
@@ -1006,23 +1008,32 @@ describe('useVim hook', () => {
         });
     });
     describe('Shell command pass-through', () => {
-        it('should pass through ctrl+r in INSERT mode', () => {
+        it('should pass through ctrl+r in INSERT mode', async () => {
             mockVimContext.vimMode = 'INSERT';
             const { result } = renderVimHook();
+            await waitFor(() => {
+                expect(result.current.mode).toBe('INSERT');
+            });
             const handled = result.current.handleInput(createKey({ name: 'r', ctrl: true }));
             expect(handled).toBe(false);
         });
-        it('should pass through ! in INSERT mode when buffer is empty', () => {
+        it('should pass through ! in INSERT mode when buffer is empty', async () => {
             mockVimContext.vimMode = 'INSERT';
             const emptyBuffer = createMockBuffer('');
             const { result } = renderVimHook(emptyBuffer);
+            await waitFor(() => {
+                expect(result.current.mode).toBe('INSERT');
+            });
             const handled = result.current.handleInput(createKey({ sequence: '!' }));
             expect(handled).toBe(false);
         });
-        it('should handle ! as input in INSERT mode when buffer is not empty', () => {
+        it('should handle ! as input in INSERT mode when buffer is not empty', async () => {
             mockVimContext.vimMode = 'INSERT';
             const nonEmptyBuffer = createMockBuffer('not empty');
             const { result } = renderVimHook(nonEmptyBuffer);
+            await waitFor(() => {
+                expect(result.current.mode).toBe('INSERT');
+            });
             const key = createKey({ sequence: '!', name: '!' });
             act(() => {
                 result.current.handleInput(key);
@@ -1032,357 +1043,226 @@ describe('useVim hook', () => {
     });
     // Line operations (dd, cc) are tested in text-buffer.test.ts
     describe('Reducer-based integration tests', () => {
-        describe('de (delete word end)', () => {
-            it('should delete from cursor to end of current word', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['hello world test'],
-                    cursorRow: 0,
-                    cursorCol: 1, // cursor on 'e' in "hello"
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_delete_word_end',
-                    payload: { count: 1 },
-                });
-                // Should delete "ello" (from cursor to end of word), leaving "h world test"
-                expect(result.lines).toEqual(['h world test']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(1);
-            });
-            it('should delete multiple word ends with count', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['hello world test more'],
-                    cursorRow: 0,
-                    cursorCol: 1, // cursor on 'e' in "hello"
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_delete_word_end',
-                    payload: { count: 2 },
-                });
-                // Should delete "ello world" (to end of second word), leaving "h test more"
-                expect(result.lines).toEqual(['h test more']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(1);
-            });
-        });
-        describe('db (delete word backward)', () => {
-            it('should delete from cursor to start of previous word', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['hello world test'],
-                    cursorRow: 0,
-                    cursorCol: 11, // cursor on 't' in "test"
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_delete_word_backward',
-                    payload: { count: 1 },
-                });
-                // Should delete "world" (previous word only), leaving "hello  test"
-                expect(result.lines).toEqual(['hello  test']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(6);
-            });
-            it('should delete multiple words backward with count', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['hello world test more'],
-                    cursorRow: 0,
-                    cursorCol: 17, // cursor on 'm' in "more"
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_delete_word_backward',
-                    payload: { count: 2 },
-                });
-                // Should delete "world test " (two words backward), leaving "hello more"
-                expect(result.lines).toEqual(['hello more']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(6);
-            });
-        });
-        describe('cw (change word forward)', () => {
-            it('should delete from cursor to start of next word', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['hello world test'],
-                    cursorRow: 0,
-                    cursorCol: 0, // cursor on 'h' in "hello"
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_change_word_forward',
-                    payload: { count: 1 },
-                });
-                // Should delete "hello " (word + space), leaving "world test"
-                expect(result.lines).toEqual(['world test']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(0);
-            });
-            it('should change multiple words with count', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['hello world test more'],
-                    cursorRow: 0,
-                    cursorCol: 0,
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_change_word_forward',
-                    payload: { count: 2 },
-                });
-                // Should delete "hello world " (two words), leaving "test more"
-                expect(result.lines).toEqual(['test more']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(0);
-            });
-        });
-        describe('ce (change word end)', () => {
-            it('should change from cursor to end of current word', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['hello world test'],
-                    cursorRow: 0,
-                    cursorCol: 1, // cursor on 'e' in "hello"
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_change_word_end',
-                    payload: { count: 1 },
-                });
-                // Should delete "ello" (from cursor to end of word), leaving "h world test"
-                expect(result.lines).toEqual(['h world test']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(1);
-            });
-            it('should change multiple word ends with count', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['hello world test'],
-                    cursorRow: 0,
-                    cursorCol: 1, // cursor on 'e' in "hello"
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_change_word_end',
-                    payload: { count: 2 },
-                });
-                // Should delete "ello world" (to end of second word), leaving "h test"
-                expect(result.lines).toEqual(['h test']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(1);
-            });
-        });
-        describe('cb (change word backward)', () => {
-            it('should change from cursor to start of previous word', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['hello world test'],
-                    cursorRow: 0,
-                    cursorCol: 11, // cursor on 't' in "test"
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_change_word_backward',
-                    payload: { count: 1 },
-                });
-                // Should delete "world" (previous word only), leaving "hello  test"
-                expect(result.lines).toEqual(['hello  test']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(6);
-            });
-        });
-        describe('cc (change line)', () => {
-            it('should clear the line and place cursor at the start', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['  hello world'],
-                    cursorRow: 0,
-                    cursorCol: 5, // cursor on 'o'
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_change_line',
-                    payload: { count: 1 },
-                });
-                expect(result.lines).toEqual(['']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(0);
-            });
-        });
-        describe('dd (delete line)', () => {
-            it('should delete the current line', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['line1', 'line2', 'line3'],
-                    cursorRow: 1,
-                    cursorCol: 2,
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_delete_line',
-                    payload: { count: 1 },
-                });
-                expect(result.lines).toEqual(['line1', 'line3']);
-                expect(result.cursorRow).toBe(1);
-                expect(result.cursorCol).toBe(0);
-            });
-            it('should delete multiple lines with count', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['line1', 'line2', 'line3', 'line4'],
-                    cursorRow: 1,
-                    cursorCol: 2,
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_delete_line',
-                    payload: { count: 2 },
-                });
-                // Should delete lines 1 and 2
-                expect(result.lines).toEqual(['line1', 'line4']);
-                expect(result.cursorRow).toBe(1);
-                expect(result.cursorCol).toBe(0);
-            });
-            it('should handle deleting last line', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['only line'],
-                    cursorRow: 0,
-                    cursorCol: 3,
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_delete_line',
-                    payload: { count: 1 },
-                });
-                // Should leave an empty line when deleting the only line
-                expect(result.lines).toEqual(['']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(0);
-            });
-        });
-        describe('D (delete to end of line)', () => {
-            it('should delete from cursor to end of line', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['hello world test'],
-                    cursorRow: 0,
-                    cursorCol: 6, // cursor on 'w' in "world"
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_delete_to_end_of_line',
-                });
-                // Should delete "world test", leaving "hello "
-                expect(result.lines).toEqual(['hello ']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(6);
-            });
-            it('should handle D at end of line', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['hello world'],
-                    cursorRow: 0,
-                    cursorCol: 11, // cursor at end
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_delete_to_end_of_line',
-                });
-                // Should not change anything when at end of line
-                expect(result.lines).toEqual(['hello world']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(11);
-            });
-        });
-        describe('C (change to end of line)', () => {
-            it('should change from cursor to end of line', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['hello world test'],
-                    cursorRow: 0,
-                    cursorCol: 6, // cursor on 'w' in "world"
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_change_to_end_of_line',
-                });
-                // Should delete "world test", leaving "hello "
-                expect(result.lines).toEqual(['hello ']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(6);
-            });
-            it('should handle C at beginning of line', () => {
-                const initialState = createMockTextBufferState({
-                    lines: ['hello world'],
-                    cursorRow: 0,
-                    cursorCol: 0,
-                    preferredCol: null,
-                    undoStack: [],
-                    redoStack: [],
-                    clipboard: null,
-                    selectionAnchor: null,
-                });
-                const result = textBufferReducer(initialState, {
-                    type: 'vim_change_to_end_of_line',
-                });
-                // Should delete entire line content
-                expect(result.lines).toEqual(['']);
-                expect(result.cursorRow).toBe(0);
-                expect(result.cursorCol).toBe(0);
-            });
+        const testCases = [
+            {
+                command: 'de',
+                desc: 'delete from cursor to end of current word',
+                lines: ['hello world test'],
+                cursorRow: 0,
+                cursorCol: 1,
+                actionType: 'vim_delete_word_end',
+                count: 1,
+                expectedLines: ['h world test'],
+                expectedCursorRow: 0,
+                expectedCursorCol: 1,
+            },
+            {
+                command: 'de',
+                desc: 'delete multiple word ends with count',
+                lines: ['hello world test more'],
+                cursorRow: 0,
+                cursorCol: 1,
+                actionType: 'vim_delete_word_end',
+                count: 2,
+                expectedLines: ['h test more'],
+                expectedCursorRow: 0,
+                expectedCursorCol: 1,
+            },
+            {
+                command: 'db',
+                desc: 'delete from cursor to start of previous word',
+                lines: ['hello world test'],
+                cursorRow: 0,
+                cursorCol: 11,
+                actionType: 'vim_delete_word_backward',
+                count: 1,
+                expectedLines: ['hello  test'],
+                expectedCursorRow: 0,
+                expectedCursorCol: 6,
+            },
+            {
+                command: 'db',
+                desc: 'delete multiple words backward with count',
+                lines: ['hello world test more'],
+                cursorRow: 0,
+                cursorCol: 17,
+                actionType: 'vim_delete_word_backward',
+                count: 2,
+                expectedLines: ['hello more'],
+                expectedCursorRow: 0,
+                expectedCursorCol: 6,
+            },
+            {
+                command: 'cw',
+                desc: 'delete from cursor to start of next word',
+                lines: ['hello world test'],
+                cursorRow: 0,
+                cursorCol: 0,
+                actionType: 'vim_change_word_forward',
+                count: 1,
+                expectedLines: ['world test'],
+                expectedCursorRow: 0,
+                expectedCursorCol: 0,
+            },
+            {
+                command: 'cw',
+                desc: 'change multiple words with count',
+                lines: ['hello world test more'],
+                cursorRow: 0,
+                cursorCol: 0,
+                actionType: 'vim_change_word_forward',
+                count: 2,
+                expectedLines: ['test more'],
+                expectedCursorRow: 0,
+                expectedCursorCol: 0,
+            },
+            {
+                command: 'ce',
+                desc: 'change from cursor to end of current word',
+                lines: ['hello world test'],
+                cursorRow: 0,
+                cursorCol: 1,
+                actionType: 'vim_change_word_end',
+                count: 1,
+                expectedLines: ['h world test'],
+                expectedCursorRow: 0,
+                expectedCursorCol: 1,
+            },
+            {
+                command: 'ce',
+                desc: 'change multiple word ends with count',
+                lines: ['hello world test'],
+                cursorRow: 0,
+                cursorCol: 1,
+                actionType: 'vim_change_word_end',
+                count: 2,
+                expectedLines: ['h test'],
+                expectedCursorRow: 0,
+                expectedCursorCol: 1,
+            },
+            {
+                command: 'cb',
+                desc: 'change from cursor to start of previous word',
+                lines: ['hello world test'],
+                cursorRow: 0,
+                cursorCol: 11,
+                actionType: 'vim_change_word_backward',
+                count: 1,
+                expectedLines: ['hello  test'],
+                expectedCursorRow: 0,
+                expectedCursorCol: 6,
+            },
+            {
+                command: 'cc',
+                desc: 'clear the line and place cursor at the start',
+                lines: ['  hello world'],
+                cursorRow: 0,
+                cursorCol: 5,
+                actionType: 'vim_change_line',
+                count: 1,
+                expectedLines: [''],
+                expectedCursorRow: 0,
+                expectedCursorCol: 0,
+            },
+            {
+                command: 'dd',
+                desc: 'delete the current line',
+                lines: ['line1', 'line2', 'line3'],
+                cursorRow: 1,
+                cursorCol: 2,
+                actionType: 'vim_delete_line',
+                count: 1,
+                expectedLines: ['line1', 'line3'],
+                expectedCursorRow: 1,
+                expectedCursorCol: 0,
+            },
+            {
+                command: 'dd',
+                desc: 'delete multiple lines with count',
+                lines: ['line1', 'line2', 'line3', 'line4'],
+                cursorRow: 1,
+                cursorCol: 2,
+                actionType: 'vim_delete_line',
+                count: 2,
+                expectedLines: ['line1', 'line4'],
+                expectedCursorRow: 1,
+                expectedCursorCol: 0,
+            },
+            {
+                command: 'dd',
+                desc: 'handle deleting last line',
+                lines: ['only line'],
+                cursorRow: 0,
+                cursorCol: 3,
+                actionType: 'vim_delete_line',
+                count: 1,
+                expectedLines: [''],
+                expectedCursorRow: 0,
+                expectedCursorCol: 0,
+            },
+            {
+                command: 'D',
+                desc: 'delete from cursor to end of line',
+                lines: ['hello world test'],
+                cursorRow: 0,
+                cursorCol: 6,
+                actionType: 'vim_delete_to_end_of_line',
+                expectedLines: ['hello '],
+                expectedCursorRow: 0,
+                expectedCursorCol: 6,
+            },
+            {
+                command: 'D',
+                desc: 'handle D at end of line',
+                lines: ['hello world'],
+                cursorRow: 0,
+                cursorCol: 11,
+                actionType: 'vim_delete_to_end_of_line',
+                expectedLines: ['hello world'],
+                expectedCursorRow: 0,
+                expectedCursorCol: 11,
+            },
+            {
+                command: 'C',
+                desc: 'change from cursor to end of line',
+                lines: ['hello world test'],
+                cursorRow: 0,
+                cursorCol: 6,
+                actionType: 'vim_change_to_end_of_line',
+                expectedLines: ['hello '],
+                expectedCursorRow: 0,
+                expectedCursorCol: 6,
+            },
+            {
+                command: 'C',
+                desc: 'handle C at beginning of line',
+                lines: ['hello world'],
+                cursorRow: 0,
+                cursorCol: 0,
+                actionType: 'vim_change_to_end_of_line',
+                expectedLines: [''],
+                expectedCursorRow: 0,
+                expectedCursorCol: 0,
+            },
+        ];
+        it.each(testCases)('$command: should $desc', ({ lines, cursorRow, cursorCol, actionType, count, expectedLines, expectedCursorRow, expectedCursorCol, }) => {
+            const initialState = createMockTextBufferState({
+                lines,
+                cursorRow,
+                cursorCol,
+                preferredCol: null,
+                undoStack: [],
+                redoStack: [],
+                clipboard: null,
+                selectionAnchor: null,
+            });
+            const action = (count
+                ? { type: actionType, payload: { count } }
+                : { type: actionType });
+            const result = textBufferReducer(initialState, action);
+            expect(result.lines).toEqual(expectedLines);
+            expect(result.cursorRow).toBe(expectedCursorRow);
+            expect(result.cursorCol).toBe(expectedCursorCol);
         });
     });
 });