feat: 100% coverage

Author: bramstroker

custom_components/powercalc/analytics/analytics.py

@@ -125,9 +125,6 @@ def add(self, key: str, value: Any) -> None:  # noqa: ANN401
 
     def set_flag(self, key: str) -> None:
         """Set a boolean flag to True"""
-        if self._already_seen(key):
-            return
-
         self._data[key] = True  # type:ignore
 
 

custom_components/powercalc/discovery.py

@@ -133,7 +133,7 @@ async def initialize_existing_entries(self) -> None:
         """Build a list of config entries which are already setup, to prevent duplicate discovery flows"""
         for entry in self.hass.config_entries.async_entries(DOMAIN):
             if not entry.unique_id:
-                continue
+                continue  # pragma: no cover
 
             self.initialized_flows.add(entry.unique_id)
             entity_id = entry.data.get(CONF_ENTITY_ID)

custom_components/powercalc/select.py

@@ -52,8 +52,8 @@ async def async_setup_platform(
     key = _key(None)
     pending = hass.data[DOMAIN].setdefault(DATA_PENDING_SELECT_ENTITIES, {}).pop(key, [])
     if pending:
-        _LOGGER.debug("Adding TariffSelect entities pending")
-        async_add_entities(pending)
+        _LOGGER.debug("Adding TariffSelect entities pending")  # pragma: no cover
+        async_add_entities(pending)  # pragma: no cover
 
     delayed_add_entities_handler(hass, async_add_entities)
 

tests/test_sensor.py

@@ -1,6 +1,6 @@
 from datetime import timedelta
 import logging
-from unittest.mock import patch
+from unittest.mock import AsyncMock, patch
 
 from homeassistant.components import light, sensor
 from homeassistant.components.integration.sensor import ATTR_SOURCE_ID
@@ -59,6 +59,7 @@
     CalculationStrategy,
     SensorType,
 )
+from custom_components.powercalc.errors import SensorConfigurationError
 
 from .common import (
     create_input_boolean,
@@ -195,6 +196,46 @@ async def test_create_nested_group_sensor(hass: HomeAssistant) -> None:
     assert group2.state == "0.00"
 
 
+async def test_create_nested_without_group(hass: HomeAssistant) -> None:
+    await run_powercalc_setup(
+        hass,
+        {
+            CONF_ENTITIES: [
+                get_simple_fixed_config("switch.test1", 50),
+                {
+                    CONF_ENTITIES: [
+                        get_simple_fixed_config("switch.test2", 50),
+                    ],
+                },
+            ],
+        },
+    )
+    assert hass.states.get("sensor.test1_power")
+    assert hass.states.get("sensor.test2_power")
+
+
+async def test_create_nested_handles_exception(hass: HomeAssistant, caplog: pytest.LogCaptureFixture) -> None:
+    caplog.set_level(logging.ERROR)
+
+    with patch(
+        "custom_components.powercalc.sensor.attach_entities_to_source_device",
+        new=AsyncMock(side_effect=SensorConfigurationError("My custom error message")),
+    ):
+        await run_powercalc_setup(
+            hass,
+            {
+                CONF_ENTITIES: [
+                    get_simple_fixed_config("switch.test1", 50),
+                    get_simple_fixed_config("switch.test2", 50),
+                ],
+            },
+        )
+    assert not hass.states.get("sensor.test1_power")
+    assert not hass.states.get("sensor.test2_power")
+
+    assert "My custom error message" in caplog.text
+
+
 async def test_light_lut_strategy(
     hass: HomeAssistant,
     mock_entity_with_model_information: MockEntityWithModel,