mirror of
https://github.com/mauriceboe/TREK.git
synced 2026-11-13 07:40:59 +00:00
Maurice
2d398568da
Every page of an install with the assistant and AI parsing off asked /api/ai/status and logged the 404 the addon guard answers. The store asks only while an addon opens the route and the install is not managed, and one hook on the desktop window and both phone screens asks again when the assistant or AI parsing is switched.
78 lines
3.2 KiB
TypeScript
78 lines
3.2 KiB
TypeScript
/**
|
|
* FE-AI-STATUS-001 to FE-AI-STATUS-003
|
|
*
|
|
* The views of the assistant ask for its status through this hook: not while
|
|
* both addons that open /api/ai are off, where the server answers 404, and
|
|
* as soon as one comes on, which the addon list can say after the view
|
|
* mounted (the first load, an admin's switch). The assistant's own addon
|
|
* switching asks again while AI parsing keeps /api/ai open.
|
|
*/
|
|
import { act, renderHook, waitFor } from '@testing-library/react';
|
|
import type { AiStatusResponse } from '@trek/shared';
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
|
|
|
|
import { resetAllStores, seedStore } from '../../../tests/helpers/store';
|
|
import { aiConnectionsRepo } from '../../repo/aiConnectionsRepo';
|
|
import { useAddonStore } from '../../store/addonStore';
|
|
import { useAssistantStore } from '../../store/assistantStore';
|
|
import { useAssistantStatus } from './useAssistantStatus';
|
|
|
|
const INITIAL = useAssistantStore.getState();
|
|
const STATUS = { available: true, retention: { mode: 'days', days: 90 } } as AiStatusResponse;
|
|
|
|
/** The addon list as it lands from the server, with these addons on. */
|
|
const addonsOn = (...ids: string[]) =>
|
|
seedStore(useAddonStore, { loaded: true, addons: ids.map((id) => ({ id, enabled: true })) });
|
|
|
|
beforeEach(() => {
|
|
resetAllStores();
|
|
});
|
|
|
|
afterEach(() => {
|
|
useAssistantStore.setState(INITIAL, true);
|
|
vi.restoreAllMocks();
|
|
});
|
|
|
|
describe('useAssistantStatus', () => {
|
|
it('FE-AI-STATUS-001: asks nothing before the addon list or with both addons off, and asks once one comes on', async () => {
|
|
const status = vi.spyOn(aiConnectionsRepo, 'status').mockResolvedValue(STATUS);
|
|
renderHook(() => useAssistantStatus());
|
|
act(() => addonsOn('mcp'));
|
|
expect(status).not.toHaveBeenCalled();
|
|
expect(useAssistantStore.getState().status).toBeNull();
|
|
|
|
act(() => addonsOn('ai_assistant'));
|
|
await waitFor(() => expect(useAssistantStore.getState().status).toEqual(STATUS));
|
|
expect(status).toHaveBeenCalledTimes(1);
|
|
|
|
// The list read again with the assistant still on asks nothing more.
|
|
act(() => addonsOn('ai_assistant', 'mcp'));
|
|
expect(status).toHaveBeenCalledTimes(1);
|
|
});
|
|
|
|
it('FE-AI-STATUS-002: with AI parsing on it asks at once, and the last addon going off hides the assistant', async () => {
|
|
const status = vi.spyOn(aiConnectionsRepo, 'status').mockResolvedValue(STATUS);
|
|
addonsOn('llm_parsing');
|
|
renderHook(() => useAssistantStatus());
|
|
await waitFor(() => expect(useAssistantStore.getState().status).toEqual(STATUS));
|
|
expect(status).toHaveBeenCalledTimes(1);
|
|
|
|
act(() => addonsOn());
|
|
expect(useAssistantStore.getState().status).toBeNull();
|
|
expect(status).toHaveBeenCalledTimes(1);
|
|
});
|
|
|
|
it('FE-AI-STATUS-003: with AI parsing on, the assistant switched on or off asks again', async () => {
|
|
const status = vi.spyOn(aiConnectionsRepo, 'status').mockResolvedValue(STATUS);
|
|
addonsOn('llm_parsing');
|
|
renderHook(() => useAssistantStatus());
|
|
await waitFor(() => expect(status).toHaveBeenCalledTimes(1));
|
|
|
|
act(() => addonsOn('llm_parsing', 'ai_assistant'));
|
|
await waitFor(() => expect(status).toHaveBeenCalledTimes(2));
|
|
|
|
act(() => addonsOn('llm_parsing'));
|
|
await waitFor(() => expect(status).toHaveBeenCalledTimes(3));
|
|
});
|
|
});
|