jessey
4bd51a40c3
Expand diagnostic/report collection (all stored per-diagnostic, in the Report zip; logs also fed to the AI on "Explain"): - syslogs: nvidia-smi -q snapshot (driver/throttle/clocks/power/temps/PCIe/ECC/ retired pages) + display-server log auto-detected — Xorg.0.log on X11, or the compositor user-journal slice (gnome-shell/kwin/sway/gamescope) on Wayland. - diagstore: include the full M5 inventory (inventory.txt + .json) — invaluable for larger/shared debugging. inventory.collect() degrades gracefully (no root prompt). Best-effort throughout. - Tests for nvidia/display + inventory in store; docs (M15/SPEC). Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
96 lines
4.1 KiB
Python
96 lines
4.1 KiB
Python
"""Tests for M15 session-scoped system-log collection (kernel + coredumps)."""
|
|
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
from rigdoctor.core import syslogs
|
|
|
|
|
|
class KernelLogTests(unittest.TestCase):
|
|
def test_passes_since_and_tails(self):
|
|
with mock.patch("shutil.which", return_value="/usr/bin/journalctl"), \
|
|
mock.patch.object(syslogs, "_run", return_value="X" * 100 + "TAILLINE") as run:
|
|
out = syslogs.kernel_log(since=1_000_000_000, max_bytes=8)
|
|
self.assertEqual(out, "TAILLINE")
|
|
cmd = run.call_args[0][0]
|
|
self.assertIn("-k", cmd)
|
|
self.assertIn("--since", cmd)
|
|
|
|
def test_missing_tool_returns_empty(self):
|
|
with mock.patch("shutil.which", return_value=None):
|
|
self.assertEqual(syslogs.kernel_log(), "")
|
|
|
|
|
|
class CoredumpTests(unittest.TestCase):
|
|
def test_empty_when_no_coredumps(self):
|
|
with mock.patch("shutil.which", return_value="/usr/bin/coredumpctl"), \
|
|
mock.patch.object(syslogs, "_run", return_value="No coredumps found."):
|
|
self.assertEqual(syslogs.coredumps(), "")
|
|
|
|
def test_returns_list(self):
|
|
with mock.patch("shutil.which", return_value="/usr/bin/coredumpctl"), \
|
|
mock.patch.object(syslogs, "_run", return_value="TIME PID SIG EXE\n... SEGV PathOfExile"):
|
|
out = syslogs.coredumps()
|
|
self.assertIn("PathOfExile", out)
|
|
|
|
|
|
class NvidiaTests(unittest.TestCase):
|
|
def test_missing_tool(self):
|
|
with mock.patch("shutil.which", return_value=None):
|
|
self.assertEqual(syslogs.nvidia_snapshot(), "")
|
|
|
|
def test_snapshot_head_truncated(self):
|
|
with mock.patch("shutil.which", return_value="/usr/bin/nvidia-smi"), \
|
|
mock.patch.object(syslogs, "_run", return_value="DRIVER\n" + "x" * 99999):
|
|
out = syslogs.nvidia_snapshot(max_bytes=10)
|
|
self.assertEqual(out, "DRIVER\nxxx") # head, not tail
|
|
|
|
|
|
class DisplayTests(unittest.TestCase):
|
|
def test_session_type_env(self):
|
|
with mock.patch.dict("os.environ", {"XDG_SESSION_TYPE": "wayland"}):
|
|
self.assertEqual(syslogs._session_type(), "wayland")
|
|
|
|
def test_x11_tails_xorg_log(self):
|
|
import tempfile
|
|
from pathlib import Path
|
|
log = Path(tempfile.mkdtemp()) / "Xorg.0.log"
|
|
log.write_text("(EE) NVIDIA(GPU-0): something failed")
|
|
with mock.patch.object(syslogs, "_session_type", return_value="x11"), \
|
|
mock.patch.object(syslogs, "_xorg_log", return_value=log):
|
|
out = syslogs.display_log()
|
|
self.assertIn("(EE) NVIDIA", out)
|
|
|
|
def test_wayland_uses_user_journal(self):
|
|
with mock.patch.object(syslogs, "_session_type", return_value="wayland"), \
|
|
mock.patch("shutil.which", return_value="/usr/bin/journalctl"), \
|
|
mock.patch.object(syslogs, "_run", return_value="gnome-shell: GPU error") as run:
|
|
out = syslogs.display_log(since=1_000_000_000)
|
|
self.assertIn("GPU error", out)
|
|
cmd = run.call_args[0][0]
|
|
self.assertIn("--user", cmd)
|
|
self.assertTrue(any(a.startswith("_COMM=") for a in cmd))
|
|
|
|
|
|
class CollectTests(unittest.TestCase):
|
|
def test_collect_combines_sections(self):
|
|
with mock.patch.object(syslogs, "kernel_log", return_value="NVRM: Xid 79"), \
|
|
mock.patch.object(syslogs, "coredumps", return_value="game SIGSEGV"), \
|
|
mock.patch.object(syslogs, "nvidia_snapshot", return_value="Driver Version 595"), \
|
|
mock.patch.object(syslogs, "display_log", return_value="(EE) NVIDIA"):
|
|
out = syslogs.collect()
|
|
for needle in ("Kernel log", "Xid 79", "Crashed processes", "SIGSEGV",
|
|
"NVIDIA snapshot", "595", "Display server log"):
|
|
self.assertIn(needle, out)
|
|
|
|
def test_collect_empty_when_nothing(self):
|
|
with mock.patch.object(syslogs, "kernel_log", return_value=""), \
|
|
mock.patch.object(syslogs, "coredumps", return_value=""), \
|
|
mock.patch.object(syslogs, "nvidia_snapshot", return_value=""), \
|
|
mock.patch.object(syslogs, "display_log", return_value=""):
|
|
self.assertEqual(syslogs.collect(), "")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|