import { QueryClient } from "@tanstack/react-query";import { EnvironmentId } from "@t3tools/contracts";import { afterEach, beforeEach, describe, expect, it, vi } from "vite-plus/test"; const mockCreateEnvironmentConnection = vi.fn();const mockCreateWsRpcClient = vi.fn();const mockFetchRemoteSessionState = vi.fn();const mockResolveRemoteWebSocketConnectionUrl = vi.fn(() => "ws://remote.example.test");const mockRemoteHttpRunPromise = vi.fn((effect: Promise<unknown>) => effect);const mockWaitForSavedEnvironmentRegistryHydration = vi.fn();const mockListSavedEnvironmentRecords = vi.fn();const mockSavedEnvironmentRegistrySubscribe = vi.fn();const mockReadSavedEnvironmentBearerToken = vi.fn();const mockGetSavedEnvironmentRecord = vi.fn(); function MockWsTransport() {  return undefined;} vi.mock("../primary", () => ({  getPrimaryKnownEnvironment: vi.fn(() => ({    id: "env-1",    label: "Primary environment",    source: "window-origin",    target: {      httpBaseUrl: "http://127.0.0.1:3000/",      wsBaseUrl: "ws://127.0.0.1:3000/",    },    environmentId: EnvironmentId.make("env-1"),  })),})); vi.mock("../../lib/runtime", () => ({  remoteHttpRuntime: {    runPromise: mockRemoteHttpRunPromise,  },})); vi.mock("./catalog", () => ({  getSavedEnvironmentRecord: mockGetSavedEnvironmentRecord,  hasSavedEnvironmentRegistryHydrated: vi.fn(() => true),  listSavedEnvironmentRecords: mockListSavedEnvironmentRecords,  persistSavedEnvironmentRecord: vi.fn(),  readSavedEnvironmentBearerToken: mockReadSavedEnvironmentBearerToken,  removeSavedEnvironmentBearerToken: vi.fn(),  useSavedEnvironmentRegistryStore: {    subscribe: mockSavedEnvironmentRegistrySubscribe,    getState: () => ({      upsert: vi.fn(),      remove: vi.fn(),      markConnected: vi.fn(),      rename: vi.fn(),    }),  },  useSavedEnvironmentRuntimeStore: {    getState: () => ({      ensure: vi.fn(),      patch: vi.fn(),      clear: vi.fn(),    }),  },  waitForSavedEnvironmentRegistryHydration: mockWaitForSavedEnvironmentRegistryHydration,  writeSavedEnvironmentBearerToken: vi.fn(),})); vi.mock("./connection", async (importOriginal) => ({  ...(await importOriginal<typeof import("./connection")>()),  createEnvironmentConnection: mockCreateEnvironmentConnection,})); vi.mock("@t3tools/client-runtime", async (importOriginal) => {  const actual = await importOriginal<typeof import("@t3tools/client-runtime")>();  return {    ...actual,    createWsRpcClient: mockCreateWsRpcClient,    fetchRemoteSessionState: mockFetchRemoteSessionState,    resolveRemoteWebSocketConnectionUrl: mockResolveRemoteWebSocketConnectionUrl,  };}); vi.mock("../../rpc/wsTransport", () => ({  WsTransport: MockWsTransport,})); vi.mock("~/composerDraftStore", () => ({  markPromotedDraftThreadByRef: vi.fn(),  markPromotedDraftThreadsByRef: vi.fn(),  useComposerDraftStore: {    getState: () => ({      getDraftThreadByRef: vi.fn(() => null),      clearDraftThread: vi.fn(),    }),  },})); vi.mock("~/localApi", () => ({  ensureLocalApi: vi.fn(() => ({    persistence: {      setSavedEnvironmentRegistry: vi.fn(async () => undefined),    },  })),})); vi.mock("~/lib/terminalStateCleanup", () => ({  collectActiveTerminalThreadIds: vi.fn(() => []),})); vi.mock("~/orchestrationEventEffects", () => ({  deriveOrchestrationBatchEffects: vi.fn(() => ({    promotedThreadRefs: [],    invalidatedProviderState: false,  })),})); vi.mock("~/store", () => ({  useStore: {    getState: () => ({      syncServerShellSnapshot: vi.fn(),      syncServerThreadDetail: vi.fn(),      removeServerThreadDetail: vi.fn(),      applyServerShellEvent: vi.fn(),    }),  },  selectProjectsAcrossEnvironments: vi.fn(() => []),  selectSidebarThreadSummaryByRef: vi.fn(() => null),  selectThreadByRef: vi.fn(() => null),  selectThreadsAcrossEnvironments: vi.fn(() => []),})); vi.mock("~/terminalStateStore", () => ({  useTerminalStateStore: {    getState: () => ({      applyTerminalEvent: vi.fn(),      removeTerminalState: vi.fn(),      clearTerminalSelection: vi.fn(),    }),  },})); vi.mock("~/uiStateStore", () => ({  useUiStateStore: {    getState: () => ({      clearThreadUi: vi.fn(),      syncPromotedDraftThreadRefs: vi.fn(),    }),  },})); const savedRecord = {  environmentId: EnvironmentId.make("env-saved"),  label: "Remote environment",  httpBaseUrl: "https://remote.example.test/",  wsBaseUrl: "wss://remote.example.test/",}; const configSnapshot = {  environment: {    environmentId: savedRecord.environmentId,    label: "Remote environment",  },}; function createClient() {  return {    dispose: vi.fn(async () => undefined),    reconnect: vi.fn(async () => undefined),    server: {      getConfig: vi.fn(async () => configSnapshot),      subscribeConfig: vi.fn(() => () => undefined),      subscribeLifecycle: vi.fn(() => () => undefined),      subscribeAuthAccess: vi.fn(() => () => undefined),      refreshProviders: vi.fn(async () => undefined),      upsertKeybinding: vi.fn(async () => undefined),      getSettings: vi.fn(async () => undefined),      updateSettings: vi.fn(async () => undefined),    },    orchestration: {      subscribeShell: vi.fn(() => () => undefined),      subscribeThread: vi.fn(() => () => undefined),      dispatchCommand: vi.fn(async () => undefined),      getTurnDiff: vi.fn(async () => undefined),      getFullThreadDiff: vi.fn(async () => undefined),    },    terminal: {      open: vi.fn(async () => undefined),      write: vi.fn(async () => undefined),      resize: vi.fn(async () => undefined),      clear: vi.fn(async () => undefined),      restart: vi.fn(async () => undefined),      close: vi.fn(async () => undefined),      onMetadata: vi.fn(() => () => undefined),    },    projects: {      searchEntries: vi.fn(async () => []),      writeFile: vi.fn(async () => undefined),    },    shell: {      openInEditor: vi.fn(async () => undefined),    },    git: {      pull: vi.fn(async () => undefined),      refreshStatus: vi.fn(async () => undefined),      onStatus: vi.fn(() => () => undefined),      runStackedAction: vi.fn(async () => ({})),      listBranches: vi.fn(async () => []),      createWorktree: vi.fn(async () => undefined),      removeWorktree: vi.fn(async () => undefined),      createBranch: vi.fn(async () => undefined),      checkout: vi.fn(async () => undefined),      init: vi.fn(async () => undefined),      resolvePullRequest: vi.fn(async () => undefined),      preparePullRequestThread: vi.fn(async () => undefined),    },  };} describe("saved environment startup", () => {  beforeEach(() => {    vi.useFakeTimers();    vi.resetModules();    vi.clearAllMocks();     mockFetchRemoteSessionState.mockResolvedValue({      authenticated: true,      scopes: ["orchestration:read", "access:write"],    });    mockGetSavedEnvironmentRecord.mockImplementation((environmentId: EnvironmentId) =>      environmentId === savedRecord.environmentId ? savedRecord : null,    );    mockListSavedEnvironmentRecords.mockReturnValue([savedRecord]);    mockSavedEnvironmentRegistrySubscribe.mockReturnValue(() => undefined);    mockWaitForSavedEnvironmentRegistryHydration.mockResolvedValue(undefined);    mockReadSavedEnvironmentBearerToken.mockResolvedValue("saved-bearer-token");    mockCreateWsRpcClient.mockImplementation(() => createClient());    mockCreateEnvironmentConnection.mockImplementation((input) => {      if (input.kind === "saved") {        queueMicrotask(() => {          input.onConfigSnapshot?.(configSnapshot);        });      }       return {        kind: input.kind,        environmentId: input.knownEnvironment.environmentId,        knownEnvironment: input.knownEnvironment,        client: input.client,        ensureBootstrapped: vi.fn(async () => undefined),        reconnect: vi.fn(async () => undefined),        dispose: vi.fn(async () => undefined),      };    });  });   afterEach(async () => {    const { resetEnvironmentServiceForTests } = await import("./service");    await resetEnvironmentServiceForTests();    vi.useRealTimers();  });   it("uses the initial config snapshot instead of issuing an extra getConfig call", async () => {    const { startEnvironmentConnectionService, resetEnvironmentServiceForTests } =      await import("./service");     const stop = startEnvironmentConnectionService(new QueryClient());    await vi.runAllTimersAsync();     const savedConnectionCall = mockCreateEnvironmentConnection.mock.calls.find(      ([input]) => input.kind === "saved",    );    expect(savedConnectionCall).toBeDefined();     const savedClient = savedConnectionCall?.[0]?.client;    expect(savedClient.server.getConfig).not.toHaveBeenCalled();    expect(mockFetchRemoteSessionState).toHaveBeenCalledTimes(1);     stop();    await resetEnvironmentServiceForTests();  });   it("coalesces hydration and registry sync so the initial saved connection only starts once", async () => {    let finishHydration!: () => void;    let finishTokenRead!: (token: string) => void;     mockWaitForSavedEnvironmentRegistryHydration.mockImplementation(      () =>        new Promise<void>((resolve) => {          finishHydration = () => resolve();        }),    );    mockReadSavedEnvironmentBearerToken.mockImplementation(      () =>        new Promise<string>((resolve) => {          finishTokenRead = resolve;        }),    );     const { startEnvironmentConnectionService, resetEnvironmentServiceForTests } =      await import("./service");     const stop = startEnvironmentConnectionService(new QueryClient());    const registryListener = mockSavedEnvironmentRegistrySubscribe.mock.calls[0]?.[0];    expect(registryListener).toBeTypeOf("function");     registryListener?.();    finishHydration();    await vi.waitFor(() => {      expect(mockReadSavedEnvironmentBearerToken).toHaveBeenCalledTimes(1);    });     finishTokenRead("saved-bearer-token");    await vi.runAllTimersAsync();     const savedConnectionCalls = mockCreateEnvironmentConnection.mock.calls.filter(      ([input]) => input.kind === "saved",    );    expect(savedConnectionCalls).toHaveLength(1);    expect(mockFetchRemoteSessionState).toHaveBeenCalledTimes(1);     stop();    await resetEnvironmentServiceForTests();  });});