改善活動看板手機導覽排版

tests/test_ssh_helper.py

@@ -0,0 +1,111 @@
+from utils.ssh_helper import SshExecResult, build_ssh_command
+
+
+def test_build_ssh_command_supports_jump_host_and_argv_command():
+    argv = build_ssh_command(
+        host="192.168.0.188",
+        user="ollama",
+        command=["docker", "ps"],
+        port=22,
+        key_path="/tmp/autoheal.key",
+        connect_timeout=10,
+        jump_host="192.168.0.110",
+        jump_user="wooo",
+        batch_mode=True,
+        server_alive_interval=15,
+        server_alive_count_max=3,
+    )
+
+    assert argv == [
+        "ssh",
+        "-p",
+        "22",
+        "-i",
+        "/tmp/autoheal.key",
+        "-o",
+        "StrictHostKeyChecking=no",
+        "-o",
+        "BatchMode=yes",
+        "-o",
+        "ConnectTimeout=10",
+        "-o",
+        "ServerAliveInterval=15",
+        "-o",
+        "ServerAliveCountMax=3",
+        "-J",
+        "wooo@192.168.0.110",
+        "ollama@192.168.0.188",
+        "--",
+        "docker",
+        "ps",
+    ]
+
+
+def test_build_ssh_command_preserves_direct_shell_command():
+    argv = build_ssh_command(
+        host="192.168.0.110",
+        user="wooo",
+        command="cd /home/wooo/ewoooc && git status",
+        port=2222,
+        key_path="/tmp/heal.key",
+        connect_timeout=5,
+    )
+
+    assert "--" not in argv
+    assert argv[-2:] == [
+        "wooo@192.168.0.110",
+        "cd /home/wooo/ewoooc && git status",
+    ]
+
+
+def test_auto_heal_ssh_wrapper_keeps_legacy_dict_contract(monkeypatch):
+    from services import auto_heal_service
+
+    captured = {}
+
+    def fake_run_ssh_command(**kwargs):
+        captured.update(kwargs)
+        return SshExecResult(0, "ok", "", ["ssh"])
+
+    monkeypatch.setattr(auto_heal_service, "run_ssh_command", fake_run_ssh_command)
+
+    result = auto_heal_service._ssh_exec(
+        jump_host="192.168.0.110",
+        jump_user="wooo",
+        target_host="192.168.0.188",
+        target_user="ollama",
+        command=["df", "-h"],
+        key_path="/tmp/autoheal.key",
+    )
+
+    assert result == {
+        "success": True,
+        "exit_code": 0,
+        "stdout": "ok",
+        "stderr": "",
+        "command": ["df", "-h"],
+    }
+    assert captured["jump_host"] == "192.168.0.110"
+    assert captured["jump_user"] == "wooo"
+    assert captured["batch_mode"] is True
+    assert captured["server_alive_interval"] == 15
+
+
+def test_aider_heal_ssh_wrapper_keeps_tuple_contract(monkeypatch):
+    from services import aider_heal_executor
+
+    captured = {}
+
+    def fake_run_ssh_command(**kwargs):
+        captured.update(kwargs)
+        return SshExecResult(0, "ready", "", ["ssh"])
+
+    monkeypatch.setattr(aider_heal_executor, "run_ssh_command", fake_run_ssh_command)
+
+    rc, out, err = aider_heal_executor._ssh_exec("echo ready", timeout=5)
+
+    assert (rc, out, err) == (0, "ready", "")
+    assert captured["host"] == aider_heal_executor.HEAL_SSH_HOST
+    assert captured["user"] == aider_heal_executor.HEAL_SSH_USER
+    assert captured["command"] == "echo ready"
+    assert captured["command_timeout"] == 5