p34709852
/
monkey
forked from p15670423/monkey
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

Common: Add maximum_depth to PropagationConfiguration

This commit is contained in:
Mike Salvatore 2022-06-17 09:07:13 -04:00
parent a41b2e3ea4
commit bd7ea7fdb1
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monkey/common/configuration/agent_configuration.py
View File

@ -156,11 +156,13 @@ class ExploitationConfigurationSchema(Schema):
@dataclass(frozen=True)
class PropagationConfiguration:
maximum_depth: int
network_scan: NetworkScanConfiguration
exploitation: ExploitationConfiguration
class PropagationConfigurationSchema(Schema):
maximum_depth = fields.Int()
network_scan = fields.Nested(NetworkScanConfigurationSchema)
exploitation = fields.Nested(ExploitationConfigurationSchema)

3
monkey/tests/unit_tests/common/test_agent_configuration.py
View File

@ -181,6 +181,7 @@ def test_exploitation_configuration():
PROPAGATION_CONFIGURATION = {
"maximum_depth": 5,
"network_scan": NETWORK_SCAN_CONFIGURATION,
"exploitation": EXPLOITATION_CONFIGURATION,
}
@ -195,5 +196,5 @@ def test_propagation_configuration():
assert isinstance(config, PropagationConfiguration)
assert isinstance(config.network_scan, NetworkScanConfiguration)
assert isinstance(config.exploitation, ExploitationConfiguration)
assert config.maximum_depth == 5
assert config_dict == PROPAGATION_CONFIGURATION