Bring coverage up

Author: puddly

tests/test_application.py

@@ -2607,3 +2607,25 @@ def test_on_packet_received_multicast(app: ControllerApplication) -> None:
 
     # Verify packet_received was called with the same packet (dst already set)
     assert packet_received_mock.mock_calls == [call(event.packet)]
+
+
+async def test_on_message_sent_via_binding(app: ControllerApplication) -> None:
+    """Test _on_message_sent with OUTGOING_VIA_BINDING message type."""
+    # Create a pending request future
+    future = asyncio.get_running_loop().create_future()
+    app._pending_requests[(0x1234, 0x42)] = future
+
+    event = MessageSentEvent(
+        status=t.sl_Status.OK,
+        message_type=t.EmberOutgoingMessageType.OUTGOING_VIA_BINDING,
+        destination=0x1234,
+        aps_frame=t.EmberApsFrame(),
+        message_tag=0x42,
+        message_contents=b"test",
+    )
+
+    app._on_message_sent(event)
+
+    # Verify the future was resolved
+    assert future.done()
+    assert future.result() == (t.sl_Status.OK, "message send success")

tests/test_ezsp_protocol.py

@@ -6,7 +6,12 @@
 import zigpy.types
 
 from bellows.ezsp import EZSP
-from bellows.ezsp.protocol import PacketReceivedEvent
+from bellows.ezsp.protocol import (
+    IdConflictEvent,
+    PacketReceivedEvent,
+    RouteRecordEvent,
+    TrustCenterJoinEvent,
+)
 import bellows.ezsp.v4
 import bellows.ezsp.v9
 from bellows.ezsp.v9.commands import GetTokenDataRsp
@@ -423,3 +428,76 @@ def test_incoming_message_ignored_type(prot_hndl, caplog) -> None:
     # No event should be emitted for ignored message types
     assert len(handler.mock_calls) == 0
     assert "Ignoring message type" in caplog.text
+
+
+def test_trust_center_join_handler(prot_hndl) -> None:
+    """Test trustCenterJoinHandler callback."""
+    handler = MagicMock()
+    prot_hndl.on_event(TrustCenterJoinEvent.event_type, handler)
+
+    ieee = t.EUI64.convert("aa:bb:cc:dd:ee:ff:00:11")
+    prot_hndl.handle_parsed_callback(
+        "trustCenterJoinHandler",
+        {
+            "newNodeId": t.EmberNodeId(0x1234),
+            "newNodeEui64": ieee,
+            "status": t.EmberDeviceUpdate.STANDARD_SECURITY_UNSECURED_JOIN,
+            "policyDecision": t.EmberJoinDecision.NO_ACTION,
+            "parentOfNewNodeId": t.EmberNodeId(0x0000),
+        }.values(),
+    )
+
+    assert handler.mock_calls == [
+        call(
+            TrustCenterJoinEvent(
+                nwk=t.EmberNodeId(0x1234),
+                ieee=ieee,
+                device_update_status=t.EmberDeviceUpdate.STANDARD_SECURITY_UNSECURED_JOIN,
+                decision=t.EmberJoinDecision.NO_ACTION,
+                parent_nwk=t.EmberNodeId(0x0000),
+            )
+        )
+    ]
+
+
+def test_incoming_route_record_handler(prot_hndl) -> None:
+    """Test incomingRouteRecordHandler callback."""
+    handler = MagicMock()
+    prot_hndl.on_event(RouteRecordEvent.event_type, handler)
+
+    ieee = t.EUI64.convert("aa:bb:cc:dd:ee:ff:00:11")
+    prot_hndl.handle_parsed_callback(
+        "incomingRouteRecordHandler",
+        {
+            "source": t.EmberNodeId(0x1234),
+            "sourceEui": ieee,
+            "lastHopLqi": t.uint8_t(200),
+            "lastHopRssi": t.int8s(-40),
+            "relayList": [t.EmberNodeId(0x0001), t.EmberNodeId(0x0002)],
+        }.values(),
+    )
+
+    assert handler.mock_calls == [
+        call(
+            RouteRecordEvent(
+                nwk=t.EmberNodeId(0x1234),
+                ieee=ieee,
+                lqi=t.uint8_t(200),
+                rssi=t.int8s(-40),
+                relays=[t.EmberNodeId(0x0001), t.EmberNodeId(0x0002)],
+            )
+        )
+    ]
+
+
+def test_id_conflict_handler(prot_hndl) -> None:
+    """Test idConflictHandler callback."""
+    handler = MagicMock()
+    prot_hndl.on_event(IdConflictEvent.event_type, handler)
+
+    prot_hndl.handle_parsed_callback(
+        "idConflictHandler",
+        {"conflictingId": t.EmberNodeId(0x1234)}.values(),
+    )
+
+    assert handler.mock_calls == [call(IdConflictEvent(nwk=t.EmberNodeId(0x1234)))]