Register endpoint with OTA input cluster on startup

Author: puddly

tests/test_application.py

@@ -177,7 +177,7 @@ def application(znp_server):
         ],
     )
 
-    active_eps = [100, 12, 11, 8, 1]
+    active_eps = [100, 13, 12, 11, 8, 1]
 
     znp_server.reply_to(
         request=c.ZDO.ActiveEpReq.Req(DstAddr=0x0000, NWKAddrOfInterest=0x0000),
@@ -320,7 +320,7 @@ async def test_application_startup_endpoints(application):
 
     await app.startup(auto_form=False)
 
-    assert len(endpoints) == 5
+    assert len(endpoints) == 6
 
 
 @pytest_mark_asyncio_timeout(seconds=5)

zigpy_znp/zigbee/application.py

@@ -14,8 +14,8 @@
 
 from zigpy.zdo.types import ZDOCmd, ZDOHeader, CLUSTERS as ZDO_CLUSTERS
 
+from zigpy.zcl import clusters
 from zigpy.types import ExtendedPanId, deserialize as list_deserialize
-from zigpy.zcl.clusters.security import IasZone
 from zigpy.exceptions import DeliveryError
 
 import zigpy_znp.config as conf
@@ -394,10 +394,14 @@ async def startup(self, auto_form=False):
         await self._register_endpoint(
             endpoint=8,
             device_id=zigpy.profiles.zha.DeviceType.IAS_CONTROL,
-            output_clusters=[IasZone.cluster_id],
+            output_clusters=[clusters.security.IasZone.cluster_id],
         )
         await self._register_endpoint(endpoint=11)
         await self._register_endpoint(endpoint=12)
+        await self._register_endpoint(
+            endpoint=13, input_clusters=[clusters.general.Ota.cluster_id]
+        )
+
         await self._register_endpoint(
             endpoint=100, profile_id=zigpy.profiles.zll.PROFILE_ID, device_id=0x0005
         )