File size: 2,962 Bytes
9332ccf
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
/**
 * @license
 * Copyright 2026 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */

import { describe, expect, it } from 'vitest';
import { CoreToolCallStatus } from '@google/gemini-cli-core';
import { getPendingAttentionNotification } from './pendingAttentionNotification.js';

describe('getPendingAttentionNotification', () => {
  it('returns tool confirmation notification for awaiting tool approvals', () => {
    const notification = getPendingAttentionNotification(
      [
        {
          type: 'tool_group',
          tools: [
            {
              callId: 'tool-1',
              status: CoreToolCallStatus.AwaitingApproval,
              description: 'Run command',
              confirmationDetails: {
                type: 'exec',
                title: 'Run shell command',
                command: 'ls',
                rootCommand: 'ls',
                rootCommands: ['ls'],
              },
            },
          ],
        } as never,
      ],
      null,
      null,
      null,
      false,
      false,
    );

    expect(notification?.key).toBe('tool_confirmation:tool-1');
    expect(notification?.event.type).toBe('attention');
  });

  it('returns ask-user notification for ask_user confirmations', () => {
    const notification = getPendingAttentionNotification(
      [
        {
          type: 'tool_group',
          tools: [
            {
              callId: 'ask-user-1',
              status: CoreToolCallStatus.AwaitingApproval,
              description: 'Ask user',
              confirmationDetails: {
                type: 'ask_user',
                questions: [
                  {
                    header: 'Need approval?',
                    question: 'Proceed?',
                    options: [],
                    id: 'q1',
                  },
                ],
              },
            },
          ],
        } as never,
      ],
      null,
      null,
      null,
      false,
      false,
    );

    expect(notification?.key).toBe('ask_user:ask-user-1');
    expect(notification?.event).toEqual({
      type: 'attention',
      heading: 'Answer requested by agent',
      detail: 'Need approval?',
    });
  });

  it('uses request content in command/auth keys', () => {
    const commandNotification = getPendingAttentionNotification(
      [],
      {
        prompt: 'Approve command?',
        onConfirm: () => {},
      },
      null,
      null,
      false,
      false,
    );

    const authNotification = getPendingAttentionNotification(
      [],
      null,
      {
        prompt: 'Authorize sign-in?',
        onConfirm: () => {},
      },
      null,
      false,
      false,
    );

    expect(commandNotification?.key).toContain('command_confirmation:');
    expect(commandNotification?.key).toContain('Approve command?');
    expect(authNotification?.key).toContain('auth_consent:');
    expect(authNotification?.key).toContain('Authorize sign-in?');
  });
});