p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

BB: Add replace_propagation_credentials()

This commit is contained in:
Mike Salvatore 2022-07-18 14:52:05 -04:00
parent d11fbe92e8
commit dbc138d263
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
envs/monkey_zoo/blackbox/test_configurations/utils.py
View File

@ -1,5 +1,5 @@
from dataclasses import replace from dataclasses import replace
from typing import Sequence from typing import Sequence, Tuple
from common.configuration import ( from common.configuration import (
AgentConfiguration, AgentConfiguration,
@ -9,6 +9,7 @@ from common.configuration import (
PropagationConfiguration, PropagationConfiguration,
ScanTargetConfiguration, ScanTargetConfiguration,
) )
from common.credentials import Credentials
from . import TestConfiguration from . import TestConfiguration
@ -87,3 +88,9 @@ def replace_agent_configuration(
test_configuration: TestConfiguration, agent_configuration: AgentConfiguration test_configuration: TestConfiguration, agent_configuration: AgentConfiguration
) -> TestConfiguration: ) -> TestConfiguration:
return replace(test_configuration, agent_configuration=agent_configuration) return replace(test_configuration, agent_configuration=agent_configuration)
def replace_propagation_credentials(
test_configuration: TestConfiguration, propagation_credentials: Tuple[Credentials, ...]
):
return replace(test_configuration, propagation_credentials=propagation_credentials)