111 lines
3.7 KiB
TypeScript
111 lines
3.7 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2025 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import { vi, describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
import { getIdeInstaller, IdeInstaller } from './ide-installer.js';
|
|
import * as child_process from 'child_process';
|
|
import * as fs from 'fs';
|
|
import * as os from 'os';
|
|
import { DetectedIde } from './detect-ide.js';
|
|
|
|
vi.mock('child_process');
|
|
vi.mock('fs');
|
|
vi.mock('os');
|
|
|
|
describe('ide-installer', () => {
|
|
describe('getIdeInstaller', () => {
|
|
it('should return a VsCodeInstaller for "vscode"', () => {
|
|
const installer = getIdeInstaller(DetectedIde.VSCode);
|
|
expect(installer).not.toBeNull();
|
|
// A more specific check might be needed if we export the class
|
|
expect(installer).toBeInstanceOf(Object);
|
|
});
|
|
|
|
it('should return an OpenVSXInstaller for "vscodium"', () => {
|
|
const installer = getIdeInstaller(DetectedIde.VSCodium);
|
|
expect(installer).not.toBeNull();
|
|
expect(installer).toBeInstanceOf(Object);
|
|
});
|
|
|
|
it('should return a DefaultIDEInstaller for an unknown IDE', () => {
|
|
const installer = getIdeInstaller('unknown' as DetectedIde);
|
|
// Assuming DefaultIDEInstaller is the fallback
|
|
expect(installer).not.toBeNull();
|
|
expect(installer).toBeInstanceOf(Object);
|
|
});
|
|
});
|
|
|
|
describe('VsCodeInstaller', () => {
|
|
let installer: IdeInstaller;
|
|
|
|
beforeEach(() => {
|
|
// We get a new installer for each test to reset the find command logic
|
|
installer = getIdeInstaller(DetectedIde.VSCode)!;
|
|
vi.spyOn(child_process, 'execSync').mockImplementation(() => '');
|
|
vi.spyOn(fs, 'existsSync').mockReturnValue(false);
|
|
vi.spyOn(os, 'homedir').mockReturnValue('/home/user');
|
|
});
|
|
|
|
afterEach(() => {
|
|
vi.restoreAllMocks();
|
|
});
|
|
|
|
describe('install', () => {
|
|
it('should return a failure message if VS Code is not installed', async () => {
|
|
vi.spyOn(child_process, 'execSync').mockImplementation(() => {
|
|
throw new Error('Command not found');
|
|
});
|
|
vi.spyOn(fs, 'existsSync').mockReturnValue(false);
|
|
// Re-create the installer so it re-runs findVsCodeCommand
|
|
installer = getIdeInstaller(DetectedIde.VSCode)!;
|
|
const result = await installer.install();
|
|
expect(result.success).toBe(false);
|
|
expect(result.message).toContain('VS Code CLI not found');
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('OpenVSXInstaller', () => {
|
|
let installer: IdeInstaller;
|
|
|
|
beforeEach(() => {
|
|
installer = getIdeInstaller(DetectedIde.VSCodium)!;
|
|
});
|
|
|
|
afterEach(() => {
|
|
vi.restoreAllMocks();
|
|
});
|
|
|
|
describe('install', () => {
|
|
it('should call execSync with the correct command and return success', async () => {
|
|
const execSyncSpy = vi
|
|
.spyOn(child_process, 'execSync')
|
|
.mockImplementation(() => '');
|
|
const result = await installer.install();
|
|
expect(execSyncSpy).toHaveBeenCalledWith(
|
|
'npx ovsx get google.gemini-cli-vscode-ide-companion',
|
|
{ stdio: 'pipe' },
|
|
);
|
|
expect(result.success).toBe(true);
|
|
expect(result.message).toContain(
|
|
'VS Code companion extension was installed successfully from OpenVSX',
|
|
);
|
|
});
|
|
|
|
it('should return a failure message on failed installation', async () => {
|
|
vi.spyOn(child_process, 'execSync').mockImplementation(() => {
|
|
throw new Error('Command failed');
|
|
});
|
|
const result = await installer.install();
|
|
expect(result.success).toBe(false);
|
|
expect(result.message).toContain(
|
|
'Failed to install VS Code companion extension from OpenVSX',
|
|
);
|
|
});
|
|
});
|
|
});
|
|
});
|