Improve code coverage for cli package (#13724)

packages/cli/src/ui/utils/terminalSetup.test.ts

@@ -0,0 +1,161 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { terminalSetup, VSCODE_SHIFT_ENTER_SEQUENCE } from './terminalSetup.js';
+
+// Mock dependencies
+const mocks = vi.hoisted(() => ({
+  exec: vi.fn(),
+  mkdir: vi.fn(),
+  readFile: vi.fn(),
+  writeFile: vi.fn(),
+  copyFile: vi.fn(),
+  homedir: vi.fn(),
+  platform: vi.fn(),
+}));
+
+vi.mock('node:child_process', () => ({
+  exec: mocks.exec,
+  execFile: vi.fn(),
+}));
+
+vi.mock('node:fs', () => ({
+  promises: {
+    mkdir: mocks.mkdir,
+    readFile: mocks.readFile,
+    writeFile: mocks.writeFile,
+    copyFile: mocks.copyFile,
+  },
+}));
+
+vi.mock('node:os', () => ({
+  homedir: mocks.homedir,
+  platform: mocks.platform,
+}));
+
+vi.mock('./kittyProtocolDetector.js', () => ({
+  isKittyProtocolEnabled: vi.fn().mockReturnValue(false),
+}));
+
+describe('terminalSetup', () => {
+  const originalEnv = process.env;
+
+  beforeEach(() => {
+    vi.resetAllMocks();
+    process.env = { ...originalEnv };
+
+    // Default mocks
+    mocks.homedir.mockReturnValue('/home/user');
+    mocks.platform.mockReturnValue('darwin');
+    mocks.mkdir.mockResolvedValue(undefined);
+    mocks.copyFile.mockResolvedValue(undefined);
+    mocks.exec.mockImplementation((cmd, cb) => cb(null, { stdout: '' }));
+  });
+
+  afterEach(() => {
+    process.env = originalEnv;
+  });
+
+  describe('detectTerminal', () => {
+    it('should detect VS Code from env var', async () => {
+      process.env['TERM_PROGRAM'] = 'vscode';
+      const result = await terminalSetup();
+      expect(result.message).toContain('VS Code');
+    });
+
+    it('should detect Cursor from env var', async () => {
+      process.env['CURSOR_TRACE_ID'] = 'some-id';
+      const result = await terminalSetup();
+      expect(result.message).toContain('Cursor');
+    });
+
+    it('should detect Windsurf from env var', async () => {
+      process.env['VSCODE_GIT_ASKPASS_MAIN'] = '/path/to/windsurf/askpass';
+      const result = await terminalSetup();
+      expect(result.message).toContain('Windsurf');
+    });
+
+    it('should detect from parent process', async () => {
+      mocks.platform.mockReturnValue('linux');
+      mocks.exec.mockImplementation((cmd, cb) => {
+        cb(null, { stdout: 'code\n' });
+      });
+
+      const result = await terminalSetup();
+      expect(result.message).toContain('VS Code');
+    });
+  });
+
+  describe('configureVSCodeStyle', () => {
+    it('should create new keybindings file if none exists', async () => {
+      process.env['TERM_PROGRAM'] = 'vscode';
+      mocks.readFile.mockRejectedValue(new Error('ENOENT'));
+
+      const result = await terminalSetup();
+
+      expect(result.success).toBe(true);
+      expect(mocks.writeFile).toHaveBeenCalled();
+
+      const writtenContent = JSON.parse(mocks.writeFile.mock.calls[0][1]);
+      expect(writtenContent).toMatchSnapshot();
+    });
+
+    it('should append to existing keybindings', async () => {
+      process.env['TERM_PROGRAM'] = 'vscode';
+      mocks.readFile.mockResolvedValue('[]');
+
+      const result = await terminalSetup();
+
+      expect(result.success).toBe(true);
+      const writtenContent = JSON.parse(mocks.writeFile.mock.calls[0][1]);
+      expect(writtenContent).toHaveLength(2); // Shift+Enter and Ctrl+Enter
+    });
+
+    it('should not modify if bindings already exist', async () => {
+      process.env['TERM_PROGRAM'] = 'vscode';
+      const existingBindings = [
+        {
+          key: 'shift+enter',
+          command: 'workbench.action.terminal.sendSequence',
+          args: { text: VSCODE_SHIFT_ENTER_SEQUENCE },
+        },
+        {
+          key: 'ctrl+enter',
+          command: 'workbench.action.terminal.sendSequence',
+          args: { text: VSCODE_SHIFT_ENTER_SEQUENCE },
+        },
+      ];
+      mocks.readFile.mockResolvedValue(JSON.stringify(existingBindings));
+
+      const result = await terminalSetup();
+
+      expect(result.success).toBe(true);
+      expect(mocks.writeFile).not.toHaveBeenCalled();
+    });
+
+    it('should fail gracefully if json is invalid', async () => {
+      process.env['TERM_PROGRAM'] = 'vscode';
+      mocks.readFile.mockResolvedValue('{ invalid json');
+
+      const result = await terminalSetup();
+
+      expect(result.success).toBe(false);
+      expect(result.message).toContain('invalid JSON');
+    });
+
+    it('should handle comments in JSON', async () => {
+      process.env['TERM_PROGRAM'] = 'vscode';
+      const jsonWithComments = '// This is a comment\n[]';
+      mocks.readFile.mockResolvedValue(jsonWithComments);
+
+      const result = await terminalSetup();
+
+      expect(result.success).toBe(true);
+      expect(mocks.writeFile).toHaveBeenCalled();
+    });
+  });
+});