Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion bellows/ezsp/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -601,13 +601,17 @@ async def set_source_routing(self, enabled: bool) -> None:
LOGGER.warning("Couldn't set concentrator type %s: %s", True, res)

if self._ezsp_version >= 8:
await self.setSourceRouteDiscoveryMode(
(remaining_ms,) = await self.setSourceRouteDiscoveryMode(
mode=(
t.SourceRouteDiscoveryMode.ON
if enabled
else t.SourceRouteDiscoveryMode.OFF
)
)
LOGGER.debug(
"Source route discovery: %d ms until next MTORR broadcast",
remaining_ms,
)

def start_ezsp(self):
"""Mark EZSP as running."""
Expand Down
4 changes: 2 additions & 2 deletions tests/test_ezsp.py
Original file line number Diff line number Diff line change
Expand Up @@ -425,7 +425,7 @@ async def replacement(command_name, tokenId=None, valueId=None):
async def test_set_enable_source_routing(ezsp_f):
"""Test enabling source routing."""
ezsp_f.setConcentrator = AsyncMock(return_value=(t.EmberStatus.SUCCESS,))
ezsp_f.setSourceRouteDiscoveryMode = AsyncMock()
ezsp_f.setSourceRouteDiscoveryMode = AsyncMock(return_value=(12345,))

await ezsp_f.set_source_routing(enabled=True)
assert len(ezsp_f.setSourceRouteDiscoveryMode.mock_calls) == 1
Expand All @@ -438,7 +438,7 @@ async def test_set_enable_source_routing(ezsp_f):
async def test_set_disable_source_routing(ezsp_f):
"""Test disabling source routing."""
ezsp_f.setConcentrator = AsyncMock(return_value=(t.EmberStatus.SUCCESS,))
ezsp_f.setSourceRouteDiscoveryMode = AsyncMock()
ezsp_f.setSourceRouteDiscoveryMode = AsyncMock(return_value=(0,))

await ezsp_f.set_source_routing(enabled=False)
assert len(ezsp_f.setSourceRouteDiscoveryMode.mock_calls) == 1
Expand Down
Loading