Enhance logging functionality in CLI and tunnel operations. Updated gpu-rent logs to provide a complete journal output, including cloud-init logs by default. Added log digest printing for better visibility during VM operations, especially on failure scenarios. Improved error handling and added tests for new log digest features.

tests/test_vm_logs.py

@@ -0,0 +1,93 @@
+"""Unit tests for vm_logs digest / CLI log command builders."""
+
+from __future__ import annotations
+
+import pytest
+
+from gpu_rent.vm_logs import (
+    DIGEST_LINES,
+    KILLER_UNIT,
+    build_logs_remote_cmd,
+    fetch_unit_logs,
+    resolve_unit_alias,
+    units_for,
+)
+
+
+class _Cfg:
+    def __init__(self, *, enable_swarmui: bool = True, llm_runtime: str = "none"):
+        self.enable_swarmui = enable_swarmui
+        self.llm_runtime = llm_runtime
+
+
+def test_units_for_swarm_and_ollama():
+    assert units_for(_Cfg(enable_swarmui=True, llm_runtime="ollama")) == [
+        "swarmui",
+        "ollama",
+        KILLER_UNIT,
+    ]
+
+
+def test_units_for_swarm_only():
+    assert units_for(_Cfg(enable_swarmui=True, llm_runtime="none")) == [
+        "swarmui",
+        KILLER_UNIT,
+    ]
+
+
+def test_units_for_llm_only():
+    assert units_for(_Cfg(enable_swarmui=False, llm_runtime="ollama")) == [
+        "ollama",
+        KILLER_UNIT,
+    ]
+
+
+def test_digest_cmd_has_journal_no_cloud_init():
+    units = units_for(_Cfg(enable_swarmui=True, llm_runtime="ollama"))
+    cmd = build_logs_remote_cmd(
+        lines=DIGEST_LINES,
+        include_cloud_init=False,
+        journal_units=units,
+    )
+    assert "cloud-init" not in cmd
+    assert "journalctl -u swarmui" in cmd
+    assert "journalctl -u ollama" in cmd
+    assert f"journalctl -u {KILLER_UNIT}" in cmd
+    assert f"-n {DIGEST_LINES}" in cmd
+    assert "systemctl is-active swarmui" in cmd
+
+
+def test_full_logs_cmd_includes_cloud_init():
+    cmd = build_logs_remote_cmd(
+        lines=80,
+        include_cloud_init=True,
+        journal_units=["swarmui", "ollama", KILLER_UNIT],
+    )
+    assert "cloud-init-output.log" in cmd
+    assert "journalctl -u swarmui" in cmd
+
+
+def test_resolve_unit_alias():
+    assert resolve_unit_alias(None) == "all"
+    assert resolve_unit_alias("swarm") == "swarmui"
+    assert resolve_unit_alias("killer") == KILLER_UNIT
+    with pytest.raises(ValueError, match="неизвестный"):
+        resolve_unit_alias("bogus")
+
+
+def test_fetch_unit_logs_uses_ssh(monkeypatch):
+    calls: list[tuple] = []
+
+    def fake_ssh(cfg, host, command, check=False, timeout=60):
+        calls.append((host, command, check, timeout))
+        return "active\nok"
+
+    monkeypatch.setattr("gpu_rent.vm_logs.run_ssh", fake_ssh)
+    cfg = _Cfg(enable_swarmui=True, llm_runtime="none")
+    out = fetch_unit_logs(cfg, "1.2.3.4", units_for(cfg), lines=20)
+    assert out == "active\nok"
+    assert calls[0][0] == "1.2.3.4"
+    assert "cloud-init" not in calls[0][1]
+    assert "journalctl -u swarmui" in calls[0][1]
+    assert f"journalctl -u {KILLER_UNIT}" in calls[0][1]
+    assert "journalctl -u ollama" not in calls[0][1]