frontend/test/jobs.test.ts

Rama master · réplica de lectura del repositorio original.

import { describe, it, expect, vi, afterEach } from 'vitest';
import { watchJob, followJobProgress, jobActivityHtml } from '../src/jobs';

afterEach(() => { vi.useRealTimers(); sessionStorage.clear(); document.body.innerHTML = ''; });

describe('后台任务状态恢复', () => {
  it('默认任务面板展示真实进度，失败原因持续可见', async () => {
    vi.useFakeTimers();const host=document.createElement('div');document.body.append(host);const complete=vi.fn();
    let state={status:'running',job_id:'default',checked:2,total:5,error:''};
    followJobProgress({host,active:()=>true,read:async()=>state,busy:()=>{},complete});
    await vi.advanceTimersByTimeAsync(0);
    expect(host.querySelector('[data-geist-fieldset]')).not.toBeNull();
    expect(host.querySelector('[role="progressbar"]')?.getAttribute('aria-valuenow')).toBe('2');
    state={...state,status:'failed',error:'来源不可用'};await vi.advanceTimersByTimeAsync(4000);
    expect(host.hidden).toBe(false);expect(host.textContent).toContain('来源不可用');expect(complete).toHaveBeenCalledOnce();
    expect(jobActivityHtml('正在扫描')).not.toContain('role="progressbar"');
  });
  it('后台阶段说明随轮询更新并保留真实计数', async () => {
    vi.useFakeTimers();
    const host=document.createElement('div');document.body.append(host);
    let state={status:'running',job_id:'taste',checked:0,total:3,message:'分析浏览记录：已处理 0 / 3 条'};
    followJobProgress({host,active:()=>true,read:async()=>state,busy:()=>{},complete:()=>{},
      title:'读取浏览记录',note:text=>text,loading:text=>text,progress:(value,max,label)=>`${label} ${value}/${max}`});
    await vi.advanceTimersByTimeAsync(0);
    expect(host.textContent).toContain(state.message);
    state={...state,checked:2,message:'分析浏览记录：已处理 2 / 3 条'};
    await vi.advanceTimersByTimeAsync(4000);
    expect(host.textContent).toContain(state.message);
    expect(host.textContent).toContain('2/3');
  });
  it('页面重建后恢复来源进度并只接收一次完成回执', async () => {
    vi.useFakeTimers();
    const host = document.createElement('div'); document.body.append(host);
    const complete = vi.fn();
    let state = {status: 'running', job_id: 'same', checked: 1, total: 3,
      current: {label: '演示来源', attempt: 3, max_attempts: 5, retry_in: 2}};
    const mount = (element: HTMLElement) => followJobProgress({host: element,
      active: () => true, read: async () => state, busy: () => {}, complete,
      note: text => text, loading: text => text, progress: (_value,_max,label) => label||'',
      container: content => `<section data-geist-fieldset>${content}</section>`});
    mount(host); await vi.advanceTimersByTimeAsync(0);
    expect(host.textContent).toContain('1/3');
    expect(host.querySelector('[data-geist-fieldset]')).not.toBeNull();
    expect(host.textContent).toContain('第 3/5 次尝试');
    host.remove();
    const fresh = document.createElement('div'); document.body.append(fresh);
    mount(fresh); await vi.advanceTimersByTimeAsync(0);
    expect(fresh.textContent).toContain('演示来源');
    state = {...state, status: 'complete', checked: 3};
    await vi.advanceTimersByTimeAsync(4000);
    expect(complete).toHaveBeenCalledTimes(1);
    expect(sessionStorage.getItem('peach-follow-job')).toBeNull();
  });
  it('连接中断后继续读取相同任务直至终态', async () => {
    const states: string[] = [], waits: number[] = [];
    let calls = 0, failures = 0;
    await watchJob({ active: () => true,
      read: async () => { calls++; if (calls === 2) throw new Error('offline');
        return { status: calls < 3 ? 'running' : 'complete', job_id: 'one' }; },
      render: state => states.push(state.status),
      disconnected: () => { failures++; }, pause: async ms => { waits.push(ms); },
    });
    expect(states).toEqual(['running', 'complete']);
    expect(failures).toBe(1);
    expect(waits).toEqual([2000, 4000]);
  });
  it('离开页面后丢弃在途响应', async () => {
    let active = true, rendered = false;
    await watchJob({ active: () => active,
      read: async () => { active = false; return { status: 'running' }; },
      render: () => { rendered = true; }, disconnected: () => {},
    });
    expect(rendered).toBe(false);
  });
  it('首读恢复只交接一次运行任务', async () => {
    let calls = 0;
    await watchJob({ active: () => true, once: true,
      read: async () => { calls++; return { status: 'running' }; },
      render: () => {}, disconnected: () => {},
    });
    expect(calls).toBe(1);
  });
});