fix none args.modules case

bretfourbe · Nov 21, 2023 · cd6e9e1 · cd6e9e1
1 parent 5cd689f
commit cd6e9e1
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/tests/cli/test_options.py b/tests/cli/test_options.py
@@ -203,3 +203,17 @@ async def test_is_mod_cms_set(mock_is_mod_cms_set, _, __):
     with mock.patch.object(sys, "argv", testargs):
         await wapiti_main()
         mock_is_mod_cms_set.assert_called_once()
+
+
+@pytest.mark.asyncio
+@mock.patch("wapitiCore.main.wapiti.Wapiti.browse")
+@mock.patch("wapitiCore.main.wapiti.Wapiti.attack")
+async def test_basic_usage(_, __):
+    """Test without option"""
+    testsagrs = [
+        "wapiti",
+        "--url", "http://testphp.vulnweb.com/"
+    ]
+
+    with mock.patch.object(sys, "argv", testsagrs):
+        await wapiti_main()
diff --git a/wapitiCore/main/wapiti.py b/wapitiCore/main/wapiti.py
@@ -74,7 +74,7 @@ def is_valid_endpoint(url_type, url: str):
 
 
 def is_mod_cms_set(args):
-    if "cms" in args.modules:
+    if args.modules and "cms" in args.modules:
         return True
     logging.error("Error: Invalid option --cms, module cms is required when this option is used")
     return False
@@ -301,7 +301,7 @@ async def wapiti_main():
                 )
             attack_options["cms"] = args.cms
 
-        if "cms" in args.modules and not args.cms:
+        if args.modules and "cms" in args.modules and not args.cms:
             attack_options["cms"] = "drupal,joomla,prestashop"
 
         if args.skipped_parameters: