diff --git a/homeassistant/components/zwave_js/__init__.py b/homeassistant/components/zwave_js/__init__.py index b56298e36ba762..2ff4f40a5ad46a 100644 --- a/homeassistant/components/zwave_js/__init__.py +++ b/homeassistant/components/zwave_js/__init__.py @@ -741,6 +741,7 @@ def async_on_notification(self, event: dict[str, Any]) -> None: ATTR_DOMAIN: DOMAIN, ATTR_NODE_ID: notification.node.node_id, ATTR_HOME_ID: driver.controller.home_id, + ATTR_ENDPOINT: notification.endpoint_idx, ATTR_DEVICE_ID: device.id, ATTR_COMMAND_CLASS: notification.command_class, } diff --git a/tests/components/zwave_js/test_events.py b/tests/components/zwave_js/test_events.py index f91250700ddd73..80b179248d8aa0 100644 --- a/tests/components/zwave_js/test_events.py +++ b/tests/components/zwave_js/test_events.py @@ -173,6 +173,7 @@ async def test_notifications( assert len(events) == 1 assert events[0].data["home_id"] == client.driver.controller.home_id assert events[0].data["node_id"] == 32 + assert events[0].data["endpoint"] == 0 assert events[0].data["type"] == 6 assert events[0].data["event"] == 5 assert events[0].data["label"] == "Access Control" @@ -206,6 +207,7 @@ async def test_notifications( assert len(events) == 2 assert events[1].data["home_id"] == client.driver.controller.home_id assert events[1].data["node_id"] == 32 + assert events[0].data["endpoint"] == 0 assert events[1].data["event_type"] == 5 assert events[1].data["event_type_label"] == "test1" assert events[1].data["data_type"] == 2 @@ -233,6 +235,7 @@ async def test_notifications( assert len(events) == 3 assert events[2].data["home_id"] == client.driver.controller.home_id assert events[2].data["node_id"] == 32 + assert events[0].data["endpoint"] == 0 assert events[2].data["event_type"] == 4 assert events[2].data["event_type_label"] == "test1" assert events[2].data["direction"] == "up"