import { describe, expect, it } from 'vitest';
import { aggregateFiveDriveTraces } from '../../scripts/lib/performance-aggregation.mjs';

const run = (lcp: number, tbt: number, runIndex: number, gatingReferenceResult = true) => ({
  schema: 'drive-trace.v1',
  profile: 'desktop-high',
  referenceRun: { runId: `run-${runIndex}`, runIndex, attestationDigest: `${runIndex}`.repeat(64) },
  gatingReferenceResult,
  validReferenceCadence: true,
  metrics: { lcp, tbt },
  budgetResults: { cls: true, scene: true },
});

describe('five-run drive trace aggregation', () => {
  it('uses worst and median gates across exactly five reference runs', () => {
    expect(aggregateFiveDriveTraces([
      run(2_000, 100, 1), run(2_100, 110, 2), run(2_500, 200, 3), run(2_900, 250, 4), run(3_000, 300, 5),
    ])).toMatchObject({
      runCount: 5,
      passes: true,
      summary: { lcpWorst: 3_000, lcpMedian: 2_500, tbtWorst: 300, tbtMedian: 200 },
    });
  });

  it('rejects non-reference developer evidence and wrong run counts', () => {
    expect(aggregateFiveDriveTraces([
      run(2_000, 100, 1, false), run(2_000, 100, 2), run(2_000, 100, 3), run(2_000, 100, 4), run(2_000, 100, 5),
    ]).passes).toBe(false);
    expect(() => aggregateFiveDriveTraces([run(2_000, 100, 1)])).toThrow(/five/u);
  });

  it('rejects duplicate or incomplete authenticated reference identities', () => {
    const duplicate = [1, 2, 3, 4, 4].map((index) => run(2_000, 100, index));
    expect(() => aggregateFiveDriveTraces(duplicate)).toThrow(/run indexes 1 through 5/u);
  });
});