import { describe, expect, test } from "bun:test";
import { getHeartbeatBatchSize, shouldSendHeartbeatPing } from "./maintenance-loops";

describe("maintenance heartbeat batching", () => {
  test("spreads large fleets across the heartbeat interval", () => {
    expect(getHeartbeatBatchSize(1000, 30_000)).toBe(34);
  });

  test("keeps small fleets responsive", () => {
    expect(getHeartbeatBatchSize(10, 30_000)).toBe(1);
  });

  test("handles empty fleets", () => {
    expect(getHeartbeatBatchSize(0, 30_000)).toBe(0);
  });

  test("only sends heartbeat pings after the per-client interval", () => {
    const now = Date.now();
    expect(shouldSendHeartbeatPing({}, now, 30_000)).toBe(true);
    expect(shouldSendHeartbeatPing({ lastPingSent: now - 29_999 }, now, 30_000)).toBe(false);
    expect(shouldSendHeartbeatPing({ lastPingSent: now - 30_000 }, now, 30_000)).toBe(true);
  });
});