Skip to content

Commit 708c929

Browse files
authored
Update controller inclusion state in handle_inclusion_state_changed (#1173)
* Update controller inclusion state in `handle_inclusion_state_changed` * add test
1 parent a6af850 commit 708c929

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

test/model/test_controller.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2293,3 +2293,18 @@ async def test_identify(client, multisensor_6):
22932293
)
22942294
client.driver.controller.receive_event(event)
22952295
assert event.data["node"] == multisensor_6
2296+
2297+
2298+
async def test_inclusion_state_changed(controller):
2299+
"""Test that inclusion state is updated after receiving inclusion state changed event."""
2300+
assert controller.inclusion_state == InclusionState.IDLE
2301+
event = Event(
2302+
"inclusion state changed",
2303+
{
2304+
"source": "controller",
2305+
"event": "inclusion state changed",
2306+
"state": InclusionState.INCLUDING.value,
2307+
},
2308+
)
2309+
controller.receive_event(event)
2310+
assert controller.inclusion_state == InclusionState.INCLUDING

zwave_js_server/model/controller/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -909,6 +909,7 @@ def handle_inclusion_started(self, event: Event) -> None:
909909

910910
def handle_inclusion_state_changed(self, event: Event) -> None:
911911
"""Process an inclusion state changed event."""
912+
self.data["inclusionState"] = event.data["state"]
912913

913914
def handle_exclusion_started(self, event: Event) -> None:
914915
"""Process an exclusion started event."""

0 commit comments

Comments
 (0)