pytrap: add edge cases test for copyMessage()

Author: cejkato2

pytrap/test/unirectemplate_unittest.py

@@ -890,6 +890,49 @@ def runTest(self):
             pytrap.UnirecIPAddr("10.0.0.1"), "TIME_FIRST": pytrap.UnirecTime(1669885132, 853),
             "ABC": 123, "TEXT2": "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaa", "STREAMBYTES2": b"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaa", "UNKNOWN": "unknown text"})
 
+class CopyMessageEdgeCasesTest(unittest.TestCase):
+    def runTest(self):
+        import pytrap
+        data = { "SRC_IP": pytrap.UnirecIPAddr("10.0.0.1"), "DST_IP": pytrap.UnirecIPAddr("10.0.0.1"),
+                  "TIME_FIRST": pytrap.UnirecTime(1669885132, 853),
+                  "TIME_LAST": pytrap.UnirecTime(1669885132, 853),
+                  "ABC": 123, "BCD": 321, "TEXT": "some_text", "TEXT2": "some_text2",
+                  "STREAMBYTES": b"abc\x01\x02\x00\x03", "STREAMBYTES2": b"abc\x02\x03\x00\x04"}
+        a = pytrap.UnirecTemplate("ipaddr DST_IP,ipaddr SRC_IP,time TIME_FIRST,time TIME_LAST,uint32 ABC,uint32 BCD,string TEXT,string TEXT2,bytes STREAMBYTES,bytes STREAMBYTES2")
+        b = pytrap.UnirecTemplate("ipaddr SRC_IP,time TIME_FIRST,uint32 ABC,string TEXT2,bytes STREAMBYTES2,string UNKNOWN")
+        with self.assertRaises(TypeError):
+            # both a and b have no allocated data
+            b.copyMessage(a)
+        a.createMessage(100)
+        a.setFromDict(data)
+        with self.assertRaises(TypeError):
+            # b has no allocated data
+            b.copyMessage(a)
+
+        b.createMessage(100)
+        # recreate a
+        a = pytrap.UnirecTemplate("ipaddr DST_IP,ipaddr SRC_IP,time TIME_FIRST,time TIME_LAST,uint32 ABC,uint32 BCD,string TEXT,string TEXT2,bytes STREAMBYTES,bytes STREAMBYTES2")
+        with self.assertRaises(TypeError):
+            # a has no allocated data
+            b.copyMessage(a)
+
+        a.createMessage(100)
+        a.setFromDict(data)
+        b = pytrap.UnirecTemplate("")
+        b.createMessage()
+        b.copyMessage(a)
+        self.assertEqual(b.getDict(), {})
+
+        del(a)
+        del(b)
+
+        a = pytrap.UnirecTemplate("")
+        a.createMessage()
+        b = pytrap.UnirecTemplate("ipaddr DST_IP,time TIME_FIRST,string TEXT")
+        b.createMessage(100)
+        b.copyMessage(a)
+        self.assertEqual(b.getDict(), {"DST_IP": pytrap.UnirecIPAddr("::"), "TIME_FIRST": pytrap.UnirecTime(0, 0), "TEXT": ""})
+
 
 class AllocateBigMessage(unittest.TestCase):
     def runTest(self):