Reolink translate errors (#132301)

2025-11-10 11:29:46 +00:00 · 2024-12-18 20:22:33 +01:00
parent c8f050ecbc
commit 19e6867f1a
14 changed files with 331 additions and 152 deletions
--- a/tests/components/reolink/test_util.py
+++ b/tests/components/reolink/test_util.py
@@ -0,0 +1,115 @@
+"""Test the Reolink util functions."""
+
+from unittest.mock import MagicMock, patch
+
+import pytest
+from reolink_aio.exceptions import (
+    ApiError,
+    CredentialsInvalidError,
+    InvalidContentTypeError,
+    InvalidParameterError,
+    LoginError,
+    NoDataError,
+    NotSupportedError,
+    ReolinkConnectionError,
+    ReolinkError,
+    ReolinkTimeoutError,
+    SubscriptionError,
+    UnexpectedDataError,
+)
+
+from homeassistant.components.number import (
+    ATTR_VALUE,
+    DOMAIN as NUMBER_DOMAIN,
+    SERVICE_SET_VALUE,
+)
+from homeassistant.config_entries import ConfigEntryState
+from homeassistant.const import ATTR_ENTITY_ID, Platform
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
+
+from .conftest import TEST_NVR_NAME
+
+from tests.common import MockConfigEntry
+
+
+@pytest.mark.parametrize(
+    ("side_effect", "expected"),
+    [
+        (
+            ApiError("Test error"),
+            HomeAssistantError,
+        ),
+        (
+            CredentialsInvalidError("Test error"),
+            HomeAssistantError,
+        ),
+        (
+            InvalidContentTypeError("Test error"),
+            HomeAssistantError,
+        ),
+        (
+            InvalidParameterError("Test error"),
+            ServiceValidationError,
+        ),
+        (
+            LoginError("Test error"),
+            HomeAssistantError,
+        ),
+        (
+            NoDataError("Test error"),
+            HomeAssistantError,
+        ),
+        (
+            NotSupportedError("Test error"),
+            HomeAssistantError,
+        ),
+        (
+            ReolinkConnectionError("Test error"),
+            HomeAssistantError,
+        ),
+        (
+            ReolinkError("Test error"),
+            HomeAssistantError,
+        ),
+        (
+            ReolinkTimeoutError("Test error"),
+            HomeAssistantError,
+        ),
+        (
+            SubscriptionError("Test error"),
+            HomeAssistantError,
+        ),
+        (
+            UnexpectedDataError("Test error"),
+            HomeAssistantError,
+        ),
+    ],
+)
+async def test_try_function(
+    hass: HomeAssistant,
+    config_entry: MockConfigEntry,
+    reolink_connect: MagicMock,
+    side_effect: ReolinkError,
+    expected: Exception,
+) -> None:
+    """Test try_function error translations using number entity."""
+    reolink_connect.volume.return_value = 80
+
+    with patch("homeassistant.components.reolink.PLATFORMS", [Platform.NUMBER]):
+        assert await hass.config_entries.async_setup(config_entry.entry_id)
+    await hass.async_block_till_done()
+    assert config_entry.state is ConfigEntryState.LOADED
+
+    entity_id = f"{Platform.NUMBER}.{TEST_NVR_NAME}_volume"
+
+    reolink_connect.set_volume.side_effect = side_effect
+    with pytest.raises(expected):
+        await hass.services.async_call(
+            NUMBER_DOMAIN,
+            SERVICE_SET_VALUE,
+            {ATTR_ENTITY_ID: entity_id, ATTR_VALUE: 50},
+            blocking=True,
+        )
+
+    reolink_connect.set_volume.reset_mock(side_effect=True)