diff --git a/envs/monkey_zoo/blackbox/test_configurations/wmi_mimikatz.py b/envs/monkey_zoo/blackbox/test_configurations/wmi_mimikatz.py index fe4a1414e..c525f5ca5 100644 --- a/envs/monkey_zoo/blackbox/test_configurations/wmi_mimikatz.py +++ b/envs/monkey_zoo/blackbox/test_configurations/wmi_mimikatz.py @@ -7,8 +7,6 @@ from .utils import ( add_exploiters, add_subnets, add_tcp_ports, - replace_agent_configuration, - replace_propagation_credentials, set_maximum_depth, ) @@ -40,17 +38,12 @@ 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, 1) -test_configuration = _add_exploiters(test_configuration) -test_configuration = _add_subnets(test_configuration) -test_configuration = _add_credential_collectors(test_configuration) -test_configuration = _add_tcp_ports(test_configuration) -test_configuration = _add_credential_collectors(test_configuration) - -wmi_mimikatz_test_configuration = replace_agent_configuration( - noop_test_configuration, test_configuration -) - +test_agent_configuration = set_maximum_depth(noop_test_configuration.agent_configuration, 1) +test_agent_configuration = _add_exploiters(test_agent_configuration) +test_agent_configuration = _add_subnets(test_agent_configuration) +test_agent_configuration = _add_credential_collectors(test_agent_configuration) +test_agent_configuration = _add_tcp_ports(test_agent_configuration) +test_agent_configuration = _add_credential_collectors(test_agent_configuration) CREDENTIALS = ( Credentials(Username("Administrator"), None), @@ -59,6 +52,7 @@ CREDENTIALS = ( Credentials(None, Password("Ivrrw5zEzs")), Credentials(None, Password("Password1!")), ) -wmi_mimikatz_test_configuration = replace_propagation_credentials( - wmi_mimikatz_test_configuration, CREDENTIALS -) + +wmi_mimikatz_test_configuration = noop_test_configuration.copy() +wmi_mimikatz_test_configuration.agent_configuration = test_agent_configuration +wmi_mimikatz_test_configuration.propagation_credentials = CREDENTIALS