PlantAI-rag-new/ui/tests/r3debug/reveal.test.js

// tests/r3debug/reveal.test.js —— 缺陷 #7a reveal 超时（fake-timer 断言）
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
import { setActivePinia, createPinia } from 'pinia';
import { useChatStore } from '../../src/stores/chat.js';

vi.mock('../../src/api/sessions.js', () => ({
  createMessage: vi.fn(async () => ({ id: 'm1' })),
  updateMessage: vi.fn(async () => ({ updated_at: new Date().toISOString() })),
}));
vi.mock('../../src/api/qa.js', () => ({
  chatStream: vi.fn(async () => {}),
  search: vi.fn(async () => ({ answer: '' })),
}));
vi.mock('../../src/api/private.js', () => ({
  askPrivate: vi.fn(async () => ({ hit: false })),
}));

function fakeCtx(msgList, aMsg) {
  return { sessionId: 's1', msgList, aMsg, events: [], controller: null, reveal: null };
}

describe('#7a reveal 总时长上限', () => {
  beforeEach(() => {
    setActivePinia(createPinia());
    vi.useFakeTimers();
    // jsdom 无 matchMedia：revealText 判定 reduced 直接定格——此处 stub 返回不命中 reduce
    vi.stubGlobal('matchMedia', vi.fn(() => ({ matches: false })));
    Object.defineProperty(window, 'matchMedia', {
      writable: true,
      value: vi.fn(() => ({ matches: false })),
    });
  });
  afterEach(() => {
    vi.useRealTimers();
    vi.unstubAllGlobals();
    vi.restoreAllMocks();
  });

  it('≤2000 字符：30ms/字符节奏不变', async () => {
    const chat = useChatStore();
    const text = 'x'.repeat(500); // 500 字符 → 预期 500 × 30ms = 15000ms
    const aMsg = { blocks: [], revealId: 0, status: 'streaming' };
    const ctx = fakeCtx([aMsg], aMsg);
    const p = chat.revealText(aMsg, text, ctx);
    await vi.advanceTimersByTimeAsync(15000);
    await p;
    // 用真实时间轴：15000ms 恰好收尾；中途（14970ms）未收尾
    expect(aMsg.status).toBe('done');
    expect(aMsg.blocks.some((b) => b.type === 'text')).toBe(true);
    expect(aMsg.blocks.some((b) => b.type === 'stream')).toBe(false);
  });

  it('≤2000 字符路径：总时长 = 字符数 × 30ms（±1 tick）', async () => {
    const chat = useChatStore();
    const text = 'x'.repeat(1999); // 需 1999 tick × 30ms = 59970ms
    const aMsg = { blocks: [], revealId: 0, status: 'streaming' };
    const ctx = fakeCtx([aMsg], aMsg);
    const p = chat.revealText(aMsg, text, ctx);
    // 1998 tick（59940ms）后未完成
    await vi.advanceTimersByTimeAsync(1998 * 30);
    await Promise.resolve();
    expect(aMsg.status).toBe('streaming');
    // 再推进 2 tick（60ms）必完成
    await vi.advanceTimersByTimeAsync(60);
    await p;
    expect(aMsg.status).toBe('done');
  });

  it('>2000 字符：总时长 ≤8s（interval 下限 12ms）', async () => {
    const chat = useChatStore();
    const text = 'x'.repeat(5000); // 长文：加速分支
    const aMsg = { blocks: [], revealId: 0, status: 'streaming' };
    const ctx = fakeCtx([aMsg], aMsg);
    const p = chat.revealText(aMsg, text, ctx);
    await vi.advanceTimersByTimeAsync(8000);
    await p;
    expect(aMsg.status).toBe('done');
    expect(aMsg.blocks.some((b) => b.type === 'text')).toBe(true);
  });

  it('长文单 tick 步进不超过 interval 允许的上限（不空转）', async () => {
    const chat = useChatStore();
    const text = 'x'.repeat(100000);
    const aMsg = { blocks: [], revealId: 0, status: 'streaming' };
    const ctx = fakeCtx([aMsg], aMsg);
    const p = chat.revealText(aMsg, text, ctx);
    // 观察若干 tick：stream 块内容单调递增且每 tick 增量 > 0
    let prev = 0;
    for (let i = 0; i < 10; i++) {
      await vi.advanceTimersByTimeAsync(100);
      const sb = aMsg.blocks[aMsg.blocks.length - 1];
      expect(sb.text.length).toBeGreaterThan(prev);
      prev = sb.text.length;
    }
    await vi.advanceTimersByTimeAsync(8000);
    await p;
    expect(aMsg.status).toBe('done');
  });

  it('revealId 守卫：skipReveal 后旧定时器下一 tick 自愈退出', async () => {
    const chat = useChatStore();
    chat.sessionId = 's1'; // skipReveal 经 _ctx getter 取当前会话流上下文
    const text = 'x'.repeat(3000);
    const aMsg = { blocks: [], revealId: 0, status: 'streaming' };
    const ctx = fakeCtx([aMsg], aMsg);
    chat.streamCtx['s1'] = ctx;
    const p = chat.revealText(aMsg, text, ctx);
    await vi.advanceTimersByTimeAsync(100);
    chat.skipReveal();
    await vi.advanceTimersByTimeAsync(500);
    await p;
    expect(aMsg.status).toBe('done');
    expect(aMsg.blocks.some((b) => b.type === 'stream')).toBe(false);
    expect(aMsg.blocks.some((b) => b.type === 'text')).toBe(true);
  });
});