openclaw/src/agents/cli-runner.test-support.ts

/** Shared CLI runner test doubles for supervisor, bootstrap, and heartbeat seams. */
import type { Mock } from "vitest";
import { beforeEach, vi } from "vitest";
import { getClaudeGeneration } from "./cli-runner/claude-live-registry.js";
import { createManagedRun, supervisorSpawnMock } from "./cli-runner/execute.test-support.js";
import { setCliRunnerPrepareTestDeps } from "./cli-runner/prepare.test-support.js";
import type { EmbeddedContextFile } from "./embedded-agent-helpers.js";
import type { WorkspaceBootstrapFile } from "./workspace.js";

export {
  createManagedRun,
  enqueueSystemEventMock,
  requestHeartbeatMock,
  supervisorSpawnMock,
} from "./cli-runner/execute.test-support.js";

// Shared CLI runner test doubles. They replace supervisor/process and bootstrap
// dependencies so CLI runner tests can assert process behavior deterministically.
type BootstrapContext = {
  bootstrapFiles: WorkspaceBootstrapFile[];
  contextFiles: EmbeddedContextFile[];
};
type ResolveBootstrapContextForRunMock = Mock<() => Promise<BootstrapContext>>;

const hoisted = vi.hoisted(
  (): {
    resolveBootstrapContextForRunMock: ResolveBootstrapContextForRunMock;
  } => {
    return {
      resolveBootstrapContextForRunMock: vi.fn<() => Promise<BootstrapContext>>(async () => ({
        bootstrapFiles: [],
        contextFiles: [],
      })),
    };
  },
);

setCliRunnerPrepareTestDeps({
  makeBootstrapWarn: () => () => {},
  resolveBootstrapContextForRun: hoisted.resolveBootstrapContextForRunMock,
  resolveOpenClawReferencePaths: async () => ({ docsPath: null, sourcePath: null }),
});

/** Queue one successful CLI supervisor run. */
export function mockSuccessfulCliRun(stdout = "ok") {
  supervisorSpawnMock.mockResolvedValueOnce(
    createManagedRun({
      reason: "exit",
      exitCode: 0,
      exitSignal: null,
      durationMs: 50,
      stdout,
      stderr: "",
      timedOut: false,
      noOutputTimedOut: false,
    }),
  );
}

/** Restore prepare-time CLI runner test dependencies after a test overrides them. */
export function restoreCliRunnerPrepareTestDeps() {
  setCliRunnerPrepareTestDeps({
    makeBootstrapWarn: () => () => {},
    resolveBootstrapContextForRun: hoisted.resolveBootstrapContextForRunMock,
    resolveOpenClawReferencePaths: async () => ({ docsPath: null, sourcePath: null }),
    getClaudeGeneration,
  });
}

beforeEach(() => {
  vi.unstubAllEnvs();
});