Tests

Author: andrew-codechimp

custom_components/label_state/binary_sensor.py

@@ -149,6 +149,7 @@ async def async_added_to_hass(self) -> None:
 
         ent_reg = er.async_get(self.hass)
         entries = er.async_entries_for_label(ent_reg, self._label)
+
         for entity_entry in entries:
             for label in entity_entry.labels:
                 if label == self._label:

tests/test_binary_sensor.py

@@ -63,13 +63,21 @@ async def test_state_sensor(
     sensor1_entity_entry = entity_registry.async_get_or_create(
         "sensor", "test_1", "unique", suggested_object_id="test_1"
     )
-    sensor1_entity_entry.labels.add(test_label.label_id)
+    await hass.async_block_till_done()
+    sensor1_entity_entry = entity_registry.async_update_entity(
+        sensor1_entity_entry.entity_id, labels={test_label.label_id}
+    )
+    await hass.async_block_till_done()
     assert sensor1_entity_entry.entity_id == "sensor.test_1"
 
     sensor2_entity_entry = entity_registry.async_get_or_create(
         "sensor", "test_2", "unique", suggested_object_id="test_2"
     )
-    sensor2_entity_entry.labels.add(test_label.label_id)
+    await hass.async_block_till_done()
+    sensor2_entity_entry = entity_registry.async_update_entity(
+        sensor2_entity_entry.entity_id, labels={test_label.label_id}
+    )
+    await hass.async_block_till_done()
     assert sensor2_entity_entry.entity_id == "sensor.test_2"
 
     config = MockConfigEntry(
@@ -110,36 +118,46 @@ async def test_state_sensor(
     [
         ("11", "12", 10, 20, "off"),
         ("1", "12", 10, 20, "on"),
+        ("11", "12", 10, None, "off"),
+        ("1", "12", 10, None, "on"),
+        ("1", "19", None, 20, "off"),
+        ("1", "22", None, 20, "on"),
     ],
 )
 async def test_numeric_state_sensor(
     hass: HomeAssistant,
     state_1: str,
     state_2: str,
-    state_lower_limit: float,
-    state_upper_limit: float,
+    state_lower_limit: float | None,
+    state_upper_limit: float | None,
     expected_state: str,
     entity_registry: er.EntityRegistry,
     label_registry: lr.LabelRegistry,
 ) -> None:
     """Test the numeric state sensor."""
 
     test_label = label_registry.async_create(
-        "test_numeric_state",
+        "test_numeric_state_label",
     )
 
     sensor1_entity_entry = entity_registry.async_get_or_create(
         "sensor", "test_1", "unique", suggested_object_id="test_1"
     )
-    sensor1_entity_entry.labels.add(test_label.label_id)
+    await hass.async_block_till_done()
+    sensor1_entity_entry = entity_registry.async_update_entity(
+        sensor1_entity_entry.entity_id, labels={test_label.label_id}
+    )
     await hass.async_block_till_done()
     assert sensor1_entity_entry.entity_id == "sensor.test_1"
     assert test_label.label_id in sensor1_entity_entry.labels
 
     sensor2_entity_entry = entity_registry.async_get_or_create(
         "sensor", "test_2", "unique", suggested_object_id="test_2"
     )
-    sensor2_entity_entry.labels.add(test_label.label_id)
+    await hass.async_block_till_done()
+    sensor2_entity_entry = entity_registry.async_update_entity(
+        sensor2_entity_entry.entity_id, labels={test_label.label_id}
+    )
     await hass.async_block_till_done()
     assert sensor2_entity_entry.entity_id == "sensor.test_2"
     assert test_label.label_id in sensor2_entity_entry.labels