Add Tests for Pre Nak

TD22057 · Dec 3, 2020 · 33ac375 · 33ac375
1 parent 9ba2ffe
commit 33ac375
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/tests/handler/test_BroadcastCmdResponse.py b/tests/handler/test_BroadcastCmdResponse.py
@@ -67,6 +67,12 @@ def callback(msg, on_done=None):
         r = handler.msg_received(proto, msg)
         assert r == Msg.FINISHED
 
+        # direct Pre NAK
+        flags = Msg.Flags(Msg.Flags.Type.DIRECT_NAK, False)
+        msg = Msg.InpStandard(addr, addr, flags, 0x10, 0xFC)
+        r = handler.msg_received(proto, msg)
+        assert r == Msg.CONTINUE
+
         # unexpected
         flags = Msg.Flags(Msg.Flags.Type.ALL_LINK_BROADCAST, False)
         msg = Msg.InpStandard(addr, addr, flags, 0x10, 0x00)

diff --git a/tests/handler/test_DeviceDbGet.py b/tests/handler/test_DeviceDbGet.py
@@ -36,6 +36,12 @@ def callback(success, msg, value):
         r = handler.msg_received(proto, std_ack)
         assert r == Msg.UNKNOWN
 
+        # direct Pre NAK
+        flags = Msg.Flags(Msg.Flags.Type.DIRECT_NAK, False)
+        msg = Msg.InpStandard(addr, addr, flags, 0x2f, 0xFC)
+        r = handler.msg_received(proto, msg)
+        assert r == Msg.CONTINUE
+
         # Try w/ an extended msg.
         ext_data = bytes(14)
         ext_ack = Msg.OutExtended.direct(addr, 0x2f, 0x00, ext_data)

diff --git a/tests/handler/test_ExtendedCmdResponse.py b/tests/handler/test_ExtendedCmdResponse.py
@@ -69,6 +69,12 @@ def callback(msg, on_done=None):
         r = handler.msg_received(proto, msg)
         assert r == Msg.FINISHED
 
+        # direct Pre NAK
+        flags = Msg.Flags(Msg.Flags.Type.DIRECT_NAK, False)
+        msg = Msg.InpStandard(addr, addr, flags, 0x2e, 0xFC)
+        r = handler.msg_received(proto, msg)
+        assert r == Msg.CONTINUE
+
         # unexpected
         flags = Msg.Flags(Msg.Flags.Type.BROADCAST, False)
         msg = Msg.InpStandard(addr, addr, flags, 0x2e, 0x00)