Ensure NmtBase.state return a string for unknown states

Fixes christiansandberg#500
erlend-aasland · Jul 9, 2024 · 991af60 · 991af60
1 parent fa74236
commit 991af60
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/canopen/nmt.py b/canopen/nmt.py
@@ -86,10 +86,10 @@ def state(self) -> str:
         - 'RESET'
         - 'RESET COMMUNICATION'
         """
-        if self._state in NMT_STATES:
+        try:
             return NMT_STATES[self._state]
-        else:
-            return self._state
+        except KeyError:
+            return f"UNKNOWN STATE '{self._state}'"
 
     @state.setter
     def state(self, new_state: str):

diff --git a/test/test_nmt.py b/test/test_nmt.py
@@ -88,7 +88,6 @@ def test_nmt_master_on_heartbeat_initialising(self):
         state = self.node.nmt.wait_for_heartbeat(self.TIMEOUT)
         self.assertEqual(state, "PRE-OPERATIONAL")
 
-    @unittest.expectedFailure
     def test_nmt_master_on_heartbeat_unknown_state(self):
         task = self.net.send_periodic(self.COB_ID, [0xcb], self.PERIOD)
         self.addCleanup(task.stop)