Add test to listen for new sensor being labelled

Author: andrew-codechimp

tests/test_binary_sensor.py

@@ -31,15 +31,18 @@
     (
         "state_1",
         "state_2",
+        "state_3",
         "expected_state",
     ),
     [
         (
             "unavailable",
             "on",
             "on",
+            "on",
         ),
         (
+            "on",
             "on",
             "on",
             "off",
@@ -50,6 +53,7 @@ async def test_state_sensor(
     hass: HomeAssistant,
     state_1: str,
     state_2: str,
+    state_3: str,
     expected_state: str,
     entity_registry: er.EntityRegistry,
     label_registry: lr.LabelRegistry,
@@ -106,6 +110,25 @@ async def test_state_sensor(
     assert state is not None
     assert state.state == expected_state
 
+    # Add a third sensor to test the listener change
+    sensor3_entity_entry = entity_registry.async_get_or_create(
+        "sensor", "test_3", "unique", suggested_object_id="test_3"
+    )
+    await hass.async_block_till_done()
+    sensor3_entity_entry = entity_registry.async_update_entity(
+        sensor3_entity_entry.entity_id, labels={test_label.label_id}
+    )
+    await hass.async_block_till_done()
+    assert sensor3_entity_entry.entity_id == "sensor.test_3"
+
+    hass.states.async_set(sensor3_entity_entry.entity_id, state_3)
+    await hass.async_block_till_done()
+
+    state = hass.states.get("binary_sensor.test_state")
+
+    assert state is not None
+    assert state.state == expected_state
+
 
 @pytest.mark.parametrize(
     (