Add reboot button to SFRBox (#86514)

tests/components/sfr_box/test_button.py

@@ -0,0 +1,71 @@
+"""Test the SFR Box buttons."""
+from collections.abc import Generator
+from unittest.mock import patch
+
+import pytest
+from sfrbox_api.exceptions import SFRBoxError
+
+from homeassistant.components.button import DOMAIN as BUTTON_DOMAIN, SERVICE_PRESS
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import ATTR_ENTITY_ID, Platform
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import HomeAssistantError
+
+from . import check_device_registry, check_entities
+from .const import EXPECTED_ENTITIES
+
+from tests.common import mock_device_registry, mock_registry
+
+pytestmark = pytest.mark.usefixtures("system_get_info", "dsl_get_info")
+
+
+@pytest.fixture(autouse=True)
+def override_platforms() -> Generator[None, None, None]:
+    """Override PLATFORMS_WITH_AUTH."""
+    with patch(
+        "homeassistant.components.sfr_box.PLATFORMS_WITH_AUTH", [Platform.BUTTON]
+    ), patch("homeassistant.components.sfr_box.coordinator.SFRBox.authenticate"):
+        yield
+
+
+async def test_buttons(
+    hass: HomeAssistant, config_entry_with_auth: ConfigEntry
+) -> None:
+    """Test for SFR Box buttons."""
+    entity_registry = mock_registry(hass)
+    device_registry = mock_device_registry(hass)
+
+    await hass.config_entries.async_setup(config_entry_with_auth.entry_id)
+    await hass.async_block_till_done()
+
+    check_device_registry(device_registry, EXPECTED_ENTITIES["expected_device"])
+
+    expected_entities = EXPECTED_ENTITIES[Platform.BUTTON]
+    assert len(entity_registry.entities) == len(expected_entities)
+
+    check_entities(hass, entity_registry, expected_entities)
+
+    # Reboot success
+    service_data = {ATTR_ENTITY_ID: "button.sfr_box_reboot"}
+    with patch(
+        "homeassistant.components.sfr_box.button.SFRBox.system_reboot"
+    ) as mock_action:
+        await hass.services.async_call(
+            BUTTON_DOMAIN, SERVICE_PRESS, service_data=service_data, blocking=True
+        )
+
+    assert len(mock_action.mock_calls) == 1
+    assert mock_action.mock_calls[0][1] == ()
+
+    # Reboot failed
+    service_data = {ATTR_ENTITY_ID: "button.sfr_box_reboot"}
+    with patch(
+        "homeassistant.components.sfr_box.button.SFRBox.system_reboot",
+        side_effect=SFRBoxError,
+    ) as mock_action, pytest.raises(HomeAssistantError):
+        await hass.services.async_call(
+            BUTTON_DOMAIN, SERVICE_PRESS, service_data=service_data, blocking=True
+        )
+
+    assert len(mock_action.mock_calls) == 1
+    assert mock_action.mock_calls[0][1] == ()