import { describe, expect, it, vi } from 'vitest';

import { withDmsPreviewTicketRefresh } from '../helpers/dmsPreviewTicket';
import type { NormalizedOfficeDocumentConfig } from '../types';

const expired = () => new Error('DMS 预览票据无效或已过期');
function setup() {
  const refresh = vi
    .fn()
    .mockResolvedValue({ accessTicket: 'new', expiresAt: new Date(Date.now() + 300000).toISOString(), fileVersionId: 'v1', fileName: 'a.pdf' });
  const config: NormalizedOfficeDocumentConfig = {
    file: { fileId: 'v1', name: 'a.pdf' },
    fileType: 'dms',
    mode: 'view',
    title: 'a',
    dmsAccessTicket: 'old',
    refreshDmsPreviewTicket: refresh,
  };
  return { config, refresh };
}

describe('DMS preview ticket recovery', () => {
  it('refreshes immediately and retries with the new ticket', async () => {
    const { config, refresh } = setup();
    const load = vi
      .fn()
      .mockRejectedValueOnce(expired())
      .mockImplementationOnce(async () => config.dmsAccessTicket);
    await expect(withDmsPreviewTicketRefresh(config, () => true, load)).resolves.toBe('new');
    expect(refresh).toHaveBeenCalledTimes(1);
    expect(load).toHaveBeenCalledTimes(2);
  });
  it('refreshes a known expired ticket before loading', async () => {
    const { config, refresh } = setup();
    config.dmsPreviewExpiresAt = new Date(0).toISOString();
    const load = vi.fn(async () => config.dmsAccessTicket);
    await expect(withDmsPreviewTicketRefresh(config, () => true, load)).resolves.toBe('new');
    expect(refresh).toHaveBeenCalledTimes(1);
    expect(load).toHaveBeenCalledTimes(1);
  });
  it('stops after three retries', async () => {
    const { config, refresh } = setup();
    const load = vi.fn().mockRejectedValue(expired());
    await expect(withDmsPreviewTicketRefresh(config, () => true, load)).rejects.toThrow('已过期');
    expect(refresh).toHaveBeenCalledTimes(3);
    expect(load).toHaveBeenCalledTimes(4);
  });
  it('can recover on the third retry', async () => {
    const { config, refresh } = setup();
    const load = vi.fn().mockRejectedValueOnce(expired()).mockRejectedValueOnce(expired()).mockRejectedValueOnce(expired()).mockResolvedValueOnce('ready');
    await expect(withDmsPreviewTicketRefresh(config, () => true, load)).resolves.toBe('ready');
    expect(refresh).toHaveBeenCalledTimes(3);
    expect(load).toHaveBeenCalledTimes(4);
  });
  it('does not retry permission errors', async () => {
    const { config, refresh } = setup();
    await expect(withDmsPreviewTicketRefresh(config, () => true, vi.fn().mockRejectedValue(new Error('没有权限')))).rejects.toThrow('没有权限');
    expect(refresh).not.toHaveBeenCalled();
  });
  it('does not load after the window is closed during refresh', async () => {
    const { config, refresh } = setup();
    let current = true;
    refresh.mockImplementation(async () => {
      current = false;
      return {};
    });
    const load = vi.fn().mockRejectedValue(expired());
    await withDmsPreviewTicketRefresh(config, () => current, load);
    expect(load).toHaveBeenCalledTimes(1);
    expect(config.dmsAccessTicket).toBe('old');
  });
});