murashit_codex-panel/tests/features/chat/host/connection-bundle.test.ts

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

import {
  createServerRequestResponderRegistry,
  scheduleDeferredDiagnosticsRefresh,
} from "../../../../src/features/chat/host/bundles/connection-bundle";

describe("connection bundle server request responders", () => {
  it("responds through the client that delivered the request", () => {
    const currentClient = { respondToServerRequest: vi.fn() };
    const originalResponder = { respond: vi.fn(), reject: vi.fn() };
    const registry = createServerRequestResponderRegistry();
    registry.remember(1, originalResponder);

    registry.respond(1, { decision: "accept" });

    expect(originalResponder.respond).toHaveBeenCalledWith({ decision: "accept" });
    expect(currentClient.respondToServerRequest).not.toHaveBeenCalled();
  });

  it("cannot reuse a responder after the connection generation is cleared", () => {
    const responder = { respond: vi.fn(), reject: vi.fn() };
    const registry = createServerRequestResponderRegistry();
    registry.remember(1, responder);
    registry.clear();

    expect(registry.respond(1, { decision: "accept" })).toBe(false);
    expect(registry.reject(1, -32000, "cancelled")).toBe(false);
    expect(responder.respond).not.toHaveBeenCalled();
    expect(responder.reject).not.toHaveBeenCalled();
  });
});

describe("connection bundle deferred diagnostics", () => {
  it("reports deferred diagnostics failures without returning a blocking promise", async () => {
    const callbacks: Array<() => void> = [];
    const error = new Error("diagnostics failed");
    const refreshServerDiagnostics = vi.fn().mockRejectedValue(error);
    const addSystemMessage = vi.fn();

    scheduleDeferredDiagnosticsRefresh({
      scheduleDiagnostics: (scheduled) => {
        callbacks.push(scheduled);
      },
      isConnected: () => true,
      refreshServerDiagnostics,
      addSystemMessage,
    });

    expect(callbacks).toHaveLength(1);
    const scheduled = callbacks[0];
    if (!scheduled) throw new Error("Expected deferred diagnostics callback.");
    scheduled();
    await Promise.resolve();

    expect(refreshServerDiagnostics).toHaveBeenCalledWith({ appServerMetadataSnapshot: true });
    expect(addSystemMessage).toHaveBeenCalledWith("diagnostics failed");
  });

  it("does not refresh deferred diagnostics after disconnect", () => {
    const callbacks: Array<() => void> = [];
    const refreshServerDiagnostics = vi.fn().mockResolvedValue(undefined);

    scheduleDeferredDiagnosticsRefresh({
      scheduleDiagnostics: (scheduled) => {
        callbacks.push(scheduled);
      },
      isConnected: () => false,
      refreshServerDiagnostics,
      addSystemMessage: vi.fn(),
    });

    const scheduled = callbacks[0];
    if (!scheduled) throw new Error("Expected deferred diagnostics callback.");
    scheduled();

    expect(refreshServerDiagnostics).not.toHaveBeenCalled();
  });
});