@krishivpb60/aether-ai-cli 1.1.6 → 1.1.7

package/test/agent.test.js

@@ -0,0 +1,62 @@
+import { test } from "node:test";
+import assert from "node:assert";
+import { resolve, join } from "node:path";
+import {
+  isSafeCommand,
+  isInsideWorkspace,
+  processAgentBlocks
+} from "../src/agent.js";
+
+test("Agent Autopilot Engine Suite", async (t) => {
+  await t.test("isSafeCommand classifies inspection commands as safe", () => {
+    assert.strictEqual(isSafeCommand("git status"), true);
+    assert.strictEqual(isSafeCommand("git diff"), true);
+    assert.strictEqual(isSafeCommand("ls"), true);
+    assert.strictEqual(isSafeCommand("dir -a"), true);
+    assert.strictEqual(isSafeCommand("npm test"), true);
+    assert.strictEqual(isSafeCommand("node -v"), true);
+  });
+
+  await t.test("isSafeCommand classifies destructive/unsafe commands as unsafe", () => {
+    assert.strictEqual(isSafeCommand("rm -rf src"), false);
+    assert.strictEqual(isSafeCommand("del /f /q *"), false);
+    assert.strictEqual(isSafeCommand("format c:"), false);
+    assert.strictEqual(isSafeCommand("npm run dev"), false);
+  });
+
+  await t.test("isInsideWorkspace verifies if path is inside current workspace", () => {
+    const ws = resolve(process.cwd());
+    assert.strictEqual(isInsideWorkspace("src/config.js"), true);
+    assert.strictEqual(isInsideWorkspace(join(ws, "package.json")), true);
+    assert.strictEqual(isInsideWorkspace("../../somefile.txt"), false);
+  });
+
+  await t.test("processAgentBlocks extracts and processes tools sequentially", async () => {
+    const aiOutput = `
+I will read a file and run a command.
+[READ_FILE: src/git.js]
+And also run:
+[RUN_COMMAND: node -v]
+    `;
+
+    const aiConfig = { AUTOPILOT: "safe" };
+    // Mock readline interface
+    const rl = {
+      pause: () => {},
+      resume: () => {},
+      question: (query, cb) => cb("y")
+    };
+
+    const results = await processAgentBlocks(aiOutput, aiConfig, rl);
+    assert.strictEqual(results.length, 2);
+    
+    // First tool result should be READ_FILE
+    assert.strictEqual(results[0].tool, "READ_FILE");
+    assert.strictEqual(results[0].arg, "src/git.js");
+    
+    // Second tool result should be RUN_COMMAND
+    assert.strictEqual(results[1].tool, "RUN_COMMAND");
+    assert.strictEqual(results[1].arg, "node -v");
+    assert.strictEqual(results[1].success, true);
+  });
+});

package/test/config.test.js

@@ -25,6 +25,9 @@ const {
   loadHistory,
   saveHistory,
   clearHistory,
+  listSessions,
+  switchSession,
+  startNewSession,
 } = await import("../src/config.js");
 
 const { getAllConfigKeys } = await import("../src/ai/providers.js");
@@ -170,6 +173,30 @@ test("Configuration Loading Suite", async (t) => {
     assert.deepStrictEqual(cleared, []);
   });
 
+  await t.test("listSessions, switchSession, and startNewSession handle multi-session histories", async () => {
+    // 1. Start new session
+    const file1 = startNewSession();
+    await saveHistory([{ role: "user", content: "session 1" }], "research");
+
+    // 2. Start another new session
+    const file2 = startNewSession();
+    await saveHistory([{ role: "user", content: "session 2" }], "architect");
+
+    // 3. List sessions
+    const sessions = listSessions();
+    assert.ok(sessions.length >= 2);
+    // The most recent session (file2) should be first
+    assert.strictEqual(sessions[0].file, file2);
+    assert.strictEqual(sessions[0].mode, "architect");
+    assert.strictEqual(sessions[1].file, file1);
+    assert.strictEqual(sessions[1].mode, "research");
+
+    // 4. Switch session back to file1
+    switchSession(file1);
+    const loaded = await loadHistory();
+    assert.strictEqual(loaded[0].content, "session 1");
+  });
+
   await t.test("getAIConfig supports and loads CUSTOM_COMMANDS correctly", async () => {
     const testCommands = { "/explain": "Explain this:", "/refactor": "Refactor this:" };
     await saveConfig({