fix: show multi-run sessions immediately

Register created multi-run sessions in sidebar state Preserve reduced startup refresh fanout Add regression coverage
2026-05-25 17:37:27 +03:00
parent f585ca2f60
commit 25ed8e6abb
2 changed files with 192 additions and 0 deletions
@@ -0,0 +1,142 @@
+import { beforeEach, describe, expect, mock, test } from 'bun:test';
+import type { Session } from '@opencode-ai/sdk/v2';
+
+const upsertedSessions: Session[] = [];
+const registeredDirectories: Array<{ sessionID: string; directory: string }> = [];
+const ensureChildCalls: Array<{ directory: string; bootstrap?: boolean }> = [];
+const childState = {
+  session: [] as Session[],
+  sessionTotal: 0,
+  limit: 5,
+};
+let currentDirectory = '/repo';
+
+mock.module('@/sync/session-ui-store', () => ({
+  routeMessage: mock(() => Promise.resolve()),
+  useSessionUIStore: {
+    getState: () => ({
+      markSessionAsOpenChamberCreated: mock(() => undefined),
+    }),
+  },
+}));
+
+mock.module('@/lib/opencode/client', () => ({
+  opencodeClient: {
+    withDirectory: async (directory: string, fn: () => Promise<Session>) => {
+      const previous = currentDirectory;
+      currentDirectory = directory;
+      try {
+        return await fn();
+      } finally {
+        currentDirectory = previous;
+      }
+    },
+    createSession: async (params?: { title?: string }): Promise<Session> => ({
+      id: 'ses_multirun',
+      title: params?.title ?? '',
+      directory: currentDirectory,
+      time: { created: 1, updated: 1 },
+    } as Session),
+  },
+}));
+
+mock.module('@/lib/gitApi', () => ({
+  checkIsGitRepository: mock(() => Promise.resolve(false)),
+}));
+
+mock.module('@/lib/worktrees/worktreeCreate', () => ({
+  createWorktreeWithDefaults: mock(),
+  resolveRootTrackingRemote: mock(() => Promise.resolve(null)),
+}));
+
+mock.module('@/lib/worktrees/worktreeStatus', () => ({
+  getRootBranch: mock(() => Promise.resolve('main')),
+}));
+
+mock.module('@/lib/openchamberConfig', () => ({
+  saveWorktreeSetupCommands: mock(() => Promise.resolve()),
+}));
+
+mock.module('./useDirectoryStore', () => ({
+  useDirectoryStore: {
+    getState: () => ({ currentDirectory: '/repo' }),
+  },
+}));
+
+mock.module('./useProjectsStore', () => ({
+  useProjectsStore: {
+    getState: () => ({
+      activeProjectId: 'project-1',
+      projects: [{ id: 'project-1', path: '/repo' }],
+    }),
+  },
+}));
+
+mock.module('./useSnippetsStore', () => ({
+  useSnippetsStore: {
+    getState: () => ({
+      expandText: (value: string) => Promise.resolve(value),
+    }),
+  },
+}));
+
+mock.module('./useGlobalSessionsStore', () => ({
+  useGlobalSessionsStore: {
+    getState: () => ({
+      upsertSession: (session: Session) => {
+        upsertedSessions.push(session);
+      },
+    }),
+  },
+}));
+
+mock.module('@/sync/sync-refs', () => ({
+  registerSessionDirectory: (sessionID: string, directory: string) => {
+    registeredDirectories.push({ sessionID, directory });
+  },
+  getSyncChildStores: () => ({
+    ensureChild: (directory: string, options?: { bootstrap?: boolean }) => {
+      ensureChildCalls.push({ directory, bootstrap: options?.bootstrap });
+      return {
+        setState: (updater: typeof childState | ((state: typeof childState) => Partial<typeof childState> | typeof childState)) => {
+          const patch = typeof updater === 'function' ? updater(childState) : updater;
+          if (patch !== childState) {
+            Object.assign(childState, patch);
+          }
+        },
+      };
+    },
+  }),
+}));
+
+const { useMultiRunStore } = await import('./useMultiRunStore');
+
+describe('useMultiRunStore', () => {
+  beforeEach(() => {
+    upsertedSessions.length = 0;
+    registeredDirectories.length = 0;
+    ensureChildCalls.length = 0;
+    childState.session = [];
+    childState.sessionTotal = 0;
+    childState.limit = 5;
+    currentDirectory = '/repo';
+    useMultiRunStore.setState({ isLoading: false, error: null });
+  });
+
+  test('registers created sessions without waiting for a sidebar refresh', async () => {
+    const result = await useMultiRunStore.getState().createMultiRun({
+      name: 'Fix thing',
+      isolateRuns: false,
+      groups: [{
+        prompt: 'Fix it',
+        models: [{ providerID: 'anthropic', modelID: 'claude-sonnet-4-5' }],
+      }],
+    });
+
+    expect(result?.sessionIds).toEqual(['ses_multirun']);
+    expect(upsertedSessions.map((session) => session.id)).toEqual(['ses_multirun']);
+    expect(registeredDirectories).toEqual([{ sessionID: 'ses_multirun', directory: '/repo' }]);
+    expect(ensureChildCalls).toEqual([{ directory: '/repo', bootstrap: false }]);
+    expect(childState.session.map((session) => session.id)).toEqual(['ses_multirun']);
+  });
+});