"""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()