from __future__ import annotations

import unittest
from unittest.mock import patch

from bot_bottle.backend.egress_apply import EgressApplyError
from bot_bottle.backend.smolmachines import egress_apply
from bot_bottle.backend.smolmachines.smolvm import SmolvmRunResult


class TestFetchCurrentRoutes(unittest.TestCase):
    def test_reads_routes_from_sidecar_vm(self):
        with patch.object(
            egress_apply._smolvm,
            "machine_exec",
            return_value=SmolvmRunResult(0, "routes", ""),
        ) as exec_:
            self.assertEqual("routes", egress_apply.fetch_current_routes("dev-abc"))
        exec_.assert_called_once_with(
            "bot-bottle-sidecars-dev-abc",
            ["cat", "/etc/egress/routes.yaml"],
        )

    def test_read_failure_raises_apply_error(self):
        with patch.object(
            egress_apply._smolvm,
            "machine_exec",
            return_value=SmolvmRunResult(1, "", "nope"),
        ):
            with self.assertRaises(EgressApplyError):
                egress_apply.fetch_current_routes("dev-abc")


class TestReload(unittest.TestCase):
    def test_sends_hup_to_sidecar_init(self):
        with patch.object(
            egress_apply._smolvm,
            "machine_exec",
            return_value=SmolvmRunResult(0, "", ""),
        ) as exec_:
            egress_apply.SmolmachinesEgressApplicator()._signal_bundle_reload("dev-abc")
        exec_.assert_called_once_with(
            "bot-bottle-sidecars-dev-abc",
            ["sh", "-c", "kill -HUP 1"],
        )

    def test_reload_failure_raises_apply_error(self):
        with patch.object(
            egress_apply._smolvm,
            "machine_exec",
            return_value=SmolvmRunResult(1, "", "nope"),
        ):
            with self.assertRaises(EgressApplyError):
                egress_apply.SmolmachinesEgressApplicator()._signal_bundle_reload("dev-abc")


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