Fastdotcom service optimization (#107179)

* Startup mechanic * Workable service (again) * Optimized version, for now * Minor refactoring * Test cases * Fixing test case * Adding startup comment * State_unknown added * Update homeassistant/components/fastdotcom/services.py Co-authored-by: Martin Hjelmare <marhje52@gmail.com> * Check if config entries are not found * Update tests/components/fastdotcom/test_service.py Co-authored-by: Martin Hjelmare <marhje52@gmail.com> * Update homeassistant/components/fastdotcom/services.py Co-authored-by: Martin Hjelmare <marhje52@gmail.com> * Update homeassistant/components/fastdotcom/services.py Co-authored-by: Martin Hjelmare <marhje52@gmail.com> --------- Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
2024-01-11 08:27:15 +01:00 · 2024-01-11 08:27:15 +01:00 · b08832a89a
commit b08832a89a
parent e0457590d1
5 changed files with 161 additions and 38 deletions
--- a/tests/components/fastdotcom/test_service.py
+++ b/tests/components/fastdotcom/test_service.py
@ -0,0 +1,87 @@
+"""Test Fastdotcom service."""
+from unittest.mock import patch
+
+import pytest
+
+from homeassistant.components.fastdotcom.const import DEFAULT_NAME, DOMAIN, SERVICE_NAME
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import HomeAssistantError
+
+from tests.common import MockConfigEntry
+
+
+async def test_service(hass: HomeAssistant) -> None:
+    """Test the Fastdotcom service."""
+    config_entry = MockConfigEntry(
+        domain=DOMAIN,
+        unique_id="UNIQUE_TEST_ID",
+        title=DEFAULT_NAME,
+    )
+    config_entry.add_to_hass(hass)
+
+    with patch(
+        "homeassistant.components.fastdotcom.coordinator.fast_com", return_value=0
+    ):
+        await hass.config_entries.async_setup(config_entry.entry_id)
+        await hass.async_block_till_done()
+
+    state = hass.states.get("sensor.fast_com_download")
+    assert state is not None
+    assert state.state == "0"
+
+    with patch(
+        "homeassistant.components.fastdotcom.coordinator.fast_com", return_value=5.0
+    ):
+        await hass.services.async_call(DOMAIN, SERVICE_NAME, blocking=True)
+
+    state = hass.states.get("sensor.fast_com_download")
+    assert state is not None
+    assert state.state == "5.0"
+
+
+async def test_service_unloaded_entry(hass: HomeAssistant) -> None:
+    """Test service called when config entry unloaded."""
+    config_entry = MockConfigEntry(
+        domain=DOMAIN,
+        unique_id="UNIQUE_TEST_ID",
+        title=DEFAULT_NAME,
+    )
+    config_entry.add_to_hass(hass)
+
+    with patch(
+        "homeassistant.components.fastdotcom.coordinator.fast_com", return_value=0
+    ):
+        await hass.config_entries.async_setup(config_entry.entry_id)
+        await hass.async_block_till_done()
+
+    assert config_entry
+    await config_entry.async_unload(hass)
+
+    with pytest.raises(HomeAssistantError) as exc:
+        await hass.services.async_call(DOMAIN, SERVICE_NAME, blocking=True)
+
+    assert "Fast.com is not loaded" in str(exc)
+
+
+async def test_service_removed_entry(hass: HomeAssistant) -> None:
+    """Test service called when config entry was removed and HA was not restarted yet."""
+    config_entry = MockConfigEntry(
+        domain=DOMAIN,
+        unique_id="UNIQUE_TEST_ID",
+        title=DEFAULT_NAME,
+    )
+    config_entry.add_to_hass(hass)
+
+    with patch(
+        "homeassistant.components.fastdotcom.coordinator.fast_com", return_value=0
+    ):
+        await hass.config_entries.async_setup(config_entry.entry_id)
+        await hass.async_block_till_done()
+
+    assert config_entry
+    await hass.config_entries.async_remove(config_entry.entry_id)
+
+    with pytest.raises(HomeAssistantError) as exc:
+        await hass.services.async_call(DOMAIN, SERVICE_NAME, blocking=True)
+
+    assert "No Fast.com config entries found" in str(exc)