Handle handler error during initialisation

Signed-off-by: Chris Jackson <[email protected]>
zsmartsystems · Oct 4, 2024 · fc5ea82 · fc5ea82
1 parent c038c6f
commit fc5ea82
Show file tree

Hide file tree

Showing 2 changed files with 68 additions and 2 deletions.
diff --git a/...ee.dongle.ember/src/main/java/com/zsmartsystems/zigbee/dongle/ember/ZigBeeDongleEzsp.java b/...ee.dongle.ember/src/main/java/com/zsmartsystems/zigbee/dongle/ember/ZigBeeDongleEzsp.java
@@ -457,22 +457,37 @@ public ZigBeeStatus initialize() {
         stackConfigurer.setPolicy(stackPolicies);
         policies = stackConfigurer.getPolicy(stackPolicies.keySet());
         for (Entry<EzspPolicyId, Integer> policy : policies.entrySet()) {
-            EzspDecisionId decisionId = EzspDecisionId.getEzspDecisionId(policy.getValue());
+            EzspDecisionId decisionId = null;
+            try {
+                decisionId = EzspDecisionId.getEzspDecisionId(policy.getValue());
+            } catch (Exception e) {
+                // Eat me!
+                // This is only for logging.
+            }
             logger.debug("Policy state {} = {} [{}]", policy.getKey(), decisionId,
                     String.format("%02X", policy.getValue()));
         }
 
         EmberNcp ncp = getEmberNcp();
 
         // Get the current network parameters so that any configuration updates start from here
-        networkParameters = ncp.getNetworkParameters().getParameters();
+        EzspGetNetworkParametersResponse networkParametersResponse = ncp.getNetworkParameters();
+        if (networkParametersResponse == null) {
+            return ZigBeeStatus.COMMUNICATION_ERROR;
+        }
+        networkParameters = networkParametersResponse.getParameters();
         logger.debug("Ember initial network parameters are {}", networkParameters);
 
         ieeeAddress = ncp.getIeeeAddress();
         logger.debug("Ember local IEEE Address is {}", ieeeAddress);
 
         ncp.getNetworkParameters();
 
+        if (!frameHandler.isAlive()) {
+            logger.error("Ember frame handler is not alive after initialize!");
+            return ZigBeeStatus.COMMUNICATION_ERROR;
+        }
+
         isConfigured = true;
         logger.debug("EZSP Dongle: initialize done");
 

diff --git a/...ongle.ember/src/test/java/com/zsmartsystems/zigbee/dongle/ember/ZigBeeDongleEzspTest.java b/...ongle.ember/src/test/java/com/zsmartsystems/zigbee/dongle/ember/ZigBeeDongleEzspTest.java
@@ -728,4 +728,55 @@ public EmberNcp getEmberNcp() {
         assertNull(dongle.getTcLinkKey());
         assertNotNull(dongle.getZigBeeNetworkKey());
     }
+
+    @Test
+    public void initializeNoNetorkParameters() {
+        System.out.println("--- " + Thread.currentThread().getStackTrace()[1].getMethodName());
+        final EmberNcp ncp = Mockito.mock(EmberNcp.class);
+        final EzspVersionResponse version = Mockito.mock(EzspVersionResponse.class);
+        Mockito.when(version.getProtocolVersion()).thenReturn(4);
+
+        Mockito.when(ncp.getVersion()).thenReturn(version);
+
+        final ZigBeePort port = Mockito.mock(ZigBeePort.class);
+        Mockito.when(port.open()).thenReturn(true);
+
+        ZigBeeDongleEzsp dongle = new ZigBeeDongleEzsp(port) {
+            @Override
+            public EmberNcp getEmberNcp() {
+                return ncp;
+            }
+        };
+
+        final ZigBeeTransportReceive receiver = Mockito.mock(ZigBeeTransportReceive.class);
+
+        dongle.setZigBeeTransportReceive(receiver);
+        assertEquals(ZigBeeStatus.COMMUNICATION_ERROR, dongle.initialize());
+    }
+
+    @Test
+    public void initializeSuccess() {
+        System.out.println("--- " + Thread.currentThread().getStackTrace()[1].getMethodName());
+        final EmberNcp ncp = Mockito.mock(EmberNcp.class);
+        final EzspVersionResponse version = Mockito.mock(EzspVersionResponse.class);
+        Mockito.when(version.getProtocolVersion()).thenReturn(4);
+
+        Mockito.when(ncp.getVersion()).thenReturn(version);
+        Mockito.when(ncp.getNetworkParameters()).thenReturn(Mockito.mock(EzspGetNetworkParametersResponse.class));
+
+        final ZigBeePort port = Mockito.mock(ZigBeePort.class);
+        Mockito.when(port.open()).thenReturn(true);
+
+        ZigBeeDongleEzsp dongle = new ZigBeeDongleEzsp(port) {
+            @Override
+            public EmberNcp getEmberNcp() {
+                return ncp;
+            }
+        };
+
+        final ZigBeeTransportReceive receiver = Mockito.mock(ZigBeeTransportReceive.class);
+
+        dongle.setZigBeeTransportReceive(receiver);
+        assertEquals(ZigBeeStatus.SUCCESS, dongle.initialize());
+    }
 }