initial unittest

atlas0fd00m · Apr 3, 2019 · b90c174 · b90c174
1 parent f642745
commit b90c174
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 0 deletions.
diff --git a/tests/.test_basics.py.un~ b/tests/.test_basics.py.un~
diff --git a/tests/__init__.py b/tests/__init__.py
diff --git a/tests/test_basics.py b/tests/test_basics.py
@@ -0,0 +1,9 @@
+import unittest
+
+class RfCatBasicTests(unittest.TestCase):
+
+    def test_importing(self):
+        import rflib
+        devs = rflib.getRfCatDevices()
+        self.assertEquals(type(devs), list, "rflib.getRfCatDevices() doesn't return a list!: %r" % devs)
+
diff --git a/tests/test_basics.py~ b/tests/test_basics.py~
@@ -0,0 +1,16 @@
+import os
+import unittest
+
+import PE
+import vivisect.tests.helpers as helpers
+
+class RfCatBasicTests(unittest.TestCase):
+
+    def test_export_by_name(self):
+        file_path = helpers.getTestPath('windows', 'i386', 'export_by_name.dll')
+        pe = PE.peFromFileName(file_path)
+        export_list = pe.getExports()
+        self.assertEquals(len(export_list), 2, "expecting 2 exported functions")
+        self.assertEquals(export_list[0][1], 0, "exported function with ordinal 0 not found")
+        self.assertEquals(export_list[0][2], "Func1", "exported function with name 'Func1' not found")
+