kytos-ng / telemetry_int

    """TestINTManager."""

    async def test_disable_int_metadata(self, monkeypatch) -> None:

        """Test disable INT metadata args."""

        controller = MagicMock()

        api_mock = AsyncMock()

        monkeypatch.setattr("napps.kytos.telemetry_int.managers.int.api", api_mock)

        int_manager = INTManager(controller)

        int_manager.remove_int_flows = AsyncMock()

        await int_manager.disable_int({}, False)

        assert api_mock.add_evcs_metadata.call_count == 1

        args = api_mock.add_evcs_metadata.call_args[0]

        assert args[0] == {}

        assert "telemetry" in args[1]

        telemetry_dict = args[1]["telemetry"]

        expected_keys = ["enabled", "status", "status_reason", "status_updated_at"]

        assert sorted(list(telemetry_dict.keys())) == sorted(expected_keys)

        assert not telemetry_dict["enabled"]

        assert telemetry_dict["status"] == "DOWN"

        assert telemetry_dict["status_reason"] == ["disabled"]

        assert args[2] is False

    async def test_enable_int_metadata(self, monkeypatch) -> None:

        """Test enable INT metadata args."""

        assert args[2] is False

    async def test_enable_int_metadata(self, monkeypatch) -> None:

        """Test enable INT metadata args."""

        controller = MagicMock()

        api_mock = AsyncMock()

        monkeypatch.setattr("napps.kytos.telemetry_int.managers.int.api", api_mock)

        int_manager = INTManager(controller)

        int_manager.remove_int_flows = AsyncMock()

        await int_manager.enable_int({}, False)

        assert api_mock.add_evcs_metadata.call_count == 1

        args = api_mock.add_evcs_metadata.call_args[0]

        assert args[0] == {}

        assert "telemetry" in args[1]

        telemetry_dict = args[1]["telemetry"]

        expected_keys = ["enabled", "status", "status_reason", "status_updated_at"]

        assert sorted(list(telemetry_dict.keys())) == sorted(expected_keys)

        assert telemetry_dict["enabled"] is True

        assert telemetry_dict["status"] == "UP"

        assert telemetry_dict["status_reason"] == []