From 138ce81f1b0fe26cb30d3343039e4e541cf21fc3 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Mon, 18 Jul 2022 14:52:31 -0400 Subject: [PATCH] BB: Add add_credential_collectors() --- 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 c9ef2dd01..b84b51e0f 100644 --- a/envs/monkey_zoo/blackbox/test_configurations/utils.py +++ b/envs/monkey_zoo/blackbox/test_configurations/utils.py @@ -49,6 +49,12 @@ def add_subnets( return replace_scan_target_configuration(agent_configuration, scan_target_configuration) +def add_credential_collectors( + agent_configuration: AgentConfiguration, credential_collectors: Sequence[PluginConfiguration] +) -> AgentConfiguration: + return replace(agent_configuration, credential_collectors=tuple(credential_collectors)) + + def replace_exploitation_configuration( agent_configuration: AgentConfiguration, exploitation_configuration: ExploitationConfiguration ) -> AgentConfiguration: