mirror of
https://github.com/google-gemini/gemini-cli
synced 2026-04-21 13:37:17 +00:00
120 lines
3.5 KiB
TypeScript
120 lines
3.5 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2026 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
import { setupWorktree } from './worktreeSetup.js';
|
|
import {
|
|
getProjectRootForWorktree,
|
|
createWorktreeService,
|
|
writeToStderr,
|
|
} from '@google/gemini-cli-core';
|
|
|
|
// Mock dependencies
|
|
vi.mock('@google/gemini-cli-core', async (importOriginal) => {
|
|
const actual =
|
|
await importOriginal<typeof import('@google/gemini-cli-core')>();
|
|
return {
|
|
...actual,
|
|
getProjectRootForWorktree: vi.fn(),
|
|
createWorktreeService: vi.fn(),
|
|
debugLogger: {
|
|
log: vi.fn(),
|
|
error: vi.fn(),
|
|
debug: vi.fn(),
|
|
},
|
|
writeToStdout: vi.fn(),
|
|
writeToStderr: vi.fn(),
|
|
};
|
|
});
|
|
|
|
describe('setupWorktree', () => {
|
|
const originalEnv = { ...process.env };
|
|
const originalCwd = process.cwd;
|
|
|
|
const mockService = {
|
|
setup: vi.fn(),
|
|
maybeCleanup: vi.fn(),
|
|
};
|
|
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
process.env = { ...originalEnv };
|
|
|
|
// Mock process.cwd and process.chdir
|
|
let currentPath = '/mock/project';
|
|
process.cwd = vi.fn().mockImplementation(() => currentPath);
|
|
process.chdir = vi.fn().mockImplementation((newPath) => {
|
|
currentPath = newPath;
|
|
});
|
|
|
|
// Mock successful execution of core utilities
|
|
vi.mocked(getProjectRootForWorktree).mockResolvedValue('/mock/project');
|
|
vi.mocked(createWorktreeService).mockResolvedValue(mockService as never);
|
|
mockService.setup.mockResolvedValue({
|
|
name: 'my-feature',
|
|
path: '/mock/project/.gemini/worktrees/my-feature',
|
|
baseSha: 'base-sha',
|
|
});
|
|
});
|
|
|
|
afterEach(() => {
|
|
process.env = { ...originalEnv };
|
|
process.cwd = originalCwd;
|
|
delete (process as { chdir?: typeof process.chdir }).chdir;
|
|
});
|
|
|
|
it('should create and switch to a new worktree', async () => {
|
|
await setupWorktree('my-feature');
|
|
|
|
expect(getProjectRootForWorktree).toHaveBeenCalledWith('/mock/project');
|
|
expect(createWorktreeService).toHaveBeenCalledWith('/mock/project');
|
|
expect(mockService.setup).toHaveBeenCalledWith('my-feature');
|
|
expect(process.chdir).toHaveBeenCalledWith(
|
|
'/mock/project/.gemini/worktrees/my-feature',
|
|
);
|
|
expect(process.env['GEMINI_CLI_WORKTREE_HANDLED']).toBe('1');
|
|
});
|
|
|
|
it('should generate a name if worktreeName is undefined', async () => {
|
|
mockService.setup.mockResolvedValue({
|
|
name: 'generated-name',
|
|
path: '/mock/project/.gemini/worktrees/generated-name',
|
|
baseSha: 'base-sha',
|
|
});
|
|
|
|
await setupWorktree(undefined);
|
|
|
|
expect(mockService.setup).toHaveBeenCalledWith(undefined);
|
|
});
|
|
|
|
it('should skip worktree creation if GEMINI_CLI_WORKTREE_HANDLED is set', async () => {
|
|
process.env['GEMINI_CLI_WORKTREE_HANDLED'] = '1';
|
|
|
|
await setupWorktree('my-feature');
|
|
|
|
expect(createWorktreeService).not.toHaveBeenCalled();
|
|
expect(process.chdir).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('should handle errors gracefully and exit', async () => {
|
|
const mockExit = vi.spyOn(process, 'exit').mockImplementation(() => {
|
|
throw new Error('PROCESS_EXIT');
|
|
});
|
|
|
|
mockService.setup.mockRejectedValue(new Error('Git failure'));
|
|
|
|
await expect(setupWorktree('my-feature')).rejects.toThrow('PROCESS_EXIT');
|
|
|
|
expect(writeToStderr).toHaveBeenCalledWith(
|
|
expect.stringContaining(
|
|
'Failed to create or switch to worktree: Git failure',
|
|
),
|
|
);
|
|
expect(mockExit).toHaveBeenCalledWith(1);
|
|
|
|
mockExit.mockRestore();
|
|
});
|
|
});
|