38 lines
2.2 KiB
TypeScript
38 lines
2.2 KiB
TypeScript
import type { IncomingMessage, ServerResponse } from 'node:http';
|
|
import { Readable } from 'node:stream';
|
|
import { beforeEach, expect, it, vi } from 'vitest';
|
|
import { handleCloudAgentsRoutes } from '@electron/api/routes/cloud-agents';
|
|
|
|
const service = vi.hoisted(() => ({ useAvatarPreset: vi.fn(), uploadAvatar: vi.fn() }));
|
|
vi.mock('@electron/services/cloud-agents', () => ({
|
|
CloudAgentsModule: class { useAvatarPreset = service.useAvatarPreset; uploadAvatar = service.uploadAvatar; },
|
|
CloudAgentsError: class extends Error { constructor(readonly status: number, readonly code: string) { super(code); } },
|
|
}));
|
|
|
|
beforeEach(() => vi.resetAllMocks());
|
|
|
|
async function call(path: string, body?: string, method = 'POST') {
|
|
const request = Object.assign(Readable.from(body ? [body] : []), { method, headers: {} }) as IncomingMessage;
|
|
let output = '';
|
|
const response = { setHeader: vi.fn(), statusCode: 0, end: (text: string) => { output = text; } } as unknown as ServerResponse;
|
|
const handled = await handleCloudAgentsRoutes(request, response, new URL(`http://localhost/api/cloud-agents${path}`));
|
|
return { handled, status: response.statusCode, body: JSON.parse(output) };
|
|
}
|
|
|
|
it('routes only the preset id to Main and returns the canonical icon projection', async () => {
|
|
const avatar = { icon: '/api/storage/public/images/account/avatar.png', icon_url: 'https://agents.test/api/storage/public/images/account/avatar.png' };
|
|
service.useAvatarPreset.mockResolvedValue(avatar);
|
|
expect(await call('/avatars/preset', JSON.stringify({ id: 'avatar-03' }))).toEqual({ handled: true, status: 200, body: avatar });
|
|
expect(service.useAvatarPreset).toHaveBeenCalledExactlyOnceWith({ id: 'avatar-03' });
|
|
expect(service.uploadAvatar).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('keeps the existing file picker route and rejects malformed preset JSON', async () => {
|
|
service.uploadAvatar.mockResolvedValue(null);
|
|
expect(await call('/avatars/pick')).toEqual({ handled: true, status: 200, body: null });
|
|
expect(service.uploadAvatar).toHaveBeenCalledOnce();
|
|
expect((await call('/avatars/preset', '{')).status).toBe(400);
|
|
expect(service.useAvatarPreset).not.toHaveBeenCalled();
|
|
expect((await call('/avatars/preset', undefined, 'GET')).status).toBe(404);
|
|
});
|