BB: Add missing type hints to add_exploiters()

This commit is contained in:
Mike Salvatore 2022-07-18 14:49:25 -04:00
parent 37e79f41e8
commit d11fbe92e8
1 changed files with 4 additions and 1 deletions

View File

@ -5,6 +5,7 @@ from common.configuration import (
AgentConfiguration, AgentConfiguration,
ExploitationConfiguration, ExploitationConfiguration,
NetworkScanConfiguration, NetworkScanConfiguration,
PluginConfiguration,
PropagationConfiguration, PropagationConfiguration,
ScanTargetConfiguration, ScanTargetConfiguration,
) )
@ -13,7 +14,9 @@ from . import TestConfiguration
def add_exploiters( def add_exploiters(
agent_configuration: AgentConfiguration, brute_force=[], vulnerability=[] agent_configuration: AgentConfiguration,
brute_force: Sequence[PluginConfiguration] = [],
vulnerability: Sequence[PluginConfiguration] = [],
) -> AgentConfiguration: ) -> AgentConfiguration:
exploitation_configuration = replace( exploitation_configuration = replace(
agent_configuration.propagation.exploitation, agent_configuration.propagation.exploitation,