import { createElement } from "react";
import { afterEach, describe, expect, it, vi } from "vitest";
import { cleanup, render, screen } from "@testing-library/react";

describe("useRuntimeConnection", () => {
  afterEach(() => {
    cleanup();
    vi.resetModules();
    vi.restoreAllMocks();
  });

  it("selects the hermes provider from the active adapter type", async () => {
    vi.doMock("@/lib/gateway/GatewayClient", () => ({
      useGatewayConnection: () => ({
        client: {},
        status: "connected",
        gatewayUrl: "ws://localhost:18789",
        token: "",
        selectedAdapterType: "hermes",
        detectedAdapterType: "hermes",
        activeAdapterType: "hermes",
        localGatewayDefaults: null,
        error: null,
        connectPromptReady: true,
        shouldPromptForConnect: false,
        connect: async () => {},
        disconnect: () => {},
        useLocalGatewayDefaults: () => {},
        setGatewayUrl: () => {},
        setToken: () => {},
        setSelectedAdapterType: () => {},
        clearError: () => {},
      }),
    }));

    const { useRuntimeConnection } = await import("@/lib/runtime/useRuntimeConnection");

    const Probe = () => {
      const state = useRuntimeConnection({} as never);
      return createElement(
        "div",
        null,
        createElement("div", { "data-testid": "providerId" }, state.providerId),
        createElement("div", { "data-testid": "providerLabel" }, state.providerLabel)
      );
    };

    render(createElement(Probe));

    expect(screen.getByTestId("providerId")).toHaveTextContent("hermes");
    expect(screen.getByTestId("providerLabel")).toHaveTextContent("Hermes");
  });

  it("selects the custom provider from the active adapter type", async () => {
    vi.doMock("@/lib/gateway/GatewayClient", () => ({
      useGatewayConnection: () => ({
        client: {},
        status: "connected",
        gatewayUrl: "http://127.0.0.1:7770",
        token: "",
        selectedAdapterType: "custom",
        detectedAdapterType: "custom",
        activeAdapterType: "custom",
        localGatewayDefaults: null,
        error: null,
        connectPromptReady: true,
        shouldPromptForConnect: false,
        connect: async () => {},
        disconnect: () => {},
        useLocalGatewayDefaults: () => {},
        setGatewayUrl: () => {},
        setToken: () => {},
        setSelectedAdapterType: () => {},
        clearError: () => {},
      }),
    }));

    const { useRuntimeConnection } = await import("@/lib/runtime/useRuntimeConnection");

    const Probe = () => {
      const state = useRuntimeConnection({} as never);
      return createElement(
        "div",
        null,
        createElement("div", { "data-testid": "providerId" }, state.providerId),
        createElement("div", { "data-testid": "providerLabel" }, state.providerLabel)
      );
    };

    render(createElement(Probe));

    expect(screen.getByTestId("providerId")).toHaveTextContent("custom");
    expect(screen.getByTestId("providerLabel")).toHaveTextContent("Custom");
  });
});