Common: Remove sub-configuration and sub-schemas from configuration

monkey/common/configuration/__init__.py

@@ -1,29 +1,3 @@
-from .agent_sub_configurations import (
-    PluginConfiguration,
-    CustomPBAConfiguration,
-    ICMPScanConfiguration,
-    TCPScanConfiguration,
-    ScanTargetConfiguration,
-    NetworkScanConfiguration,
-    ExploitationOptionsConfiguration,
-    ExploiterConfiguration,
-    ExploitationConfiguration,
-    PropagationConfiguration,
-)
-
-from .agent_sub_configuration_schemas import (
-    PluginConfigurationSchema,
-    CustomPBAConfigurationSchema,
-    ICMPScanConfigurationSchema,
-    TCPScanConfigurationSchema,
-    ScanTargetConfigurationSchema,
-    NetworkScanConfigurationSchema,
-    ExploitationOptionsConfigurationSchema,
-    ExploiterConfigurationSchema,
-    ExploitationConfigurationSchema,
-    PropagationConfigurationSchema,
-)
-
 from .agent_configuration import (
     AgentConfiguration,
     AgentConfigurationSchema,

monkey/tests/unit_tests/common/test_agent_configuration.py

@@ -1,23 +1,24 @@
 from common import OperatingSystems
-from common.configuration import (
-    AgentConfiguration,
-    AgentConfigurationSchema,
-    CustomPBAConfiguration,
+from common.configuration import AgentConfiguration, AgentConfigurationSchema
+from common.configuration.agent_sub_configuration_schemas import (
     CustomPBAConfigurationSchema,
-    ExploitationConfiguration,
     ExploitationConfigurationSchema,
     ExploitationOptionsConfigurationSchema,
     ExploiterConfigurationSchema,
     ICMPScanConfigurationSchema,
-    NetworkScanConfiguration,
     NetworkScanConfigurationSchema,
-    PluginConfiguration,
     PluginConfigurationSchema,
-    PropagationConfiguration,
     PropagationConfigurationSchema,
     ScanTargetConfigurationSchema,
     TCPScanConfigurationSchema,
 )
+from common.configuration.agent_sub_configurations import (
+    CustomPBAConfiguration,
+    ExploitationConfiguration,
+    NetworkScanConfiguration,
+    PluginConfiguration,
+    PropagationConfiguration,
+)
 
 NAME = "bond"
 OPTIONS = {"gun": "Walther PPK", "car": "Aston Martin DB5"}