generate_shellcode unittest

knownsec · Apr 22, 2019 · 6507473 · 6507473
1 parent b85ee16
commit 6507473
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/tests/test_generate_shellcode_list.py b/tests/test_generate_shellcode_list.py
@@ -0,0 +1,18 @@
+import unittest
+
+from pocsuite3.lib.core.enums import OS, OS_ARCH
+from pocsuite3.lib.utils import generate_shellcode_list
+
+
+class TestCase(unittest.TestCase):
+    def setUp(self):
+        pass
+
+    def test_windows_shellcode(self):
+        _list = generate_shellcode_list(listener_ip='127.0.0.1', listener_port=8088)
+        self.assertTrue(len(_list) > 0)
+
+    def test_linux_shellcode(self):
+        _list = generate_shellcode_list(listener_ip='127.0.0.1', listener_port=8088, os_target=OS.LINUX,
+                                        os_target_arch=OS_ARCH.X86)
+        self.assertTrue(len(_list) > 0)