From f93565b2cfcf69d4e413f7f490131a84c19b2ed2 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Tue, 20 Sep 2022 15:50:43 +0530 Subject: [PATCH] Island: Subscribe repository reset functions to CLEAR_SIMULATION_DATA event --- .../cc/setup/island_event_handlers.py | 22 ++++++++++++++++++- 1 file changed, 21 insertions(+), 1 deletion(-) diff --git a/monkey/monkey_island/cc/setup/island_event_handlers.py b/monkey/monkey_island/cc/setup/island_event_handlers.py index ab3e856d6..ea32e8e65 100644 --- a/monkey/monkey_island/cc/setup/island_event_handlers.py +++ b/monkey/monkey_island/cc/setup/island_event_handlers.py @@ -3,7 +3,13 @@ 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.repository import ICredentialsRepository +from monkey_island.cc.repository import ( + IAgentEventRepository, + IAgentRepository, + ICredentialsRepository, + IMachineRepository, + INodeRepository, +) from monkey_island.cc.services.database import Database @@ -32,3 +38,17 @@ def _subscribe_clear_simulation_data_events( island_event_queue.subscribe( IslandEventTopic.CLEAR_SIMULATION_DATA, credentials_repository.remove_stolen_credentials ) + + node_repository = container.resolve(INodeRepository) + island_event_queue.subscribe(IslandEventTopic.CLEAR_SIMULATION_DATA, node_repository.reset) + + agent_event_repository = container.resolve(IAgentEventRepository) + island_event_queue.subscribe( + IslandEventTopic.CLEAR_SIMULATION_DATA, agent_event_repository.reset + ) + + agent_repository = container.resolve(IAgentRepository) + island_event_queue.subscribe(IslandEventTopic.CLEAR_SIMULATION_DATA, agent_repository.reset) + + machine_repository = container.resolve(IMachineRepository) + island_event_queue.subscribe(IslandEventTopic.CLEAR_SIMULATION_DATA, machine_repository.reset)