mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-04-20 10:10:56 -07:00
116 lines
3.1 KiB
TypeScript
116 lines
3.1 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2026 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import { vi, describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
import { format } from 'node:util';
|
|
import { handleDisable, disableCommand } from './disable.js';
|
|
import {
|
|
loadSettings,
|
|
SettingScope,
|
|
type LoadedSettings,
|
|
type LoadableSettingScope,
|
|
} from '../../config/settings.js';
|
|
|
|
const emitConsoleLog = vi.hoisted(() => vi.fn());
|
|
const debugLogger = vi.hoisted(() => ({
|
|
log: vi.fn((message, ...args) => {
|
|
emitConsoleLog('log', format(message, ...args));
|
|
}),
|
|
}));
|
|
|
|
vi.mock('@google/gemini-cli-core', async (importOriginal) => {
|
|
const actual =
|
|
await importOriginal<typeof import('@google/gemini-cli-core')>();
|
|
return {
|
|
...actual,
|
|
debugLogger,
|
|
};
|
|
});
|
|
|
|
vi.mock('../../config/settings.js', async (importOriginal) => {
|
|
const actual =
|
|
await importOriginal<typeof import('../../config/settings.js')>();
|
|
return {
|
|
...actual,
|
|
loadSettings: vi.fn(),
|
|
isLoadableSettingScope: vi.fn((s) => s === 'User' || s === 'Workspace'),
|
|
};
|
|
});
|
|
|
|
vi.mock('../utils.js', () => ({
|
|
exitCli: vi.fn(),
|
|
}));
|
|
|
|
describe('skills disable command', () => {
|
|
const mockLoadSettings = vi.mocked(loadSettings);
|
|
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
});
|
|
|
|
afterEach(() => {
|
|
vi.restoreAllMocks();
|
|
});
|
|
|
|
describe('handleDisable', () => {
|
|
it('should disable an enabled skill in user scope', async () => {
|
|
const mockSettings = {
|
|
forScope: vi.fn().mockReturnValue({
|
|
settings: { skills: { disabled: [] } },
|
|
path: '/user/settings.json',
|
|
}),
|
|
setValue: vi.fn(),
|
|
};
|
|
mockLoadSettings.mockReturnValue(
|
|
mockSettings as unknown as LoadedSettings,
|
|
);
|
|
|
|
await handleDisable({
|
|
name: 'skill1',
|
|
scope: SettingScope.User as LoadableSettingScope,
|
|
});
|
|
|
|
expect(mockSettings.setValue).toHaveBeenCalledWith(
|
|
SettingScope.User,
|
|
'skills.disabled',
|
|
['skill1'],
|
|
);
|
|
expect(emitConsoleLog).toHaveBeenCalledWith(
|
|
'log',
|
|
'Skill "skill1" disabled by adding it to the disabled list in user (/user/settings.json) settings.',
|
|
);
|
|
});
|
|
|
|
it('should log a message if the skill is already disabled', async () => {
|
|
const mockSettings = {
|
|
forScope: vi.fn().mockReturnValue({
|
|
settings: { skills: { disabled: ['skill1'] } },
|
|
path: '/user/settings.json',
|
|
}),
|
|
setValue: vi.fn(),
|
|
};
|
|
vi.mocked(loadSettings).mockReturnValue(
|
|
mockSettings as unknown as LoadedSettings,
|
|
);
|
|
|
|
await handleDisable({ name: 'skill1', scope: SettingScope.User });
|
|
|
|
expect(mockSettings.setValue).not.toHaveBeenCalled();
|
|
expect(emitConsoleLog).toHaveBeenCalledWith(
|
|
'log',
|
|
'Skill "skill1" is already disabled.',
|
|
);
|
|
});
|
|
});
|
|
|
|
describe('disableCommand', () => {
|
|
it('should have correct command and describe', () => {
|
|
expect(disableCommand.command).toBe('disable <name> [--scope]');
|
|
expect(disableCommand.describe).toBe('Disables an agent skill.');
|
|
});
|
|
});
|
|
});
|