File size: 1,995 Bytes
7a1ad33 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | /**
* @license
* Copyright 2026 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import { describe, it, expect } from 'vitest';
import { ApprovalMode } from '../policy/types.js';
import {
getApprovalModeDescription,
getPlanModeExitMessage,
} from './approvalModeUtils.js';
describe('approvalModeUtils', () => {
describe('getApprovalModeDescription', () => {
it('should return correct description for DEFAULT mode', () => {
expect(getApprovalModeDescription(ApprovalMode.DEFAULT)).toBe(
'Default mode (edits will require confirmation)',
);
});
it('should return correct description for AUTO_EDIT mode', () => {
expect(getApprovalModeDescription(ApprovalMode.AUTO_EDIT)).toBe(
'Auto-Edit mode (edits will be applied automatically)',
);
});
it('should return correct description for PLAN mode', () => {
expect(getApprovalModeDescription(ApprovalMode.PLAN)).toBe(
'Plan mode (read-only planning)',
);
});
it('should return correct description for YOLO mode', () => {
expect(getApprovalModeDescription(ApprovalMode.YOLO)).toBe(
'YOLO mode (all tool calls auto-approved)',
);
});
});
describe('getPlanModeExitMessage', () => {
it('should return standard message when not manual', () => {
expect(getPlanModeExitMessage(ApprovalMode.DEFAULT, false)).toBe(
'Plan approved. Switching to Default mode (edits will require confirmation).',
);
});
it('should return manual message when manual is true', () => {
expect(getPlanModeExitMessage(ApprovalMode.AUTO_EDIT, true)).toBe(
'User has manually exited Plan Mode. Switching to Auto-Edit mode (edits will be applied automatically).',
);
});
it('should default to non-manual message', () => {
expect(getPlanModeExitMessage(ApprovalMode.YOLO)).toBe(
'Plan approved. Switching to YOLO mode (all tool calls auto-approved).',
);
});
});
});
|