import { describe, expect, it } from "vitest";

import type { SpeedtestTrackerDashboardData, SpeedtestTrackerStats } from "@homarr/integrations/types";

import { combineSpeedtestDashboards, mergeStats } from "./helpers";

const stats = (average: number, totalResults: number): SpeedtestTrackerStats => ({
  ping: { avg: average, min: average, max: average },
  download: { avg: average, avg_bits: average, min: average, max: average },
  upload: { avg: average, avg_bits: average, min: average, max: average },
  total_results: totalResults,
});

describe("speedtest dashboard aggregation", () => {
  it("weights averages by result count", () => {
    expect(mergeStats(stats(10, 1), stats(20, 3))?.ping.avg).toBe(17.5);
  });

  it("selects the newest latest result independent of integration order", () => {
    const older = { id: 1, ping: 1, download_bits: 1, upload_bits: 1, healthy: true, created_at: new Date(1) };
    const newer = { ...older, id: 2, created_at: new Date(2) };
    const dashboards: SpeedtestTrackerDashboardData[] = [
      { latestResult: newer, stats: null, recentResults: [] },
      { latestResult: older, stats: null, recentResults: [] },
    ];
    expect(combineSpeedtestDashboards(dashboards).latestResult?.id).toBe(2);
  });
});