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 ffd9ed882..d29451568 100644 --- a/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py +++ b/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py @@ -6,6 +6,7 @@ from .utils import ( add_credential_collectors, add_exploiters, add_subnets, + add_tcp_ports, replace_agent_configuration, replace_propagation_credentials, set_maximum_depth, @@ -54,6 +55,11 @@ def _add_credential_collectors(agent_configuration: AgentConfiguration) -> Agent ) +def _add_tcp_ports(agent_configuration: AgentConfiguration) -> AgentConfiguration: + ports = [445, 8088, 22] + return add_tcp_ports(agent_configuration, ports) + + agent_configuration = set_maximum_depth(noop_test_configuration.agent_configuration, 1) agent_configuration = _add_exploiters(agent_configuration) agent_configuration = _add_subnets(agent_configuration)