import { beforeEach, describe, expect, it } from "vite-plus/test";

import {
  completeRelayClientInstallDialogClose,
  finishRelayClientInstall,
  readRelayClientInstallDialogState,
  reportRelayClientInstallProgress,
  requestRelayClientInstallConfirmation,
  resetRelayClientInstallDialogForTests,
  respondToRelayClientInstallConfirmation,
} from "./relayClientInstallDialog";

describe("relay client install dialog coordinator", () => {
  beforeEach(() => {
    resetRelayClientInstallDialogForTests();
  });

  it("moves a confirmed installation through streamed progress stages", async () => {
    const confirmation = requestRelayClientInstallConfirmation("2026.5.2");
    expect(readRelayClientInstallDialogState()).toEqual({
      status: "confirming",
      version: "2026.5.2",
    });

    respondToRelayClientInstallConfirmation(true);
    await expect(confirmation).resolves.toBe(true);
    expect(readRelayClientInstallDialogState()).toEqual({
      status: "installing",
      version: "2026.5.2",
      stage: "checking",
    });

    reportRelayClientInstallProgress({ type: "progress", stage: "downloading" });
    expect(readRelayClientInstallDialogState()).toEqual({
      status: "installing",
      version: "2026.5.2",
      stage: "downloading",
    });

    finishRelayClientInstall();
    expect(readRelayClientInstallDialogState()).toEqual({
      status: "closing",
      view: {
        status: "installing",
        version: "2026.5.2",
        stage: "downloading",
      },
    });

    completeRelayClientInstallDialogClose();
    expect(readRelayClientInstallDialogState()).toEqual({ status: "idle" });
  });

  it("returns to idle when installation is declined", async () => {
    const confirmation = requestRelayClientInstallConfirmation("2026.5.2");
    respondToRelayClientInstallConfirmation(false);

    await expect(confirmation).resolves.toBe(false);
    expect(readRelayClientInstallDialogState()).toEqual({
      status: "closing",
      view: {
        status: "confirming",
        version: "2026.5.2",
      },
    });

    completeRelayClientInstallDialogClose();
    expect(readRelayClientInstallDialogState()).toEqual({ status: "idle" });
  });
});