@aaarc/handfree-ssh-mcp 1.0.0 → 1.0.2

package/build/tests/tools.test.js

@@ -1,486 +0,0 @@
-/**
- * Tools Unit Tests
- *
- * Tests for list-servers, upload, download, show-whitelist tools
- * Uses mock SSHConnectionManager to avoid actual SSH connections.
- */
-import { describe, it, beforeEach } from "node:test";
-import assert from "node:assert";
-/**
- * Mock SSHConnectionManager for testing tools without real SSH connections
- */
-class MockSSHConnectionManager {
-    configs = {};
-    enabledServers = [];
-    defaultName = "default";
-    setConfig(configs, enabledServers) {
-        this.configs = configs;
-        this.enabledServers = enabledServers || Object.keys(configs);
-        if (this.enabledServers.length > 0) {
-            this.defaultName = this.enabledServers[0];
-        }
-    }
-    getServerConfig(name) {
-        const serverName = name || this.defaultName;
-        if (!this.enabledServers.includes(serverName)) {
-            return null;
-        }
-        return this.configs[serverName] || null;
-    }
-    getAllServerInfos() {
-        return this.enabledServers.map(name => {
-            const config = this.configs[name];
-            return {
-                name,
-                host: config?.host || "unknown",
-                port: config?.port || 22,
-                username: config?.username || "unknown",
-                connected: false,
-            };
-        });
-    }
-    async upload(localPath, remotePath, connectionName) {
-        const serverName = connectionName || this.defaultName;
-        if (!this.enabledServers.includes(serverName)) {
-            throw new Error(`Server '${serverName}' not enabled`);
-        }
-        // Mock implementation - just return success message
-        return `Uploaded ${localPath} to ${remotePath} on ${serverName}`;
-    }
-    async download(remotePath, localPath, connectionName) {
-        const serverName = connectionName || this.defaultName;
-        if (!this.enabledServers.includes(serverName)) {
-            throw new Error(`Server '${serverName}' not enabled`);
-        }
-        // Mock implementation - just return success message
-        return `Downloaded ${remotePath} to ${localPath} from ${serverName}`;
-    }
-    async executeCommand(command, connectionName, options) {
-        const serverName = connectionName || this.defaultName;
-        if (!this.enabledServers.includes(serverName)) {
-            throw new Error(`Server '${serverName}' not enabled`);
-        }
-        // Mock implementation - simulate command execution
-        if (command === "pwd")
-            return "/home/testuser";
-        if (command.startsWith("ls"))
-            return "file1.txt\nfile2.txt\ndir1";
-        if (command === "hostname")
-            return "test-server";
-        if (command === "whoami")
-            return "testuser";
-        if (command === "date")
-            return "Sat Jan 24 12:00:00 UTC 2026";
-        return `Executed: ${command}`;
-    }
-    async executeCommandWithProgress(command, connectionName, options) {
-        const serverName = connectionName || this.defaultName;
-        if (!this.enabledServers.includes(serverName)) {
-            throw new Error(`Server '${serverName}' not enabled`);
-        }
-        // Simulate streaming output
-        if (options?.onProgress) {
-            options.onProgress("line 1\n");
-            options.onProgress("line 2\n");
-            options.onProgress("line 3\n");
-        }
-        return "line 1\nline 2\nline 3\n";
-    }
-    disconnect() {
-        // No-op for mock
-    }
-}
-// ============================================
-// list-servers Tool Tests
-// ============================================
-describe("list-servers Tool", () => {
-    let mockManager;
-    beforeEach(() => {
-        mockManager = new MockSSHConnectionManager();
-    });
-    it("should return empty array when no servers configured", () => {
-        mockManager.setConfig({}, []);
-        const result = mockManager.getAllServerInfos();
-        assert.deepStrictEqual(result, []);
-    });
-    it("should return single server info", () => {
-        mockManager.setConfig({
-            dev: {
-                host: "192.168.1.1",
-                port: 22,
-                username: "testuser",
-                password: "testpass",
-            }
-        }, ["dev"]);
-        const result = mockManager.getAllServerInfos();
-        assert.strictEqual(result.length, 1);
-        assert.strictEqual(result[0].name, "dev");
-        assert.strictEqual(result[0].host, "192.168.1.1");
-        assert.strictEqual(result[0].port, 22);
-        assert.strictEqual(result[0].username, "testuser");
-        assert.strictEqual(result[0].connected, false);
-    });
-    it("should return multiple servers info", () => {
-        mockManager.setConfig({
-            dev: { host: "10.0.0.1", port: 22, username: "dev" },
-            prod: { host: "10.0.0.2", port: 2222, username: "prod" },
-            staging: { host: "10.0.0.3", port: 22, username: "staging" },
-        }, ["dev", "prod", "staging"]);
-        const result = mockManager.getAllServerInfos();
-        assert.strictEqual(result.length, 3);
-        assert.strictEqual(result[0].name, "dev");
-        assert.strictEqual(result[1].name, "prod");
-        assert.strictEqual(result[2].name, "staging");
-        assert.strictEqual(result[1].port, 2222);
-    });
-    it("should only return enabled servers", () => {
-        mockManager.setConfig({
-            dev: { host: "10.0.0.1", port: 22, username: "dev" },
-            prod: { host: "10.0.0.2", port: 22, username: "prod" },
-            staging: { host: "10.0.0.3", port: 22, username: "staging" },
-        }, ["dev", "staging"]); // prod not enabled
-        const result = mockManager.getAllServerInfos();
-        assert.strictEqual(result.length, 2);
-        const names = result.map(s => s.name);
-        assert.ok(names.includes("dev"));
-        assert.ok(names.includes("staging"));
-        assert.ok(!names.includes("prod"));
-    });
-});
-// ============================================
-// show-whitelist Tool Tests
-// ============================================
-describe("show-whitelist Tool", () => {
-    let mockManager;
-    beforeEach(() => {
-        mockManager = new MockSSHConnectionManager();
-    });
-    it("should return null for non-existent server", () => {
-        mockManager.setConfig({
-            dev: { host: "192.168.1.1", username: "test", password: "test" }
-        }, ["dev"]);
-        const result = mockManager.getServerConfig("nonexistent");
-        assert.strictEqual(result, null);
-    });
-    it("should return config for valid server", () => {
-        const devConfig = {
-            host: "192.168.1.1",
-            port: 22,
-            username: "testuser",
-            password: "testpass",
-            commandWhitelist: ["^ls.*$", "^pwd$"],
-        };
-        mockManager.setConfig({ dev: devConfig }, ["dev"]);
-        const result = mockManager.getServerConfig("dev");
-        assert.ok(result);
-        assert.strictEqual(result.host, "192.168.1.1");
-        assert.deepStrictEqual(result.commandWhitelist, ["^ls.*$", "^pwd$"]);
-    });
-    it("should return default server config when name not specified", () => {
-        const devConfig = {
-            host: "192.168.1.1",
-            username: "dev",
-            password: "test",
-        };
-        mockManager.setConfig({ dev: devConfig }, ["dev"]);
-        const result = mockManager.getServerConfig(); // No name - uses default
-        assert.ok(result);
-        assert.strictEqual(result.host, "192.168.1.1");
-    });
-    it("should return null for disabled server", () => {
-        mockManager.setConfig({
-            dev: { host: "10.0.0.1", username: "dev", password: "test" },
-            prod: { host: "10.0.0.2", username: "prod", password: "test" },
-        }, ["dev"]); // Only dev enabled
-        const result = mockManager.getServerConfig("prod");
-        assert.strictEqual(result, null);
-    });
-    it("should return config with blacklist", () => {
-        const devConfig = {
-            host: "192.168.1.1",
-            username: "test",
-            password: "test",
-            commandWhitelist: ["^.*$"],
-            commandBlacklist: ["^rm.*$", "^shutdown.*$"],
-        };
-        mockManager.setConfig({ dev: devConfig }, ["dev"]);
-        const result = mockManager.getServerConfig("dev");
-        assert.ok(result);
-        assert.deepStrictEqual(result.commandBlacklist, ["^rm.*$", "^shutdown.*$"]);
-    });
-});
-// ============================================
-// upload Tool Tests
-// ============================================
-describe("upload Tool", () => {
-    let mockManager;
-    beforeEach(() => {
-        mockManager = new MockSSHConnectionManager();
-        mockManager.setConfig({
-            dev: { host: "192.168.1.1", username: "test", password: "test" }
-        }, ["dev"]);
-    });
-    it("should upload file to default server", async () => {
-        const result = await mockManager.upload("/local/file.txt", "/remote/file.txt");
-        assert.ok(result.includes("Uploaded"));
-        assert.ok(result.includes("/local/file.txt"));
-        assert.ok(result.includes("/remote/file.txt"));
-    });
-    it("should upload file to specified server", async () => {
-        const result = await mockManager.upload("/local/file.txt", "/remote/file.txt", "dev");
-        assert.ok(result.includes("dev"));
-    });
-    it("should throw error for disabled server", async () => {
-        mockManager.setConfig({
-            dev: { host: "192.168.1.1", username: "test", password: "test" },
-            prod: { host: "192.168.1.2", username: "test", password: "test" },
-        }, ["dev"]); // Only dev enabled
-        await assert.rejects(() => mockManager.upload("/local/file.txt", "/remote/file.txt", "prod"), /not enabled/);
-    });
-});
-// ============================================
-// download Tool Tests
-// ============================================
-describe("download Tool", () => {
-    let mockManager;
-    beforeEach(() => {
-        mockManager = new MockSSHConnectionManager();
-        mockManager.setConfig({
-            dev: { host: "192.168.1.1", username: "test", password: "test" }
-        }, ["dev"]);
-    });
-    it("should download file from default server", async () => {
-        const result = await mockManager.download("/remote/file.txt", "/local/file.txt");
-        assert.ok(result.includes("Downloaded"));
-        assert.ok(result.includes("/remote/file.txt"));
-        assert.ok(result.includes("/local/file.txt"));
-    });
-    it("should download file from specified server", async () => {
-        const result = await mockManager.download("/remote/file.txt", "/local/file.txt", "dev");
-        assert.ok(result.includes("dev"));
-    });
-    it("should throw error for disabled server", async () => {
-        mockManager.setConfig({
-            dev: { host: "192.168.1.1", username: "test", password: "test" },
-            prod: { host: "192.168.1.2", username: "test", password: "test" },
-        }, ["dev"]); // Only dev enabled
-        await assert.rejects(() => mockManager.download("/remote/file.txt", "/local/file.txt", "prod"), /not enabled/);
-    });
-});
-// ============================================
-// execute-command Streaming Mode Tests
-// ============================================
-describe("execute-command Streaming Mode", () => {
-    let mockManager;
-    beforeEach(() => {
-        mockManager = new MockSSHConnectionManager();
-        mockManager.setConfig({
-            dev: { host: "192.168.1.1", username: "test", password: "test" }
-        }, ["dev"]);
-    });
-    it("should execute command without streaming", async () => {
-        const result = await mockManager.executeCommand("pwd", "dev");
-        assert.strictEqual(result, "/home/testuser");
-    });
-    it("should execute command with streaming and receive chunks", async () => {
-        const chunks = [];
-        const onProgress = (chunk) => chunks.push(chunk);
-        const result = await mockManager.executeCommandWithProgress("ls -la", "dev", {
-            onProgress,
-        });
-        assert.ok(chunks.length > 0);
-        assert.ok(result.includes("line 1"));
-        assert.ok(result.includes("line 2"));
-        assert.ok(result.includes("line 3"));
-    });
-    it("should handle custom timeout", async () => {
-        const result = await mockManager.executeCommandWithProgress("long-command", "dev", {
-            timeout: 60000,
-        });
-        assert.ok(result);
-    });
-    it("should throw error for disabled server in streaming mode", async () => {
-        mockManager.setConfig({
-            dev: { host: "192.168.1.1", username: "test", password: "test" },
-            prod: { host: "192.168.1.2", username: "test", password: "test" },
-        }, ["dev"]);
-        await assert.rejects(() => mockManager.executeCommandWithProgress("pwd", "prod"), /not enabled/);
-    });
-});
-// ============================================
-// Timeout Behavior Tests (Logic Simulation)
-// ============================================
-/**
- * Helper to calculate effective timeout based on stream mode
- * Matches the logic in execute-command.ts
- */
-function calculateTimeout(stream, timeout) {
-    const useStream = stream !== false;
-    if (useStream) {
-        return timeout || 300000; // 5 min default for streaming
-    }
-    else {
-        return timeout || 30000; // 30s default for non-streaming
-    }
-}
-describe("Timeout Behavior", () => {
-    it("should use default 30s timeout for non-streaming mode", () => {
-        const effectiveTimeout = calculateTimeout(false, undefined);
-        assert.strictEqual(effectiveTimeout, 30000);
-    });
-    it("should use default 300s timeout for streaming mode", () => {
-        const effectiveTimeout = calculateTimeout(true, undefined);
-        assert.strictEqual(effectiveTimeout, 300000);
-    });
-    it("should use custom timeout when specified in streaming mode", () => {
-        const effectiveTimeout = calculateTimeout(true, 60000);
-        assert.strictEqual(effectiveTimeout, 60000);
-    });
-    it("should use custom timeout when specified in non-streaming mode", () => {
-        const effectiveTimeout = calculateTimeout(false, 5000);
-        assert.strictEqual(effectiveTimeout, 5000);
-    });
-    it("should default stream to true when not specified (undefined)", () => {
-        const effectiveTimeout = calculateTimeout(undefined, undefined);
-        // undefined !== false is true, so streaming mode is used
-        assert.strictEqual(effectiveTimeout, 300000);
-    });
-});
-// ============================================
-// Pattern Helper Function Tests (show-whitelist)
-// ============================================
-describe("patternToReadable (show-whitelist helper)", () => {
-    function patternToReadable(pattern) {
-        let readable = pattern;
-        readable = readable.replace(/^\^/, "").replace(/\$$/, "");
-        if (readable === "ls( .*)?")
-            return "ls, ls -la, ls /path, etc.";
-        if (readable === "cat .*")
-            return "cat <file>";
-        if (readable === "pwd")
-            return "pwd (current directory)";
-        if (readable === "whoami")
-            return "whoami (current user)";
-        if (readable === "hostname")
-            return "hostname";
-        if (readable === "date")
-            return "date (current time)";
-        if (readable === "docker ps.*")
-            return "docker ps, docker ps -a, etc.";
-        if (readable === "docker logs.*")
-            return "docker logs <container>";
-        if (readable === "git .*")
-            return "git <any subcommand>";
-        if (readable.includes(".*"))
-            return readable.replace(/\.\*/g, "<any>");
-        if (readable.includes(".+"))
-            return readable.replace(/\.\+/g, "<required>");
-        return readable;
-    }
-    it("should convert ls pattern to readable", () => {
-        assert.strictEqual(patternToReadable("^ls( .*)?$"), "ls, ls -la, ls /path, etc.");
-    });
-    it("should convert cat pattern to readable", () => {
-        assert.strictEqual(patternToReadable("^cat .*$"), "cat <file>");
-    });
-    it("should convert pwd pattern to readable", () => {
-        assert.strictEqual(patternToReadable("^pwd$"), "pwd (current directory)");
-    });
-    it("should convert docker ps pattern to readable", () => {
-        assert.strictEqual(patternToReadable("^docker ps.*$"), "docker ps, docker ps -a, etc.");
-    });
-    it("should convert git pattern to readable", () => {
-        assert.strictEqual(patternToReadable("^git .*$"), "git <any subcommand>");
-    });
-    it("should replace .* with <any> for unknown patterns", () => {
-        assert.strictEqual(patternToReadable("^custom .*$"), "custom <any>");
-    });
-    it("should replace .+ with <required> for unknown patterns", () => {
-        assert.strictEqual(patternToReadable("^custom .+$"), "custom <required>");
-    });
-});
-// ============================================
-// Example Generation Tests (show-whitelist)
-// ============================================
-describe("generateExamples (show-whitelist helper)", () => {
-    function generateExamples(whitelist) {
-        const examples = [];
-        for (const pattern of whitelist) {
-            if (pattern.includes("^ls"))
-                examples.push("ls -la");
-            if (pattern.includes("^cat"))
-                examples.push("cat /etc/hostname");
-            if (pattern.includes("^pwd"))
-                examples.push("pwd");
-            if (pattern.includes("^whoami"))
-                examples.push("whoami");
-            if (pattern.includes("^hostname"))
-                examples.push("hostname");
-            if (pattern.includes("^date"))
-                examples.push("date");
-            if (pattern.includes("^echo"))
-                examples.push("echo 'hello world'");
-            if (pattern.includes("^docker ps"))
-                examples.push("docker ps -a");
-            if (pattern.includes("^docker logs"))
-                examples.push("docker logs --tail 50 <container>");
-            if (pattern.includes("^git"))
-                examples.push("git status");
-        }
-        return [...new Set(examples)];
-    }
-    it("should generate ls example", () => {
-        const examples = generateExamples(["^ls( .*)?$"]);
-        assert.ok(examples.includes("ls -la"));
-    });
-    it("should generate docker examples", () => {
-        const examples = generateExamples(["^docker ps.*$", "^docker logs.*$"]);
-        assert.ok(examples.includes("docker ps -a"));
-        assert.ok(examples.includes("docker logs --tail 50 <container>"));
-    });
-    it("should generate multiple examples for mixed whitelist", () => {
-        const examples = generateExamples([
-            "^ls( .*)?$",
-            "^cat .*$",
-            "^pwd$",
-            "^git .*$",
-        ]);
-        assert.ok(examples.includes("ls -la"));
-        assert.ok(examples.includes("cat /etc/hostname"));
-        assert.ok(examples.includes("pwd"));
-        assert.ok(examples.includes("git status"));
-    });
-    it("should remove duplicates", () => {
-        const examples = generateExamples(["^ls.*$", "^ls -la$"]); // Both match ls
-        const lsCount = examples.filter(e => e === "ls -la").length;
-        assert.strictEqual(lsCount, 1);
-    });
-});
-// ============================================
-// Server Selection Tests
-// ============================================
-describe("Server Selection Logic", () => {
-    let mockManager;
-    beforeEach(() => {
-        mockManager = new MockSSHConnectionManager();
-    });
-    it("should use first enabled server as default", () => {
-        mockManager.setConfig({
-            alpha: { host: "10.0.0.1", username: "a", password: "a" },
-            beta: { host: "10.0.0.2", username: "b", password: "b" },
-        }, ["beta", "alpha"]); // beta is first
-        const config = mockManager.getServerConfig();
-        assert.ok(config);
-        assert.strictEqual(config.host, "10.0.0.2"); // beta's host
-    });
-    it("should use explicit server name over default", () => {
-        mockManager.setConfig({
-            dev: { host: "10.0.0.1", username: "dev", password: "dev" },
-            prod: { host: "10.0.0.2", username: "prod", password: "prod" },
-        }, ["dev", "prod"]);
-        const config = mockManager.getServerConfig("prod");
-        assert.ok(config);
-        assert.strictEqual(config.host, "10.0.0.2");
-    });
-});
-console.log("\n🧪 Running tools unit tests...\n");