diff --git a/monkey/tests/unit_tests/infection_monkey/master/test_ip_scanner.py b/monkey/tests/unit_tests/infection_monkey/master/test_ip_scanner.py index a9c7f601e..c55fdb85c 100644 --- a/monkey/tests/unit_tests/infection_monkey/master/test_ip_scanner.py +++ b/monkey/tests/unit_tests/infection_monkey/master/test_ip_scanner.py @@ -39,10 +39,10 @@ def scan_config(default_agent_configuration): PluginConfiguration(name="SSHFinger", options={}), ] scan_config = NetworkScanConfiguration( - tcp_config, - icmp_config, - fingerprinter_config, - default_agent_configuration.propagation.network_scan.targets, + tcp=tcp_config, + icmp=icmp_config, + fingerprinters=fingerprinter_config, + targets=default_agent_configuration.propagation.network_scan.targets, ) return scan_config diff --git a/monkey/tests/unit_tests/infection_monkey/master/test_propagator.py b/monkey/tests/unit_tests/infection_monkey/master/test_propagator.py index b8ceec5dd..53136e755 100644 --- a/monkey/tests/unit_tests/infection_monkey/master/test_propagator.py +++ b/monkey/tests/unit_tests/infection_monkey/master/test_propagator.py @@ -145,15 +145,15 @@ def get_propagation_config( default_agent_configuration, scan_target_config: ScanTargetConfiguration ): network_scan = NetworkScanConfiguration( - default_agent_configuration.propagation.network_scan.tcp, - default_agent_configuration.propagation.network_scan.icmp, - default_agent_configuration.propagation.network_scan.fingerprinters, - scan_target_config, + tcp=default_agent_configuration.propagation.network_scan.tcp, + icmp=default_agent_configuration.propagation.network_scan.icmp, + fingerprinters=default_agent_configuration.propagation.network_scan.fingerprinters, + targets=scan_target_config, ) propagation_config = PropagationConfiguration( - default_agent_configuration.propagation.maximum_depth, - network_scan, - default_agent_configuration.propagation.exploitation, + maximum_depth=default_agent_configuration.propagation.maximum_depth, + network_scan=network_scan, + exploitation=default_agent_configuration.propagation.exploitation, ) return propagation_config