"""Tests for daemon/server.py polling functions."""

import asyncio
import json
from unittest.mock import AsyncMock, patch


class TestPollIntervals:
    def test_default_intervals_loaded(self):
        from daemon.server import _POLL_INTERVALS

        assert "firewall" in _POLL_INTERVALS
        assert _POLL_INTERVALS["firewall"] == 30
        assert _POLL_INTERVALS["wireguard"] == 10
        assert _POLL_INTERVALS["dnsmasq"] == 10
        assert _POLL_INTERVALS["networkd"] == 10

    def test_env_override(self):
        """VACUUM_WALL_POLL_INTERVALS env var can override values."""
        import importlib

        with patch.dict(
            "os.environ", {"VACUUM_WALL_POLL_INTERVALS": "firewall:60,wireguard:5"}
        ):
            import daemon.server

            importlib.reload(daemon.server)
            assert daemon.server._POLL_INTERVALS["firewall"] == 60
            assert daemon.server._POLL_INTERVALS["wireguard"] == 5
            assert daemon.server._POLL_INTERVALS["dnsmasq"] == 10
        importlib.reload(daemon.server)


class TestBroadcastTick:
    def test_sends_tick_message(self):
        from daemon.server import _ws_subscribers, broadcast_tick

        mock_ws = AsyncMock()
        mock_ws.send_str = AsyncMock()
        _ws_subscribers.add(mock_ws)
        try:
            asyncio.run(broadcast_tick(["firewall", "wireguard"]))
            mock_ws.send_str.assert_called_once()
            sent = json.loads(mock_ws.send_str.call_args[0][0])
            assert sent["type"] == "tick"
            assert sent["subsystems"] == ["firewall", "wireguard"]
        finally:
            _ws_subscribers.discard(mock_ws)

    def test_prunes_dead_subscribers(self):
        from daemon.server import _ws_subscribers, broadcast_tick

        mock_ws = AsyncMock()
        mock_ws.send_str = AsyncMock(side_effect=Exception("broken"))
        _ws_subscribers.add(mock_ws)
        asyncio.run(broadcast_tick(["firewall"]))
        assert mock_ws not in _ws_subscribers


class TestPollTasks:
    def test_poll_tasks_is_a_set(self):
        from daemon.server import _poll_tasks

        assert isinstance(_poll_tasks, set)