Added defaults for prefix, suffix and verbose to tests, fixed broken import

Author: vortexau

tests/helpers/test_wordlist_helper.py

@@ -1,6 +1,6 @@
 import unittest
 import pytest
-from mock import patch
+from unittest.mock import patch
 
 from lib.helpers.wordlist_helper import WordList
 from lib.helpers.wordlist_helper import DEFAULT_WORDLIST_FILE

tests/test_input.py

@@ -32,6 +32,9 @@ def test_parse_arguments_default_value(tmpdir):
         'output_json': None,
         'output_grepable' : None,
         'ssl': False,
+        'prefix': False,
+        'suffix': False,
+        'verbose': False
     }
 
     assert vars(arguments) == expected_arguments
@@ -59,6 +62,9 @@ def test_parse_arguments_custom_arguments(tmpdir):
         '--user-agent', 'some-user-agent',
         '--waf',
         '-oN', '/tmp/on',
+        '--prefix','dev-',
+        '--suffix','test',
+        '-v'
     ]
 
     arguments = cli_argument_parser().parse(argv)
@@ -83,6 +89,9 @@ def test_parse_arguments_custom_arguments(tmpdir):
         'output_normal': '/tmp/on',
         'output_json': None,
         'output_grepable' : None,
+        'prefix': 'dev-',
+        'suffix': 'test',
+        'verbose': True
     }
 
     assert vars(arguments) == expected_arguments