frigate/frigate/test/test_config_util.py

"""Tests for the shared runtime config swap helper."""

import unittest
from unittest.mock import MagicMock

from frigate.api.config_util import swap_runtime_config


class TestSwapRuntimeConfig(unittest.TestCase):
    """swap_runtime_config rebinds every collaborator to the new config."""

    def _make_app(self) -> MagicMock:
        app = MagicMock()
        app.dispatcher.comms = [MagicMock(), MagicMock()]
        return app

    def test_rebinds_all_references(self) -> None:
        app = self._make_app()
        config = MagicMock(name="new_config")

        swap_runtime_config(app, config)

        self.assertIs(app.frigate_config, config)
        app.genai_manager.update_config.assert_called_once_with(config)
        app.profile_manager.update_config.assert_called_once_with(config)
        self.assertIs(app.stats_emitter.config, config)
        self.assertIs(app.dispatcher.config, config)
        for comm in app.dispatcher.comms:
            self.assertIs(comm.config, config)

    def test_reapplies_runtime_state_after_swap(self) -> None:
        app = self._make_app()
        config = MagicMock(name="new_config")

        swap_runtime_config(app, config)

        # the swap rebuilds cameras from yaml, so overrides must be re-layered
        app.dispatcher.reapply_runtime_state_to_config.assert_called_once_with()

    def test_tolerates_missing_optional_collaborators(self) -> None:
        app = MagicMock()
        app.profile_manager = None
        app.stats_emitter = None
        app.dispatcher = None
        config = MagicMock(name="new_config")

        # must not raise when the optional collaborators are absent
        swap_runtime_config(app, config)

        self.assertIs(app.frigate_config, config)
        app.genai_manager.update_config.assert_called_once_with(config)


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