96 lines
3.6 KiB
TypeScript
96 lines
3.6 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2025 AionUi (aionui.com)
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
import { describe, it, expect, vi } from 'vitest';
|
|
import {
|
|
shouldShowNotification,
|
|
createBrowserNotificationController,
|
|
type NotificationGate,
|
|
} from '@/renderer/hooks/system/notification/browserNotificationCore';
|
|
|
|
const openGate: NotificationGate = {
|
|
isElectron: false,
|
|
hasNotificationApi: true,
|
|
isSecureContext: true,
|
|
permission: 'granted',
|
|
settingEnabled: true,
|
|
documentHidden: true,
|
|
};
|
|
|
|
describe('shouldShowNotification', () => {
|
|
it('returns true when all gates pass', () => {
|
|
expect(shouldShowNotification(openGate)).toBe(true);
|
|
});
|
|
|
|
it.each([
|
|
['isElectron', { isElectron: true }],
|
|
['no api', { hasNotificationApi: false }],
|
|
['insecure', { isSecureContext: false }],
|
|
['not granted', { permission: 'default' as const }],
|
|
['setting off', { settingEnabled: false }],
|
|
['tab visible', { documentHidden: false }],
|
|
])('returns false when %s', (_label, override) => {
|
|
expect(shouldShowNotification({ ...openGate, ...override })).toBe(false);
|
|
});
|
|
});
|
|
|
|
describe('createBrowserNotificationController.onStreamMessage', () => {
|
|
const makeDeps = (gate: NotificationGate = openGate) => {
|
|
const show = vi.fn();
|
|
const controller = createBrowserNotificationController({
|
|
readGate: () => gate,
|
|
show,
|
|
bodyFor: (kind) => kind,
|
|
});
|
|
return { show, controller };
|
|
};
|
|
|
|
it('shows a turn-completed notification on a finish stream message', () => {
|
|
const { show, controller } = makeDeps();
|
|
controller.onStreamMessage({ type: 'finish', conversation_id: 'c1', turn_id: 't1' });
|
|
expect(show).toHaveBeenCalledWith({ body: 'turnCompleted', conversationId: 'c1' });
|
|
});
|
|
|
|
it('shows a confirmation notification on an acp_permission stream message', () => {
|
|
const { show, controller } = makeDeps();
|
|
controller.onStreamMessage({ type: 'acp_permission', conversation_id: 'c2' });
|
|
expect(show).toHaveBeenCalledWith({ body: 'confirmation', conversationId: 'c2' });
|
|
});
|
|
|
|
it('shows a confirmation notification on a permission stream message (aionrs)', () => {
|
|
const { show, controller } = makeDeps();
|
|
controller.onStreamMessage({ type: 'permission', conversation_id: 'c3' });
|
|
expect(show).toHaveBeenCalledWith({ body: 'confirmation', conversationId: 'c3' });
|
|
});
|
|
|
|
it('ignores non-terminal stream message types', () => {
|
|
const { show, controller } = makeDeps();
|
|
controller.onStreamMessage({ type: 'thinking', conversation_id: 'c1', turn_id: 't1' });
|
|
controller.onStreamMessage({ type: 'text', conversation_id: 'c1', turn_id: 't1' });
|
|
controller.onStreamMessage({ type: 'start', conversation_id: 'c1', turn_id: 't1' });
|
|
expect(show).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('does not show when the gate is closed', () => {
|
|
const { show, controller } = makeDeps({ ...openGate, documentHidden: false });
|
|
controller.onStreamMessage({ type: 'finish', conversation_id: 'c1', turn_id: 't1' });
|
|
controller.onStreamMessage({ type: 'acp_permission', conversation_id: 'c1' });
|
|
expect(show).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('dedups repeated finish for the same turn_id', () => {
|
|
const { show, controller } = makeDeps();
|
|
controller.onStreamMessage({ type: 'finish', conversation_id: 'c1', turn_id: 't1' });
|
|
controller.onStreamMessage({ type: 'finish', conversation_id: 'c1', turn_id: 't1' });
|
|
expect(show).toHaveBeenCalledTimes(1);
|
|
});
|
|
|
|
it('ignores messages without a type', () => {
|
|
const { show, controller } = makeDeps();
|
|
controller.onStreamMessage({ conversation_id: 'c1' });
|
|
expect(show).not.toHaveBeenCalled();
|
|
});
|
|
});
|