diff --git a/envs/monkey_zoo/blackbox/test_configurations/smb_pth.py b/envs/monkey_zoo/blackbox/test_configurations/smb_pth.py index ec36bb4c7..8863013e9 100644 --- a/envs/monkey_zoo/blackbox/test_configurations/smb_pth.py +++ b/envs/monkey_zoo/blackbox/test_configurations/smb_pth.py @@ -6,8 +6,6 @@ from .utils import ( add_exploiters, add_subnets, add_tcp_ports, - replace_agent_configuration, - replace_propagation_credentials, set_keep_tunnel_open_time, set_maximum_depth, ) @@ -33,16 +31,11 @@ def _add_tcp_ports(agent_configuration: AgentConfiguration) -> AgentConfiguratio return add_tcp_ports(agent_configuration, ports) -test_configuration = set_maximum_depth(noop_test_configuration.agent_configuration, 3) -test_configuration = set_keep_tunnel_open_time(test_configuration, 20) -test_configuration = _add_exploiters(test_configuration) -test_configuration = _add_subnets(test_configuration) -test_configuration = _add_tcp_ports(test_configuration) - -smb_pth_test_configuration = replace_agent_configuration( - noop_test_configuration, test_configuration -) - +test_agent_configuration = set_maximum_depth(noop_test_configuration.agent_configuration, 3) +test_agent_configuration = set_keep_tunnel_open_time(test_agent_configuration, 20) +test_agent_configuration = _add_exploiters(test_agent_configuration) +test_agent_configuration = _add_subnets(test_agent_configuration) +test_agent_configuration = _add_tcp_ports(test_agent_configuration) CREDENTIALS = ( Credentials(Username("Administrator"), None), @@ -54,6 +47,7 @@ CREDENTIALS = ( Credentials(None, NTHash("5da0889ea2081aa79f6852294cba4a5e")), Credentials(None, NTHash("50c9987a6bf1ac59398df9f911122c9b")), ) -smb_pth_test_configuration = replace_propagation_credentials( - smb_pth_test_configuration, CREDENTIALS -) + +smb_pth_test_configuration = noop_test_configuration.copy() +smb_pth_test_configuration.agent_configuration = test_agent_configuration +smb_pth_test_configuration.propagation_credentials = CREDENTIALS