From 2352bb0d5eac0e51d82608eb9c2fac4ee3513636 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Thu, 1 Sep 2022 17:01:05 +0530 Subject: [PATCH] BB: Fix configuration modification logic in zerologon.py --- .../blackbox/test_configurations/zerologon.py | 21 +++++++------------ 1 file changed, 7 insertions(+), 14 deletions(-) diff --git a/envs/monkey_zoo/blackbox/test_configurations/zerologon.py b/envs/monkey_zoo/blackbox/test_configurations/zerologon.py index 3a956c6f3..8e9a14369 100644 --- a/envs/monkey_zoo/blackbox/test_configurations/zerologon.py +++ b/envs/monkey_zoo/blackbox/test_configurations/zerologon.py @@ -1,13 +1,7 @@ from common.agent_configuration import AgentConfiguration, PluginConfiguration from .noop import noop_test_configuration -from .utils import ( - add_exploiters, - add_subnets, - add_tcp_ports, - replace_agent_configuration, - set_maximum_depth, -) +from .utils import add_exploiters, add_subnets, add_tcp_ports, set_maximum_depth def _add_exploiters(agent_configuration: AgentConfiguration) -> AgentConfiguration: @@ -27,11 +21,10 @@ def _add_subnets(agent_configuration: AgentConfiguration) -> AgentConfiguration: return add_subnets(agent_configuration, subnets) -test_configuration = set_maximum_depth(noop_test_configuration.agent_configuration, 1) -test_configuration = _add_exploiters(test_configuration) -test_configuration = _add_tcp_ports(test_configuration) -test_configuration = _add_subnets(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_tcp_ports(test_agent_configuration) +test_agent_configuration = _add_subnets(test_agent_configuration) -zerologon_test_configuration = replace_agent_configuration( - noop_test_configuration, test_configuration -) +zerologon_test_configuration = noop_test_configuration.copy() +zerologon_test_configuration.agent_configuration = test_agent_configuration