From aed9022a7ae68d3682d7f99a391f01c97f1df000 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Tue, 20 Sep 2022 14:34:11 -0400 Subject: [PATCH] Island: Register reset_machine_repository for CLEAR_SIMULATION_DATA --- .../monkey_island/cc/setup/island_event_handlers.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/monkey/monkey_island/cc/setup/island_event_handlers.py b/monkey/monkey_island/cc/setup/island_event_handlers.py index 859483ee9..a69cf1731 100644 --- a/monkey/monkey_island/cc/setup/island_event_handlers.py +++ b/monkey/monkey_island/cc/setup/island_event_handlers.py @@ -2,12 +2,14 @@ from functools import partial from common import DIContainer from monkey_island.cc.event_queue import IIslandEventQueue, IslandEventTopic -from monkey_island.cc.island_event_handlers import reset_agent_configuration +from monkey_island.cc.island_event_handlers import ( + reset_agent_configuration, + reset_machine_repository, +) from monkey_island.cc.repository import ( IAgentEventRepository, IAgentRepository, ICredentialsRepository, - IMachineRepository, INodeRepository, ) from monkey_island.cc.services.database import Database @@ -39,11 +41,14 @@ def _subscribe_clear_simulation_data_events( IslandEventTopic.CLEAR_SIMULATION_DATA, credentials_repository.remove_stolen_credentials ) + island_event_queue.subscribe( + IslandEventTopic.CLEAR_SIMULATION_DATA, container.resolve(reset_machine_repository) + ) + for i_repository in [ INodeRepository, IAgentEventRepository, IAgentRepository, - IMachineRepository, ]: repository = container.resolve(i_repository) island_event_queue.subscribe(IslandEventTopic.CLEAR_SIMULATION_DATA, repository.reset)