Function createCoderServiceMocks

src/node/runtime/coderLifecycleHooks.test.ts:39–82

(overrides?: {
  getWorkspaceStatus?: GetWorkspaceStatusMock;
  stopWorkspace?: StopWorkspaceMock;
  startWorkspace?: StartWorkspaceMock;
  deleteWorkspace?: DeleteWorkspaceMock;
})

type DeleteWorkspaceMock = ReturnType<typeof mock<(workspaceName: string) => Promise<void>>>;

function createCoderServiceMocks(overrides?: {
  getWorkspaceStatus?: GetWorkspaceStatusMock;
  stopWorkspace?: StopWorkspaceMock;
  startWorkspace?: StartWorkspaceMock;
  deleteWorkspace?: DeleteWorkspaceMock;
}): {
  coderService: CoderService;
  getWorkspaceStatus: GetWorkspaceStatusMock;
  stopWorkspace: StopWorkspaceMock;
  startWorkspace: StartWorkspaceMock;
  deleteWorkspace: DeleteWorkspaceMock;
} {
  const getWorkspaceStatus =
    overrides?.getWorkspaceStatus ??
    mock<
      (workspaceName: string, options?: { timeoutMs?: number }) => Promise<WorkspaceStatusResult>
    >(() => Promise.resolve({ kind: "ok", status: "running" }));
  const stopWorkspace =
    overrides?.stopWorkspace ??
    mock<(workspaceName: string, options?: { timeoutMs?: number }) => Promise<Result<void>>>(() =>
      Promise.resolve(Ok(undefined))
    );
  const startWorkspace =
    overrides?.startWorkspace ??
    mock<(workspaceName: string, options?: { timeoutMs?: number }) => Promise<Result<void>>>(() =>
      Promise.resolve(Ok(undefined))
    );
  const deleteWorkspace =
    overrides?.deleteWorkspace ??
    mock<(workspaceName: string) => Promise<void>>(() => Promise.resolve());

  return {
    coderService: {
      getWorkspaceStatus,
      stopWorkspace,
      startWorkspace,
      deleteWorkspace,
    } as unknown as CoderService,
    getWorkspaceStatus,
    stopWorkspace,
    startWorkspace,
    deleteWorkspace,
  };
}

function expectError(result: Result<void>): string {
  expect(result.success).toBe(false);