packages/core/src/utils/approvalModeUtils.test.ts

/**
 * @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).',
      );
    });
  });
});