Commit f427143c by fwkz

Changes so tests can be run from PyCharm test runner.

parent fa14f730
from __future__ import print_function import unittest
from __future__ import absolute_import
try: try:
import unittest.mock as mock import unittest.mock as mock
except ImportError: except ImportError:
import mock import mock
from . import RoutersploitTestCase from routersploit.test import RoutersploitTestCase
from ..exploits import Exploit, Option from routersploit.exploits import Exploit, Option
def suffix(x): def suffix(x):
...@@ -67,3 +66,6 @@ class OptionTest(RoutersploitTestCase): ...@@ -67,3 +66,6 @@ class OptionTest(RoutersploitTestCase):
self.exploit_with_widgets.paa = "new_value" self.exploit_with_widgets.paa = "new_value"
self.assertEqual(self.exploit_with_widgets.paa, "new_value_suffix_SUFFIX") self.assertEqual(self.exploit_with_widgets.paa, "new_value_suffix_SUFFIX")
if __name__ == '__main__':
unittest.main()
from __future__ import print_function
import unittest import unittest
import os import os
......
from __future__ import print_function
from __future__ import absolute_import
import unittest import unittest
try: try:
......
from __future__ import print_function import unittest
from __future__ import absolute_import
try: try:
import unittest.mock as mock import unittest.mock as mock
except ImportError: except ImportError:
import mock import mock
from . import RoutersploitTestCase from routersploit.test import RoutersploitTestCase
from .. import widgets from routersploit import widgets
class WidgetsTest(RoutersploitTestCase): class WidgetsTest(RoutersploitTestCase):
...@@ -19,3 +18,7 @@ class WidgetsTest(RoutersploitTestCase): ...@@ -19,3 +18,7 @@ class WidgetsTest(RoutersploitTestCase):
def test_url_already_with_https_prefix(self): def test_url_already_with_https_prefix(self):
self.assertEqual(widgets.url("https://127.0.0.1"), "https://127.0.0.1") self.assertEqual(widgets.url("https://127.0.0.1"), "https://127.0.0.1")
if __name__ == '__main__':
unittest.main()
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment