Add some basic tests for the parse_arguments function

This function will be responsible from parsing the argv argument list
and build an object which encapsulates all the scan command parameters.

Author: diogoosorio

tests/test_input.py

@@ -0,0 +1,115 @@
+import argparse
+import pytest
+
+from lib.input import cli_argument_parser
+
+def test_parse_arguments_default_value(tmpdir):
+    words = ['word1', 'word2', 'word3']
+    wordlist = tmpdir.mkdir('test_command').join('default')
+    wordlist.write('\n'.join(words))
+
+    argv = ['-t', 'myhost']
+    
+    arguments = cli_argument_parser().parse(argv)
+
+    expected_arguments = {
+        'target_hosts': 'myhost',
+        'wordlists': None,
+        'base_host': False,
+        'port': 80,
+        'real_port': False,
+        'ignore_http_codes': '404',
+        'ignore_content_length': 0,
+        'unique_depth': 1,
+        'fuzzy_logic': False,
+        'no_lookup': False,
+        'rate_limit': 0,
+        'random_agent': False,
+        'user_agent': None,
+        'add_waf_bypass_headers': False,
+        'output_normal': None,
+        'output_json': None,
+        'stdin': False,
+        'ssl': False,
+    }
+    
+    assert vars(arguments) == expected_arguments
+
+
+def test_parse_arguments_custom_arguments(tmpdir):
+    words = ['some', 'other', 'words']
+    wordlist = tmpdir.mkdir('test_command').join('other_words')
+    wordlist.write('\n'.join(words))
+
+    argv = [
+        '-t', '10.11.1.1',
+        '-w', str(wordlist),
+        '-b', 'myhost',
+        '-p', '8000',
+        '-r', '8001',
+        '--ignore-http-codes', '400,500,302',
+        '--ignore-content-length', '100',
+        '--unique-depth', '5',
+        '--ssl',
+        '--fuzzy-logic',
+        '--no-lookups',
+        '--rate-limit', '10',
+        '--user-agent', 'some-user-agent',
+        '--waf',
+        '-oN', '/tmp/on',
+        '-',
+    ]
+
+    arguments = cli_argument_parser().parse(argv)
+
+    expected_arguments = {
+        'target_hosts': '10.11.1.1',
+        'wordlists': str(wordlist),
+        'base_host': 'myhost',
+        'port': 8000,
+        'real_port': 8001,
+        'ignore_http_codes': '400,500,302',
+        'ignore_content_length': 100,
+        'unique_depth': 5,
+        'ssl': True,
+        'fuzzy_logic': True,
+        'no_lookup': True,
+        'rate_limit': 10,
+        'user_agent': 'some-user-agent',
+        'random_agent': False,
+        'add_waf_bypass_headers': True,
+        'output_normal': '/tmp/on',
+        'output_json': None,
+        'stdin': True,
+    }
+
+    assert vars(arguments) == expected_arguments
+
+def test_parse_arguments_mutually_exclusive_user_agent():
+    argv = [
+        '-t', '10.11.1.1',
+        '--user-agent', 'my-user-agent',
+        '--random-agent',
+    ]
+
+    with pytest.raises(SystemExit):
+        cli_argument_parser().parse(argv)
+
+def test_parse_arguments_mutually_exclusive_output():
+    argv = [
+        '-t', '10.11.1.1',
+        '-oJ',
+        '-oN',
+    ]
+
+    with pytest.raises(SystemExit):
+        cli_argument_parser().parse(argv)
+
+def test_parse_arguments_unknown_argument():
+    argv = [
+        '-t', '10.11.1.1',
+        '-i-do-not-exist',
+    ]
+
+    with pytest.raises(SystemExit):
+        cli_argument_parser().parse(argv)