/** * BaseAdapter 单元测试（v0.6.4 P3-2 错误分类单轨化后重写） * * 契约变更说明： * - mapError 已删除（生产路径死代码，与 engine.isRetryableError 双轨漂移）。 * 错误分类的唯一事实来源是 engine.isRetryableError —— 本文件改为验证 * "BaseAdapter 抛出的错误携带可判定字段"的形状契约： * throwHttpError → error.status；fetchWithTimeout 超时 → code='ETIMEDOUT' * + 'timed out' message（命中引擎网络超时分支）。 * - 新增：错误体长度截断、外部 abort 与自身超时的区分。 */ import { describe, it, expect, vi, afterEach } from 'vitest'; import { BaseAdapter, ContentFilterError } from '../base-adapter'; import { MetonaStreamEventType } from '../../types'; import type { IMetonaProviderAdapter, AdapterConfig, MetonaRequest, MetonaResponse, MetonaStreamEvent, } from '../../types'; /** 测试用具体 Adapter 实现（暴露 protected 方法供测试） */ class TestAdapter extends BaseAdapter { readonly providerId = 'test'; readonly supportedModels = ['test-model-a', 'test-model-b']; readonly supportsToolCalling = true; readonly supportsThinking = false; async send(_request: MetonaRequest): Promise { throw new Error('not implemented'); } async *sendStream(_request: MetonaRequest): AsyncIterable { // 空实现 } /** 测试辅助: 暴露 protected throwHttpError */ async throwHttpErrorPublic(response: Response, context: string) { return this.throwHttpError(response, context); } /** 测试辅助: 暴露 protected fetchWithTimeout */ async fetchWithTimeoutPublic(url: string, init: RequestInit, timeoutMs: number) { return this.fetchWithTimeout(url, init, timeoutMs); } } function makeAdapter(config: Partial = {}): TestAdapter { return new TestAdapter({ provider: 'test', baseURL: 'https://api.test.com', apiKey: 'sk-test', defaultModel: 'test-model-a', ...config, }); } // ===== 错误形状契约（engine.isRetryableError 的输入保证） ===== describe('BaseAdapter — 抛出错误的可判定形状（单轨化契约）', () => { /** 与 engine.isRetryableError 相同的判定逻辑（镜像断言用） */ const isRetryableShape = (err: unknown): boolean => { const e = err as { status?: number; code?: string; message?: string }; if (e.status === 429) return true; if (e.status && e.status >= 500 && e.status < 600) return true; if (e.code === 'ECONNRESET' || e.code === 'ETIMEDOUT' || e.code === 'ENOTFOUND') return true; const msg = e.message?.toLowerCase() ?? ''; if (msg.includes('aborted') || msg.includes('socket hang up')) return true; return false; }; function makeResponse(status: number, body: string): Response { return new Response(body, { status, statusText: 'Status' }); } it('throwHttpError 携带 status；429/5xx 形状可重试，4xx 不可', async () => { const adapter = makeAdapter(); try { await adapter.throwHttpErrorPublic(makeResponse(429, 'rate limited'), 'T'); expect.fail('should throw'); } catch (e) { expect((e as Error & { status?: number }).status).toBe(429); expect(isRetryableShape(e)).toBe(true); } try { await adapter.throwHttpErrorPublic(makeResponse(401, ''), 'T'); expect.fail('should throw'); } catch (e) { expect((e as Error & { status?: number }).status).toBe(401); expect(isRetryableShape(e)).toBe(false); } }); it('content_filter 错误体抛出 ContentFilterError（含 status）', async () => { const adapter = makeAdapter(); const body = JSON.stringify({ error: { code: 'content_filter', message: 'high risk' } }); await expect( adapter.throwHttpErrorPublic(makeResponse(400, body), 'MiMo'), ).rejects.toBeInstanceOf(ContentFilterError); try { await adapter.throwHttpErrorPublic(makeResponse(400, body), 'MiMo'); } catch (e) { expect((e as ContentFilterError & { status: number }).status).toBe(400); expect((e as ContentFilterError).message).toContain('安全审核拦截'); } }); it('巨大 HTML 错误体在消息中被截断（v0.6.4）', async () => { const adapter = makeAdapter(); const bigHtml = `${'x'.repeat(100_000)}`; let caught: unknown; try { await adapter.throwHttpErrorPublic(makeResponse(502, bigHtml), 'GW'); expect.fail('should throw'); } catch (e) { caught = e; } expect((caught as Error).message.length).toBeLessThan(1_000); expect((caught as Error).message).toContain('[truncated'); // 截断不影响 status 判定 expect((caught as Error & { status?: number }).status).toBe(502); }); }); describe('BaseAdapter — getContextWindow', () => { it('配置了 contextWindow 时返回配置值', () => { const adapter = makeAdapter({ contextWindow: 128_000 }); expect(adapter.getContextWindow()).toBe(128_000); }); it('未配置时返回兜底默认值 1M（子类应覆盖真实窗口）', () => { const adapter = makeAdapter(); expect(adapter.getContextWindow()).toBe(1_000_000); }); }); describe('BaseAdapter — listModels / healthCheck', () => { it('listModels 将 supportedModels 映射为 MetonaModelInfo', async () => { const adapter = makeAdapter(); const models = await adapter.listModels(); expect(models.map((m) => m.id)).toEqual(['test-model-a', 'test-model-b']); }); it('healthCheck 成功返回 true', async () => { const adapter = makeAdapter(); expect(await adapter.healthCheck()).toBe(true); }); }); describe('BaseAdapter — fetchWithTimeout（P3-2 超时分类单轨化）', () => { afterEach(() => { vi.unstubAllGlobals(); vi.restoreAllMocks(); vi.useRealTimers(); }); it('正常请求返回 Response 并清理 timer', async () => { const adapter = makeAdapter(); const mockResponse = new Response('{"ok":true}'); const fetchMock = vi.fn().mockResolvedValue(mockResponse); vi.stubGlobal('fetch', fetchMock); const result = await adapter.fetchWithTimeoutPublic('https://api.test.com/v1/chat', {}, 5_000); expect(result).toBe(mockResponse); expect(fetchMock).toHaveBeenCalledOnce(); }); it('自身超时 → 显式 ETIMEDOUT + timed out 消息（命中引擎网络超时可重试分支）', async () => { const adapter = makeAdapter(); vi.useFakeTimers(); const fetchMock = vi.fn( (_url: string, init: RequestInit) => new Promise((_resolve, reject) => { init.signal?.addEventListener('abort', () => reject(new DOMException('This operation was aborted', 'AbortError')), ); }), ); vi.stubGlobal('fetch', fetchMock); const promise = adapter.fetchWithTimeoutPublic('https://api.test.com/v1/chat', {}, 100); const expectation = expect(promise).rejects.toSatisfy((e: Error & { code?: string }) => { expect(e.code).toBe('ETIMEDOUT'); expect(e.message).toContain('timed out after 100ms'); // 关键：消息不再是裸 "Aborted" —— 引擎按网络超时（而非碰巧可重试）分类 return true; }); vi.advanceTimersByTime(150); await expectation; vi.useRealTimers(); }); it('外部 abort（用户中断）→ 原样 AbortError，不被改写为超时', async () => { const adapter = makeAdapter(); const controller = new AbortController(); adapter.setAbortSignal(controller.signal); const fetchMock = vi.fn( (_url: string, init: RequestInit) => new Promise((_resolve, reject) => { init.signal?.addEventListener('abort', () => reject(new DOMException('This operation was aborted', 'AbortError')), ); }), ); vi.stubGlobal('fetch', fetchMock); const promise = adapter.fetchWithTimeoutPublic('https://api.test.com/v1/chat', {}, 30_000); controller.abort(); await promise.catch((e: Error & { code?: string }) => { expect(e.name).toBe('AbortError'); // 未被转译为字符串型 ETIMEDOUT（注意 Node DOMException 自带数字 code=20） expect(e.code).not.toBe('ETIMEDOUT'); expect(e.message).not.toContain('timed out after'); }); }); }); describe('BaseAdapter — 接口契约', () => { it('providerId / 能力声明符合 IMetonaProviderAdapter 契约', () => { const adapter: IMetonaProviderAdapter = makeAdapter(); expect(adapter.providerId).toBe('test'); expect(typeof adapter.send).toBe('function'); expect(typeof adapter.sendStream).toBe('function'); expect(typeof adapter.getContextWindow).toBe('function'); expect(typeof adapter.setAbortSignal).toBe('function'); }); it('sendStream 是 AsyncGenerator（可迭代）', async () => { const adapter = makeAdapter(); const events: MetonaStreamEvent[] = []; for await (const ev of adapter.sendStream({} as MetonaRequest)) { events.push(ev); } expect(events).toHaveLength(0); expect(MetonaStreamEventType.TEXT_DELTA).toBe('text_delta'); }); });