From a0d0f127f9c7587cfbbcdecef38b97b9b939ffeb Mon Sep 17 00:00:00 2001
From: Shreya Malviya <shreya.malviya@gmail.com>
Date: Thu, 1 Sep 2022 16:58:42 +0530
Subject: [PATCH] BB: Fix configuration modification logic in smb_pth.py

---
 .../blackbox/test_configurations/smb_pth.py   | 24 +++++++------------
 1 file changed, 9 insertions(+), 15 deletions(-)

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