Commit f427143c by fwkz

Changes so tests can be run from PyCharm test runner.

parent fa14f730
from __future__ import print_function
from __future__ import absolute_import
import unittest
try:
import unittest.mock as mock
except ImportError:
import mock
from . import RoutersploitTestCase
from ..exploits import Exploit, Option
from routersploit.test import RoutersploitTestCase
from routersploit.exploits import Exploit, Option
def suffix(x):
......@@ -67,3 +66,6 @@ class OptionTest(RoutersploitTestCase):
self.exploit_with_widgets.paa = "new_value"
self.assertEqual(self.exploit_with_widgets.paa, "new_value_suffix_SUFFIX")
if __name__ == '__main__':
unittest.main()
from __future__ import print_function
import unittest
import os
......
from __future__ import print_function
from __future__ import absolute_import
import unittest
try:
......
from __future__ import print_function
from __future__ import absolute_import
import unittest
try:
import unittest.mock as mock
except ImportError:
import mock
from . import RoutersploitTestCase
from .. import widgets
from routersploit.test import RoutersploitTestCase
from routersploit import widgets
class WidgetsTest(RoutersploitTestCase):
......@@ -19,3 +18,7 @@ class WidgetsTest(RoutersploitTestCase):
def test_url_already_with_https_prefix(self):
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