Function createSessionHarness

src/node/services/agentSession.editMessageId.test.ts:36–67

(
    workspaceId: string,
    streamHandler: StreamMessageHandler = () => Promise.resolve(Ok(undefined))
  )

  let historyCleanup: (() => Promise<void>) | undefined;

  async function createSessionHarness(
    workspaceId: string,
    streamHandler: StreamMessageHandler = () => Promise.resolve(Ok(undefined))
  ) {
    const { historyService, cleanup } = await createTestHistoryService();
    historyCleanup = cleanup;

    const streamMessage = mock(streamHandler);
    const aiService = Object.assign(new EventEmitter(), {
      isStreaming: mock((_workspaceId: string) => false),
      stopStream: mock((_workspaceId: string) => Promise.resolve(Ok(undefined))),
      streamMessage: streamMessage as unknown as AIService["streamMessage"],
    }) as unknown as AIService;

    return {
      historyService,
      streamMessage,
      session: new AgentSession({
        workspaceId,
        config,
        historyService,
        aiService,
        initStateManager: new EventEmitter() as unknown as InitStateManager,
        backgroundProcessManager: {
          cleanup: mock((_workspaceId: string) => Promise.resolve()),
          setMessageQueued: mock((_workspaceId: string, _queued: boolean) => {
            void _queued;
          }),
        } as unknown as BackgroundProcessManager,
      }),
    };
  }

  async function seedImageMessage(
    workspaceId: string,