From 6e730394bf71c7ad15ed4a56eda3ba9fcbc61b72 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Tue, 19 Jul 2022 13:24:00 -0400 Subject: [PATCH] BB: Add set_keep_tunnel_open_time() --- envs/monkey_zoo/blackbox/test_configurations/utils.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/envs/monkey_zoo/blackbox/test_configurations/utils.py b/envs/monkey_zoo/blackbox/test_configurations/utils.py index f1191f59a..eef23b329 100644 --- a/envs/monkey_zoo/blackbox/test_configurations/utils.py +++ b/envs/monkey_zoo/blackbox/test_configurations/utils.py @@ -69,6 +69,12 @@ def add_http_ports( ) +def set_keep_tunnel_open_time( + agent_configuration: AgentConfiguration, keep_tunnel_open_time: int +) -> AgentConfiguration: + return replace(agent_configuration, keep_tunnel_open_time=keep_tunnel_open_time) + + def set_maximum_depth( agent_configuration: AgentConfiguration, maximum_depth: int ) -> AgentConfiguration: