from __future__ import annotations

import unittest

from src.control_plane import SERVICE_SPECS, stack_summary


class ControlPlaneServiceTests(unittest.TestCase):
    def test_service_registry_includes_required_modules(self) -> None:
        ids = {spec.service_id for spec in SERVICE_SPECS}
        self.assertTrue(
            {
                "mitmproxy",
                "vscode_server",
                "uptime_kuma",
                "pulseaudio",
                "kmscube",
                "openclaw",
            }.issubset(ids)
        )

    def test_stack_summary_exposes_new_quick_links(self) -> None:
        links = stack_summary()["quick_links"]
        for key in (
            "mitmproxy",
            "vscode_server",
            "uptime_kuma",
            "pulseaudio_tcp",
            "openclaw",
        ):
            self.assertIn(key, links)


if __name__ == "__main__":
    unittest.main()