Agent-JAE/packages/coding-agent/test/tree-selector.test.ts

625 lines
19 KiB
TypeScript

import { setKeybindings } from "@jaeswift/jae-tui";
import { beforeAll, beforeEach, describe, expect, test } from "vitest";
import { KeybindingsManager } from "../src/core/keybindings.js";
import type {
ModelChangeEntry,
SessionEntry,
SessionMessageEntry,
SessionTreeNode,
} from "../src/core/session-manager.js";
import { TreeSelectorComponent } from "../src/modes/interactive/components/tree-selector.js";
import { initTheme } from "../src/modes/interactive/theme/theme.js";
beforeAll(() => {
initTheme("dark");
});
beforeEach(() => {
// Ensure test isolation: keybindings are a global singleton
setKeybindings(new KeybindingsManager());
});
// Helper to create a user message entry
function userMessage(id: string, parentId: string | null, content: string): SessionMessageEntry {
return {
type: "message",
id,
parentId,
timestamp: new Date().toISOString(),
message: { role: "user", content, timestamp: Date.now() },
};
}
// Helper to create an assistant message entry
function assistantMessage(id: string, parentId: string | null, text: string): SessionMessageEntry {
return {
type: "message",
id,
parentId,
timestamp: new Date().toISOString(),
message: {
role: "assistant",
content: [{ type: "text", text }],
api: "anthropic-messages",
provider: "anthropic",
model: "claude-sonnet-4",
usage: {
input: 0,
output: 0,
cacheRead: 0,
cacheWrite: 0,
totalTokens: 0,
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
},
stopReason: "stop",
timestamp: Date.now(),
},
};
}
// Helper to create a tool-call-only assistant message (filtered out in default mode)
function toolCallOnlyAssistant(id: string, parentId: string | null): SessionMessageEntry {
return {
type: "message",
id,
parentId,
timestamp: new Date().toISOString(),
message: {
role: "assistant",
content: [{ type: "toolCall", id: `tc-${id}`, name: "read", arguments: { path: "test.ts" } }],
api: "anthropic-messages",
provider: "anthropic",
model: "claude-sonnet-4",
usage: {
input: 0,
output: 0,
cacheRead: 0,
cacheWrite: 0,
totalTokens: 0,
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
},
stopReason: "toolUse",
timestamp: Date.now(),
},
};
}
// Helper to create a model_change entry
function modelChange(id: string, parentId: string | null): ModelChangeEntry {
return {
type: "model_change",
id,
parentId,
timestamp: new Date().toISOString(),
provider: "anthropic",
modelId: "claude-sonnet-4",
};
}
// Helper to build a tree from entries using parentId relationships
function buildTree(entries: Array<SessionEntry>): SessionTreeNode[] {
if (entries.length === 0) return [];
const nodes: SessionTreeNode[] = entries.map((entry) => ({
entry,
children: [],
}));
const byId = new Map<string, SessionTreeNode>();
for (const node of nodes) {
byId.set(node.entry.id, node);
}
const roots: SessionTreeNode[] = [];
for (const node of nodes) {
if (node.entry.parentId === null) {
roots.push(node);
} else {
const parent = byId.get(node.entry.parentId);
if (parent) {
parent.children.push(node);
}
}
}
return roots;
}
describe("TreeSelectorComponent", () => {
describe("initial selection with metadata entries", () => {
test("focuses nearest visible ancestor when currentLeafId is a model_change with sibling branch", () => {
// Tree structure:
// user-1
// └── asst-1
// ├── user-2 (active branch)
// │ └── model-1 (model_change, CURRENT LEAF)
// └── user-3 (sibling branch, added later chronologically)
const entries = [
userMessage("user-1", null, "hello"),
assistantMessage("asst-1", "user-1", "hi"),
userMessage("user-2", "asst-1", "active branch"), // Active branch
modelChange("model-1", "user-2"), // Current leaf (metadata)
userMessage("user-3", "asst-1", "sibling branch"), // Sibling branch
];
const tree = buildTree(entries);
const selector = new TreeSelectorComponent(
tree,
"model-1", // currentLeafId is the model_change entry
24,
() => {},
() => {},
);
const list = selector.getTreeList();
// Should focus on user-2 (parent of model-1), not user-3 (last item)
expect(list.getSelectedNode()?.entry.id).toBe("user-2");
});
test("focuses nearest visible ancestor when currentLeafId is a thinking_level_change entry", () => {
// Similar structure with thinking_level_change instead of model_change
const entries = [
userMessage("user-1", null, "hello"),
assistantMessage("asst-1", "user-1", "hi"),
userMessage("user-2", "asst-1", "active branch"),
{
type: "thinking_level_change" as const,
id: "thinking-1",
parentId: "user-2",
timestamp: new Date().toISOString(),
thinkingLevel: "high",
},
userMessage("user-3", "asst-1", "sibling branch"),
];
const tree = buildTree(entries);
const selector = new TreeSelectorComponent(
tree,
"thinking-1",
24,
() => {},
() => {},
);
const list = selector.getTreeList();
expect(list.getSelectedNode()?.entry.id).toBe("user-2");
});
});
describe("filter switching with parent traversal", () => {
test("switches to nearest visible user message when changing to user-only filter", () => {
// In user-only filter: [user-1, user-2, user-3]
const entries = [
userMessage("user-1", null, "hello"),
assistantMessage("asst-1", "user-1", "hi"),
userMessage("user-2", "asst-1", "active branch"),
assistantMessage("asst-2", "user-2", "response"),
userMessage("user-3", "asst-1", "sibling branch"),
];
const tree = buildTree(entries);
const selector = new TreeSelectorComponent(
tree,
"asst-2",
24,
() => {},
() => {},
);
const list = selector.getTreeList();
expect(list.getSelectedNode()?.entry.id).toBe("asst-2");
// Simulate Ctrl+U (user-only filter)
selector.handleInput("\x15");
// Should now be on user-2 (the parent user message), not user-3
expect(list.getSelectedNode()?.entry.id).toBe("user-2");
});
test("returns to nearest visible ancestor when switching back to default filter", () => {
// Same branching structure
const entries = [
userMessage("user-1", null, "hello"),
assistantMessage("asst-1", "user-1", "hi"),
userMessage("user-2", "asst-1", "active branch"),
assistantMessage("asst-2", "user-2", "response"),
userMessage("user-3", "asst-1", "sibling branch"),
];
const tree = buildTree(entries);
const selector = new TreeSelectorComponent(
tree,
"asst-2",
24,
() => {},
() => {},
);
const list = selector.getTreeList();
expect(list.getSelectedNode()?.entry.id).toBe("asst-2");
// Switch to user-only
selector.handleInput("\x15"); // Ctrl+U
expect(list.getSelectedNode()?.entry.id).toBe("user-2");
// Switch back to default - should stay on user-2
// (since that's what we navigated to via parent traversal)
selector.handleInput("\x04"); // Ctrl+D
expect(list.getSelectedNode()?.entry.id).toBe("user-2");
});
});
describe("empty filter preservation", () => {
test("preserves selection when switching to empty labeled filter and back", () => {
// Tree with no labels
const entries = [
userMessage("user-1", null, "hello"),
assistantMessage("asst-1", "user-1", "hi"),
userMessage("user-2", "asst-1", "bye"),
assistantMessage("asst-2", "user-2", "goodbye"),
];
const tree = buildTree(entries);
const selector = new TreeSelectorComponent(
tree,
"asst-2",
24,
() => {},
() => {},
);
const list = selector.getTreeList();
expect(list.getSelectedNode()?.entry.id).toBe("asst-2");
// Switch to labeled-only filter (no labels exist, so empty result)
selector.handleInput("\x0c"); // Ctrl+L
// The list should be empty, getSelectedNode returns undefined
expect(list.getSelectedNode()).toBeUndefined();
// Switch back to default filter
selector.handleInput("\x04"); // Ctrl+D
// Should restore to asst-2 (the selection before we switched to empty filter)
expect(list.getSelectedNode()?.entry.id).toBe("asst-2");
});
test("preserves selection through multiple empty filter switches", () => {
const entries = [userMessage("user-1", null, "hello"), assistantMessage("asst-1", "user-1", "hi")];
const tree = buildTree(entries);
const selector = new TreeSelectorComponent(
tree,
"asst-1",
24,
() => {},
() => {},
);
const list = selector.getTreeList();
expect(list.getSelectedNode()?.entry.id).toBe("asst-1");
// Switch to labeled-only (empty) - Ctrl+L toggles labeled ↔ default
selector.handleInput("\x0c"); // Ctrl+L -> labeled-only
expect(list.getSelectedNode()).toBeUndefined();
// Switch to default, then back to labeled-only
selector.handleInput("\x0c"); // Ctrl+L -> default (toggle back)
expect(list.getSelectedNode()?.entry.id).toBe("asst-1");
selector.handleInput("\x0c"); // Ctrl+L -> labeled-only again
expect(list.getSelectedNode()).toBeUndefined();
// Switch back to default with Ctrl+D
selector.handleInput("\x04"); // Ctrl+D
expect(list.getSelectedNode()?.entry.id).toBe("asst-1");
});
});
describe("branch navigation and folding with ctrl+arrow keys", () => {
// Key escape sequences
const UP = "\x1b[A";
const DOWN = "\x1b[B";
const CTRL_LEFT = "\x1b[1;5D";
const CTRL_RIGHT = "\x1b[1;5C";
const ALT_LEFT = "\x1b[1;3D";
const ALT_RIGHT = "\x1b[1;3C";
// Tree structure:
//
// user-1
// asst-1
// user-2
// asst-2 ← branch point (has 2 children)
// ├─ user-3a ← branch A (active: leaf is asst-4a)
// │ asst-3a
// │ user-4a
// │ asst-4a
// └─ user-3b ← branch B
// asst-3b
// user-4b
//
// Foldable nodes: user-1 (root), user-3a (segment start), user-3b (segment start)
function buildBranchingTree() {
const entries: SessionEntry[] = [
userMessage("user-1", null, "first message"),
assistantMessage("asst-1", "user-1", "response 1"),
userMessage("user-2", "asst-1", "second message"),
assistantMessage("asst-2", "user-2", "response 2"),
// Branch A (active)
userMessage("user-3a", "asst-2", "branch A start"),
assistantMessage("asst-3a", "user-3a", "branch A response"),
userMessage("user-4a", "asst-3a", "branch A deep"),
assistantMessage("asst-4a", "user-4a", "branch A leaf"),
// Branch B
userMessage("user-3b", "asst-2", "branch B start"),
assistantMessage("asst-3b", "user-3b", "branch B response"),
userMessage("user-4b", "asst-3b", "branch B deep"),
];
return buildTree(entries);
}
test("ctrl+right unfolds a folded node, then does segment jump when unfolded", () => {
const tree = buildBranchingTree();
const selector = new TreeSelectorComponent(
tree,
"asst-4a",
24,
() => {},
() => {},
);
const list = selector.getTreeList();
selector.handleInput(CTRL_LEFT); // asst-4a → user-3a
expect(list.getSelectedNode()?.entry.id).toBe("user-3a");
selector.handleInput(CTRL_LEFT); // fold user-3a
expect(list.getSelectedNode()?.entry.id).toBe("user-3a");
selector.handleInput(DOWN); // user-3a → user-3b (children hidden)
expect(list.getSelectedNode()?.entry.id).toBe("user-3b");
selector.handleInput(UP); // user-3b → user-3a
expect(list.getSelectedNode()?.entry.id).toBe("user-3a");
selector.handleInput(CTRL_RIGHT); // unfold user-3a
expect(list.getSelectedNode()?.entry.id).toBe("user-3a");
selector.handleInput(DOWN); // user-3a → asst-3a (children restored)
expect(list.getSelectedNode()?.entry.id).toBe("asst-3a");
selector.handleInput(CTRL_LEFT); // asst-3a → user-3a
expect(list.getSelectedNode()?.entry.id).toBe("user-3a");
selector.handleInput(CTRL_RIGHT); // user-3a → asst-4a (segment jump to leaf)
expect(list.getSelectedNode()?.entry.id).toBe("asst-4a");
});
test("alt+left/right are aliases for fold and unfold navigation", () => {
const tree = buildBranchingTree();
const selector = new TreeSelectorComponent(
tree,
"asst-4a",
24,
() => {},
() => {},
);
const list = selector.getTreeList();
selector.handleInput(ALT_LEFT); // asst-4a → user-3a
expect(list.getSelectedNode()?.entry.id).toBe("user-3a");
selector.handleInput(ALT_LEFT); // fold user-3a
expect(list.getSelectedNode()?.entry.id).toBe("user-3a");
selector.handleInput(ALT_RIGHT); // unfold user-3a
expect(list.getSelectedNode()?.entry.id).toBe("user-3a");
selector.handleInput(ALT_RIGHT); // user-3a → asst-4a
expect(list.getSelectedNode()?.entry.id).toBe("asst-4a");
});
test("folding root hides entire subtree, nested fold preserved on unfold", () => {
const tree = buildBranchingTree();
const selector = new TreeSelectorComponent(
tree,
"asst-4a",
24,
() => {},
() => {},
);
const list = selector.getTreeList();
selector.handleInput(CTRL_LEFT); // asst-4a → user-3a
expect(list.getSelectedNode()?.entry.id).toBe("user-3a");
selector.handleInput(CTRL_LEFT); // fold user-3a
expect(list.getSelectedNode()?.entry.id).toBe("user-3a");
selector.handleInput(CTRL_LEFT); // user-3a (folded) → user-1
expect(list.getSelectedNode()?.entry.id).toBe("user-1");
selector.handleInput(CTRL_LEFT); // fold user-1
expect(list.getSelectedNode()?.entry.id).toBe("user-1");
selector.handleInput(DOWN); // wrap (only visible node)
expect(list.getSelectedNode()?.entry.id).toBe("user-1");
selector.handleInput(CTRL_RIGHT); // unfold user-1
expect(list.getSelectedNode()?.entry.id).toBe("user-1");
selector.handleInput(CTRL_RIGHT); // user-1 → user-3a (segment jump, user-3a still folded)
expect(list.getSelectedNode()?.entry.id).toBe("user-3a");
selector.handleInput(DOWN); // user-3a → user-3b (user-3a still folded)
expect(list.getSelectedNode()?.entry.id).toBe("user-3b");
});
test("fold and navigate on non-active branch", () => {
const tree = buildBranchingTree();
const selector = new TreeSelectorComponent(
tree,
"asst-4a",
24,
() => {},
() => {},
);
const list = selector.getTreeList();
// Navigate down to user-3b (branch B)
let found = false;
for (let i = 0; i < 20; i++) {
selector.handleInput(DOWN);
if (list.getSelectedNode()?.entry.id === "user-3b") {
found = true;
break;
}
}
expect(found).toBe(true);
selector.handleInput(CTRL_RIGHT); // user-3b → user-4b (segment jump to leaf)
expect(list.getSelectedNode()?.entry.id).toBe("user-4b");
selector.handleInput(CTRL_LEFT); // user-4b → user-3b
expect(list.getSelectedNode()?.entry.id).toBe("user-3b");
selector.handleInput(CTRL_LEFT); // fold user-3b
expect(list.getSelectedNode()?.entry.id).toBe("user-3b");
selector.handleInput(CTRL_LEFT); // user-3b (folded) → user-1
expect(list.getSelectedNode()?.entry.id).toBe("user-1");
});
test("fold and navigate with multiple roots", () => {
const entries: SessionEntry[] = [
userMessage("user-1", null, "first root"),
assistantMessage("asst-1", "user-1", "response 1"),
userMessage("user-2", null, "second root"),
assistantMessage("asst-2", "user-2", "response 2"),
];
const tree = buildTree(entries);
const selector = new TreeSelectorComponent(
tree,
"asst-1",
24,
() => {},
() => {},
);
const list = selector.getTreeList();
expect(list.getSelectedNode()?.entry.id).toBe("asst-1");
selector.handleInput(CTRL_LEFT); // asst-1 → user-1
expect(list.getSelectedNode()?.entry.id).toBe("user-1");
selector.handleInput(CTRL_LEFT); // fold user-1
expect(list.getSelectedNode()?.entry.id).toBe("user-1");
selector.handleInput(DOWN); // user-1 → user-2 (children hidden)
expect(list.getSelectedNode()?.entry.id).toBe("user-2");
selector.handleInput(CTRL_RIGHT); // user-2 → asst-2 (segment jump to leaf)
expect(list.getSelectedNode()?.entry.id).toBe("asst-2");
selector.handleInput(CTRL_LEFT); // asst-2 → user-2
expect(list.getSelectedNode()?.entry.id).toBe("user-2");
selector.handleInput(CTRL_LEFT); // fold user-2
expect(list.getSelectedNode()?.entry.id).toBe("user-2");
selector.handleInput(CTRL_LEFT); // user-2 (folded, root) → stays on user-2
expect(list.getSelectedNode()?.entry.id).toBe("user-2");
});
test("folding root hides descendants even when intermediate nodes are filtered out", () => {
// user-1 → toolCallOnly-1 (filtered out) → user-2 → asst-2
const entries: SessionEntry[] = [
userMessage("user-1", null, "hello"),
toolCallOnlyAssistant("tool-asst-1", "user-1"),
userMessage("user-2", "tool-asst-1", "follow up"),
assistantMessage("asst-2", "user-2", "response"),
];
const tree = buildTree(entries);
const selector = new TreeSelectorComponent(
tree,
"asst-2",
24,
() => {},
() => {},
);
const list = selector.getTreeList();
selector.handleInput(CTRL_LEFT); // asst-2 → user-1
expect(list.getSelectedNode()?.entry.id).toBe("user-1");
selector.handleInput(CTRL_LEFT); // fold user-1
expect(list.getSelectedNode()?.entry.id).toBe("user-1");
selector.handleInput(DOWN); // wrap (only visible node)
expect(list.getSelectedNode()?.entry.id).toBe("user-1");
});
test("search resets fold state", () => {
const tree = buildBranchingTree();
const selector = new TreeSelectorComponent(
tree,
"asst-4a",
24,
() => {},
() => {},
);
const list = selector.getTreeList();
selector.handleInput(CTRL_LEFT); // asst-4a → user-3a
selector.handleInput(CTRL_LEFT); // fold user-3a
selector.handleInput(DOWN); // user-3a → user-3b (children hidden)
expect(list.getSelectedNode()?.entry.id).toBe("user-3b");
selector.handleInput("b"); // search resets folds
selector.handleInput("\x1b"); // clear search
// Navigate to user-3a to verify fold was reset
let currentId = "";
for (let i = 0; i < 20; i++) {
selector.handleInput(DOWN);
currentId = list.getSelectedNode()?.entry.id ?? "";
if (currentId === "user-3a") break;
}
expect(currentId).toBe("user-3a");
selector.handleInput(DOWN); // user-3a → asst-3a (not user-3b)
expect(list.getSelectedNode()?.entry.id).toBe("asst-3a");
});
test("filter mode change resets fold state", () => {
const tree = buildBranchingTree();
const selector = new TreeSelectorComponent(
tree,
"asst-4a",
24,
() => {},
() => {},
);
const list = selector.getTreeList();
selector.handleInput(CTRL_LEFT); // asst-4a → user-3a
selector.handleInput(CTRL_LEFT); // fold user-3a
selector.handleInput("\x15"); // ctrl+u: user-only filter resets folds
selector.handleInput("\x04"); // ctrl+d: back to default
// Navigate to user-3a to verify fold was reset
let currentId = "";
for (let i = 0; i < 20; i++) {
selector.handleInput(DOWN);
currentId = list.getSelectedNode()?.entry.id ?? "";
if (currentId === "user-3a") break;
}
expect(currentId).toBe("user-3a");
selector.handleInput(DOWN); // user-3a → asst-3a (not user-3b)
expect(list.getSelectedNode()?.entry.id).toBe("asst-3a");
});
});
});