Something went wrong. Try again.
fork of prime agent with changes i want but are not upstreamable probqbly (and some fixes)
Something went wrong. Try again.
5.8 kB · 144 lines
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145import { describe, expect, it, vi } from "vitest";import { SessionImportFileNotFoundError } from "../src/core/session-import-errors.js";import { InteractiveMode } from "../src/modes/interactive/interactive-mode.js";
type PathCommand = "/export" | "/import";
type InteractiveModePrototype = { getPathCommandArgument(this: unknown, text: string, command: PathCommand): string | undefined; handleImportCommand(this: ImportCommandContext, text: string): Promise<void>;};
type ImportCommandContext = { loadingAnimation?: { stop: () => void }; statusContainer: { clear: () => void }; stopWorkingLoader: () => void; agentConnection: { importFromJsonl: (inputPath: string, cwdOverride?: string) => Promise<{ cancelled: boolean }> }; showError: (message: string) => void; showStatus: (message: string) => void; showExtensionConfirm: (title: string, message: string) => Promise<boolean>; renderCurrentSessionState: () => void; handleFatalRuntimeError: (prefix: string, error: unknown) => Promise<never>; promptForMissingSessionCwd: (error: unknown) => Promise<string | undefined>; getPathCommandArgument: (text: string, command: PathCommand) => string | undefined;};
const interactiveModePrototype = InteractiveMode.prototype as unknown as InteractiveModePrototype;
describe("InteractiveMode /import parsing", () => { it("strips quotes from /import path arguments", () => { expect(interactiveModePrototype.getPathCommandArgument('/import "path/to/session.jsonl"', "/import")).toBe( "path/to/session.jsonl", ); expect( interactiveModePrototype.getPathCommandArgument('/import "path with spaces/session.jsonl"', "/import"), ).toBe("path with spaces/session.jsonl"); });
it("preserves apostrophes in unquoted /import path arguments", () => { expect(interactiveModePrototype.getPathCommandArgument("/import john's/session.jsonl", "/import")).toBe( "john's/session.jsonl", ); });
it("enforces command token boundaries", () => { expect(interactiveModePrototype.getPathCommandArgument("/important /tmp/session.jsonl", "/import")).toBe( undefined, ); expect(interactiveModePrototype.getPathCommandArgument("/exporter out.html", "/export")).toBe(undefined); expect(interactiveModePrototype.getPathCommandArgument("/import /tmp/session.jsonl", "/import")).toBe( "/tmp/session.jsonl", ); });
it("passes unquoted path to agentConnection.importFromJsonl", async () => { const importFromJsonl = vi.fn(async () => ({ cancelled: false })); const showExtensionConfirm = vi.fn(async () => true); const showStatus = vi.fn(); const showError = vi.fn();
const context: ImportCommandContext = { statusContainer: { clear: vi.fn() }, stopWorkingLoader: vi.fn(), agentConnection: { importFromJsonl }, showError, showStatus, showExtensionConfirm, renderCurrentSessionState: vi.fn(), handleFatalRuntimeError: vi.fn(async () => { throw new Error("unexpected fatal error"); }), promptForMissingSessionCwd: vi.fn(async () => undefined), getPathCommandArgument: interactiveModePrototype.getPathCommandArgument, };
await interactiveModePrototype.handleImportCommand.call(context, '/import "path/to/session.jsonl"');
expect(showExtensionConfirm).toHaveBeenCalledWith( "Import session", "Replace current session with path/to/session.jsonl?", ); expect(importFromJsonl).toHaveBeenCalledWith("path/to/session.jsonl"); expect(showError).not.toHaveBeenCalled(); expect(showStatus).toHaveBeenCalledWith("Session imported from: path/to/session.jsonl"); });
it("passes unquoted apostrophe path to agentConnection.importFromJsonl unchanged", async () => { const importFromJsonl = vi.fn(async () => ({ cancelled: false })); const showExtensionConfirm = vi.fn(async () => true); const showStatus = vi.fn(); const showError = vi.fn();
const context: ImportCommandContext = { statusContainer: { clear: vi.fn() }, stopWorkingLoader: vi.fn(), agentConnection: { importFromJsonl }, showError, showStatus, showExtensionConfirm, renderCurrentSessionState: vi.fn(), handleFatalRuntimeError: vi.fn(async () => { throw new Error("unexpected fatal error"); }), promptForMissingSessionCwd: vi.fn(async () => undefined), getPathCommandArgument: interactiveModePrototype.getPathCommandArgument, };
await interactiveModePrototype.handleImportCommand.call(context, "/import john's/session.jsonl");
expect(importFromJsonl).toHaveBeenCalledWith("john's/session.jsonl"); expect(showError).not.toHaveBeenCalled(); expect(showStatus).toHaveBeenCalledWith("Session imported from: john's/session.jsonl"); });
it("shows a non-fatal error when /import path does not exist", async () => { const importFromJsonl = vi.fn(async () => { throw new SessionImportFileNotFoundError("/tmp/missing-session.jsonl"); }); const showExtensionConfirm = vi.fn(async () => true); const showStatus = vi.fn(); const showError = vi.fn(); const handleFatalRuntimeError = vi.fn(async () => { throw new Error("unexpected fatal error"); });
const context: ImportCommandContext = { statusContainer: { clear: vi.fn() }, stopWorkingLoader: vi.fn(), agentConnection: { importFromJsonl }, showError, showStatus, showExtensionConfirm, renderCurrentSessionState: vi.fn(), handleFatalRuntimeError, promptForMissingSessionCwd: vi.fn(async () => undefined), getPathCommandArgument: interactiveModePrototype.getPathCommandArgument, };
await interactiveModePrototype.handleImportCommand.call(context, "/import /tmp/missing-session.jsonl");
expect(showError).toHaveBeenCalledWith("Failed to import session: File not found: /tmp/missing-session.jsonl"); expect(showStatus).not.toHaveBeenCalled(); expect(handleFatalRuntimeError).not.toHaveBeenCalled(); });});