From a2b8338ac78b322cb4259e59aea7767fae700cf0 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Thu, 1 Sep 2022 16:45:32 +0530 Subject: [PATCH] BB: Fix configuration modification logic in depth_1_a.py --- .../blackbox/test_configurations/depth_1_a.py | 26 ++++++++----------- 1 file changed, 11 insertions(+), 15 deletions(-) diff --git a/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py b/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py index abcf02b08..f702e3149 100644 --- a/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py +++ b/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py @@ -9,8 +9,6 @@ from .utils import ( add_http_ports, add_subnets, add_tcp_ports, - replace_agent_configuration, - replace_propagation_credentials, set_maximum_depth, ) @@ -76,22 +74,20 @@ def _add_http_ports(agent_configuration: AgentConfiguration) -> AgentConfigurati return add_http_ports(agent_configuration, HTTP_PORTS) -test_configuration = set_maximum_depth(noop_test_configuration.agent_configuration, 1) -test_configuration = _add_exploiters(test_configuration) -test_configuration = _add_fingerprinters(test_configuration) -test_configuration = _add_subnets(test_configuration) -test_configuration = _add_tcp_ports(test_configuration) -test_configuration = _add_credential_collectors(test_configuration) -test_configuration = _add_http_ports(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_fingerprinters(test_agent_configuration) +test_agent_configuration = _add_subnets(test_agent_configuration) +test_agent_configuration = _add_tcp_ports(test_agent_configuration) +test_agent_configuration = _add_credential_collectors(test_agent_configuration) +test_agent_configuration = _add_http_ports(test_agent_configuration) -depth_1_a_test_configuration = replace_agent_configuration( - noop_test_configuration, test_configuration -) CREDENTIALS = ( Credentials(Username("m0nk3y"), None), Credentials(None, Password("Ivrrw5zEzs")), Credentials(None, Password("Xk8VDTsC")), ) -depth_1_a_test_configuration = replace_propagation_credentials( - depth_1_a_test_configuration, CREDENTIALS -) + +depth_1_a_test_configuration = noop_test_configuration.copy() +depth_1_a_test_configuration.agent_configuration = test_agent_configuration +depth_1_a_test_configuration.propagation_credentials = CREDENTIALS