BB: Fix configuration modification logic in all test configurations since TestConfiguration is a dataclass

envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py

@@ -1,3 +1,5 @@
+import dataclasses
+
 from common.agent_configuration import AgentConfiguration, PluginConfiguration
 from common.credentials import Credentials, Password, Username
 
@@ -90,7 +92,7 @@ CREDENTIALS = (
     Credentials(None, Password("Xk8VDTsC")),
 )
 
-depth_1_a_test_configuration = noop_test_configuration.copy()
+depth_1_a_test_configuration = dataclasses.replace(noop_test_configuration)
 replace_agent_configuration(
     test_configuration=depth_1_a_test_configuration, agent_configuration=test_agent_configuration
 )

envs/monkey_zoo/blackbox/test_configurations/depth_2_a.py

@@ -1,3 +1,5 @@
+import dataclasses
+
 from common.agent_configuration import AgentConfiguration, PluginConfiguration
 from common.credentials import Credentials, Password, Username
 
@@ -44,7 +46,7 @@ CREDENTIALS = (
     Credentials(None, Password("^NgDvY59~8")),
 )
 
-depth_2_a_test_configuration = noop_test_configuration.copy()
+depth_2_a_test_configuration = dataclasses.replace(noop_test_configuration)
 replace_agent_configuration(
     test_configuration=depth_2_a_test_configuration, agent_configuration=test_agent_configuration
 )

envs/monkey_zoo/blackbox/test_configurations/depth_3_a.py

@@ -1,3 +1,5 @@
+import dataclasses
+
 from common.agent_configuration import AgentConfiguration, PluginConfiguration
 from common.credentials import Credentials, NTHash, Password, Username
 
@@ -66,7 +68,7 @@ CREDENTIALS = (
     Credentials(None, NTHash("50c9987a6bf1ac59398df9f911122c9b")),
 )
 
-depth_3_a_test_configuration = noop_test_configuration.copy()
+depth_3_a_test_configuration = dataclasses.replace(noop_test_configuration)
 replace_agent_configuration(
     test_configuration=depth_3_a_test_configuration, agent_configuration=test_agent_configuration
 )

envs/monkey_zoo/blackbox/test_configurations/powershell_credentials_reuse.py

@@ -1,3 +1,5 @@
+import dataclasses
+
 from common.agent_configuration import AgentConfiguration, PluginConfiguration
 
 from .noop import noop_test_configuration
@@ -35,7 +37,7 @@ 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)
 
-powershell_credentials_reuse_test_configuration = noop_test_configuration.copy()
+powershell_credentials_reuse_test_configuration = dataclasses.replace(noop_test_configuration)
 replace_agent_configuration(
     test_configuration=powershell_credentials_reuse_test_configuration,
     agent_configuration=test_agent_configuration,

envs/monkey_zoo/blackbox/test_configurations/smb_pth.py

@@ -1,3 +1,5 @@
+import dataclasses
+
 from common.agent_configuration import AgentConfiguration, PluginConfiguration
 from common.credentials import Credentials, NTHash, Password, Username
 
@@ -50,7 +52,7 @@ CREDENTIALS = (
     Credentials(None, NTHash("50c9987a6bf1ac59398df9f911122c9b")),
 )
 
-smb_pth_test_configuration = noop_test_configuration.copy()
+smb_pth_test_configuration = dataclasses.replace(noop_test_configuration)
 replace_agent_configuration(
     test_configuration=smb_pth_test_configuration, agent_configuration=test_agent_configuration
 )

envs/monkey_zoo/blackbox/test_configurations/wmi_mimikatz.py

@@ -1,3 +1,5 @@
+import dataclasses
+
 from common.agent_configuration import AgentConfiguration, PluginConfiguration
 from common.credentials import Credentials, Password, Username
 
@@ -55,7 +57,7 @@ CREDENTIALS = (
     Credentials(None, Password("Password1!")),
 )
 
-wmi_mimikatz_test_configuration = noop_test_configuration.copy()
+wmi_mimikatz_test_configuration = dataclasses.replace(noop_test_configuration)
 replace_agent_configuration(
     test_configuration=wmi_mimikatz_test_configuration, agent_configuration=test_agent_configuration
 )

envs/monkey_zoo/blackbox/test_configurations/zerologon.py

@@ -1,3 +1,5 @@
+import dataclasses
+
 from common.agent_configuration import AgentConfiguration, PluginConfiguration
 
 from .noop import noop_test_configuration
@@ -32,7 +34,7 @@ test_agent_configuration = _add_exploiters(test_agent_configuration)
 test_agent_configuration = _add_tcp_ports(test_agent_configuration)
 test_agent_configuration = _add_subnets(test_agent_configuration)
 
-zerologon_test_configuration = noop_test_configuration.copy()
+zerologon_test_configuration = dataclasses.replace(noop_test_configuration)
 replace_agent_configuration(
     test_configuration=zerologon_test_configuration, agent_configuration=test_agent_configuration
 )