1 files changed, 105 insertions, 0 deletions
diff --git a/test/suite_unittests.py b/test/suite_unittests.py
new file mode 100644
index 00000000..65abf9bc
--- /dev/null
+++ b/test/suite_unittests.py
@@ -0,0 +1,105 @@
+#
+# Wireshark tests
+# By
+# Gerald Combs <gerald@wireshark.org>
+# Gilbert Ramirez <gram [AT] alumni.rice.edu>
+#
+# Ported from a set of Bash scripts which were copyright 2005 Ulf Lamping
+#
+# SPDX-License-Identifier: GPL-2.0-or-later
+#
+'''EPAN unit tests'''
+
+import subprocess
+import pytest
+
+
+class TestUnitTests:
+    def test_unit_exntest(self, program, base_env):
+        '''exntest'''
+        subprocess.check_call(program('exntest'), env=base_env)
+
+    def test_unit_oids_test(self, program, base_env):
+        '''oids_test'''
+        subprocess.check_call(program('oids_test'), env=base_env)
+
+    def test_unit_reassemble_test(self, program, base_env):
+        '''reassemble_test'''
+        subprocess.check_call(program('reassemble_test'), env=base_env)
+
+    def test_unit_tvbtest(self, program, base_env):
+        '''tvbtest'''
+        subprocess.check_call(program('tvbtest'), env=base_env)
+
+    def test_unit_wmem_test(self, program, base_env):
+        '''wmem_test'''
+        subprocess.check_call((program('wmem_test'),
+            '--verbose'
+        ), env=base_env)
+
+    def test_unit_wscbor_test(self, program, base_env):
+        '''wscbor_test'''
+        subprocess.check_call(program('wscbor_test'), env=base_env)
+
+    def test_unit_epan(self, program, base_env):
+        '''epan unit tests'''
+        subprocess.check_call((program('test_epan'),
+            '--verbose'
+        ), env=base_env)
+
+    def test_unit_wsutil(self, program, base_env):
+        '''wsutil unit tests'''
+        subprocess.check_call((program('test_wsutil'),
+            '--verbose'
+        ), env=base_env)
+
+    def test_unit_fieldcount(self, cmd_tshark, test_env):
+        '''fieldcount'''
+        subprocess.check_call((cmd_tshark, '-G', 'fieldcount'), env=test_env)
+
+class Proto:
+    """Data for a protocol."""
+    def __init__(self, line):
+        data = line.split("\t")
+        assert len(data) == 3, "expected 3 columns in %s" % data
+        assert data[0] == "P"
+        self.name = data[1]
+        self.abbrev = data[2]
+
+class Field:
+    """Data for a field."""
+    def __init__(self, line):
+        data = line.split("\t")
+        assert len(data) == 8, "expected 8 columns in %s" % data
+        assert data[0] == "F"
+        self.name = data[1]
+        self.abbrev = data[2]
+        self.ftype = data[3]
+        self.parent = data[4]
+        self.base = data[5]
+        self.bitmask = int(data[6],0)
+        self.blurb = data[7]
+
+
+class TestUnitFtSanity:
+    def test_unit_ftsanity(self, cmd_tshark, base_env):
+        """Looks for problems in field type definitions."""
+        tshark_proc = subprocess.run((cmd_tshark, "-G", "fields"),
+                        check=True, capture_output=True, encoding='utf-8', env=base_env)
+
+        lines = tshark_proc.stdout.splitlines()
+        # XXX We don't currently check protos.
+        protos = [Proto(x) for x in lines if x[0] == "P"]
+        fields = [Field(x) for x in lines if x[0] == "F"]
+
+        err_list = []
+        for field in fields:
+            if field.bitmask != 0:
+                if field.ftype.find("FT_UINT") != 0 and \
+                        field.ftype.find("FT_INT") != 0 and \
+                        field.ftype != "FT_BOOLEAN" and \
+                        field.ftype != "FT_CHAR":
+                    err_list.append("%s has a bitmask 0x%x but is type %s" % \
+                            (field.abbrev, field.bitmask, field.ftype))
+
+        assert len(err_list) == 0, 'Found field type errors: \n' + '\n'.join(err_list)