|
| 1 | +/** |
| 2 | + * analyzePartial 页级切分测试（P1-5） |
| 3 | + * |
| 4 | + * @ai-context: 验证批内 slide_change 帧作为新页起点逐页独立分析：纯板书批 |
| 5 | + * 保持单次调用；多页批按页拆分调用、页内 [图:N] 重映射为批内编号（越界 |
| 6 | + * 幻觉编号移除）。外部 remapKeyframeMarkers 负责批内→全局的二次映射。 |
| 7 | + * English: page-level splitting of incremental keyframe batches — pure |
| 8 | + * board batches stay single-call; multi-page batches split per slide_change |
| 9 | + * with local [图:N] remapped to batch-level indices (out-of-range dropped). |
| 10 | + */ |
| 11 | +import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'; |
| 12 | + |
| 13 | +vi.mock('@/lib/auth/supabaseClient', () => ({ |
| 14 | + supabase: { |
| 15 | + auth: { |
| 16 | + getSession: vi.fn().mockResolvedValue({ data: { session: { access_token: 'test-token' } } }), |
| 17 | + }, |
| 18 | + }, |
| 19 | +})); |
| 20 | + |
| 21 | +import { analyzePartial } from './sessionAnalyzer'; |
| 22 | +import type { KeyFrame } from '@/lib/capture/captureTypes'; |
| 23 | + |
| 24 | +function makeKf(id: string, changeType: KeyFrame['changeType'], ts: number): KeyFrame { |
| 25 | + return { id, timestamp: ts, imageBase64: 'img', changeType }; |
| 26 | +} |
| 27 | + |
| 28 | +describe('analyzePartial 页级切分', () => { |
| 29 | + const invokeMock = vi.fn(); |
| 30 | + |
| 31 | + beforeEach(() => { |
| 32 | + invokeMock.mockReset(); |
| 33 | + (window as unknown as { electronAPI: unknown }).electronAPI = { invoke: invokeMock }; |
| 34 | + }); |
| 35 | + |
| 36 | + afterEach(() => { |
| 37 | + delete (window as unknown as { electronAPI?: unknown }).electronAPI; |
| 38 | + }); |
| 39 | + |
| 40 | + it('纯板书批（无 slide_change）：单次调用，编号不重映射', async () => { |
| 41 | + invokeMock.mockResolvedValue({ content: '知识点 A\n[图:1]\n知识点 B' }); |
| 42 | + const kfs = [ |
| 43 | + makeKf('a', 'writing', 1000), |
| 44 | + makeKf('b', 'writing', 2000), |
| 45 | + makeKf('c', 'scene_change', 3000), |
| 46 | + ]; |
| 47 | + |
| 48 | + const md = await analyzePartial(kfs, 0); |
| 49 | + |
| 50 | + expect(invokeMock).toHaveBeenCalledTimes(1); |
| 51 | + expect(invokeMock.mock.calls[0][0]).toBe('ai_session_analyze'); |
| 52 | + expect(invokeMock.mock.calls[0][1].keyframes).toHaveLength(3); |
| 53 | + expect(md).toBe('知识点 A\n[图:1]\n知识点 B'); |
| 54 | + }); |
| 55 | + |
| 56 | + it('含翻页批：按 slide_change 切页逐页分析，页内编号重映射为批内编号', async () => { |
| 57 | + // 批 = [w1, s1, w2, s2] → 页组 [[w1], [s1, w2], [s2]]，3 次调用 |
| 58 | + invokeMock |
| 59 | + .mockResolvedValueOnce({ content: '页1内容 [图:1]' }) |
| 60 | + .mockResolvedValueOnce({ content: '页2内容 [图:1] [图:2]' }) |
| 61 | + .mockResolvedValueOnce({ content: '页3内容 [图:1]' }); |
| 62 | + const kfs = [ |
| 63 | + makeKf('w1', 'writing', 1000), |
| 64 | + makeKf('s1', 'slide_change', 2000), |
| 65 | + makeKf('w2', 'writing', 3000), |
| 66 | + makeKf('s2', 'slide_change', 4000), |
| 67 | + ]; |
| 68 | + |
| 69 | + const md = await analyzePartial(kfs, 0); |
| 70 | + |
| 71 | + expect(invokeMock).toHaveBeenCalledTimes(3); |
| 72 | + // 各页组帧数：1 / 2 / 1（页首 slide_change 帧归入新页）；kfPayload 仅含 |
| 73 | + // timestamp/imageBase64/changeType（无 id），按时间戳断言 |
| 74 | + expect(invokeMock.mock.calls[0][1].keyframes.map((k: { timestamp: number }) => k.timestamp)).toEqual([1]); |
| 75 | + expect(invokeMock.mock.calls[1][1].keyframes.map((k: { timestamp: number }) => k.timestamp)).toEqual([2, 3]); |
| 76 | + expect(invokeMock.mock.calls[2][1].keyframes.map((k: { timestamp: number }) => k.timestamp)).toEqual([4]); |
| 77 | + // 页内编号 → 批内编号：页2 [图:1]→[图:2]、[图:2]→[图:3]；页3 [图:1]→[图:4] |
| 78 | + expect(md).toBe('页1内容 [图:1]\n\n页2内容 [图:2] [图:3]\n\n页3内容 [图:4]'); |
| 79 | + }); |
| 80 | + |
| 81 | + it('越界页内编号（模型幻觉）被移除', async () => { |
| 82 | + invokeMock |
| 83 | + .mockResolvedValueOnce({ content: '页1内容 [图:1]' }) |
| 84 | + .mockResolvedValueOnce({ content: '幻觉编号 [图:9] [图:0]' }); |
| 85 | + const kfs = [ |
| 86 | + makeKf('w1', 'writing', 1000), |
| 87 | + makeKf('s1', 'slide_change', 2000), |
| 88 | + makeKf('w2', 'writing', 3000), |
| 89 | + ]; |
| 90 | + |
| 91 | + const md = await analyzePartial(kfs, 0); |
| 92 | + |
| 93 | + // 越界编号移除后残留空格被 analyzePartial 的 trim() 收敛 |
| 94 | + expect(md).toBe('页1内容 [图:1]\n\n幻觉编号'); |
| 95 | + }); |
| 96 | + |
| 97 | + it('批首即翻页帧：首页从 slide_change 帧开始', async () => { |
| 98 | + invokeMock |
| 99 | + .mockResolvedValueOnce({ content: '页A [图:1]' }) |
| 100 | + .mockResolvedValueOnce({ content: '页B [图:1]' }); |
| 101 | + const kfs = [ |
| 102 | + makeKf('s1', 'slide_change', 1000), |
| 103 | + makeKf('w1', 'writing', 2000), |
| 104 | + makeKf('s2', 'slide_change', 3000), |
| 105 | + ]; |
| 106 | + |
| 107 | + const md = await analyzePartial(kfs, 0); |
| 108 | + |
| 109 | + // 页组 [[s1, w1], [s2]]：页1 offset 0（[图:1] 不变），页2 offset 2（[图:1]→[图:3]） |
| 110 | + expect(invokeMock).toHaveBeenCalledTimes(2); |
| 111 | + expect(md).toBe('页A [图:1]\n\n页B [图:3]'); |
| 112 | + }); |
| 113 | +}); |
0 commit comments