test: fix navigation and focus tests after rebase

- navigation.test.ts: add focusedPanel to mock store and deep-path setStore
  handler to match setActiveTask's usage of store.focusedPanel
- focus.test.ts: update ai-terminal panel expectation to include agent ID
  suffix since normalizeTaskPanel now resolves 'ai-terminal' → 'ai-terminal:<id>'

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: brooksc

src/store/focus.test.ts

@@ -204,7 +204,7 @@ describe('focus navigation neighbor map', () => {
 
     expect(mockStore.activeTaskId).toBe('task-1');
     expect(mockStore.sidebarFocused).toBe(false);
-    expect(mockStore.focusedPanel['task-1']).toBe('ai-terminal');
+    expect(mockStore.focusedPanel['task-1']).toBe('ai-terminal:agent-1');
   });
 
   it('clamps split shell-toolbar down-moves to the last available shell', () => {

src/store/navigation.test.ts

@@ -3,11 +3,12 @@ import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 type MockStore = {
   activeTaskId: string | null;
   activeAgentId: string | null;
-  tasks: Record<string, { id: string; agentIds: string[] }>;
+  tasks: Record<string, { id: string; agentIds: string[]; selectedAgentId?: string }>;
   terminals: Record<string, unknown>;
   taskOrder: string[];
   collapsedTaskOrder: string[];
   projects: Array<{ id: string }>;
+  focusedPanel: Record<string, string>;
 };
 
 let mockStore: MockStore;
@@ -22,9 +23,13 @@ vi.mock('./core', () => ({
     },
   ),
   setStore: vi.fn((...args: unknown[]) => {
-    const key = args[0] as keyof MockStore;
-    const value = args[1];
-    (mockStore as Record<string, unknown>)[key] = value;
+    const value = args[args.length - 1];
+    let target: Record<string, unknown> = mockStore as unknown as Record<string, unknown>;
+    for (let i = 0; i < args.length - 2; i++) {
+      const key = args[i] as string;
+      target = target[key] as Record<string, unknown>;
+    }
+    target[args[args.length - 2] as string] = value;
   }),
 }));
 
@@ -48,6 +53,7 @@ beforeEach(() => {
     taskOrder: ['task-1', 'task-2', 'task-3'],
     collapsedTaskOrder: [],
     projects: [],
+    focusedPanel: {},
   };
 });
 
@@ -82,6 +88,38 @@ describe('jumpToTask', () => {
     expect(mockStore.activeAgentId).toBe('agent-b');
   });
 
+  it('preserves activeAgentId when it already belongs to the target task', () => {
+    mockStore.tasks['task-2'].agentIds = ['agent-b', 'agent-b2'];
+    mockStore.activeAgentId = 'agent-b2';
+    jumpToTask(1);
+    expect(mockStore.activeAgentId).toBe('agent-b2');
+  });
+
+  it('prefers the focused AI pane when switching back to a multi-agent task', () => {
+    mockStore.tasks['task-1'].agentIds = ['agent-a', 'agent-a2'];
+    mockStore.activeTaskId = 'task-2';
+    mockStore.activeAgentId = 'agent-b';
+    mockStore.focusedPanel['task-1'] = 'ai-terminal:agent-a2';
+
+    jumpToTask(0);
+
+    expect(mockStore.activeTaskId).toBe('task-1');
+    expect(mockStore.activeAgentId).toBe('agent-a2');
+  });
+
+  it('restores the per-task selected agent when focus is on a non-agent panel', () => {
+    mockStore.tasks['task-1'].agentIds = ['agent-a', 'agent-a2'];
+    mockStore.tasks['task-1'].selectedAgentId = 'agent-a2';
+    mockStore.activeTaskId = 'task-2';
+    mockStore.activeAgentId = 'agent-b';
+    mockStore.focusedPanel['task-1'] = 'prompt';
+
+    jumpToTask(0);
+
+    expect(mockStore.activeTaskId).toBe('task-1');
+    expect(mockStore.activeAgentId).toBe('agent-a2');
+  });
+
   it('indexes taskOrder, not collapsed tasks', () => {
     // Collapsed tasks live in collapsedTaskOrder and must not be reachable
     // by index — the user can't see them, so jumping there would surprise.