From 27c8a1019b95143f5be345d330d82b4007781910 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Tue, 6 Sep 2022 17:59:17 +0530 Subject: [PATCH] Island: Register IIslandEventQueue instance in DI container --- monkey/monkey_island/cc/services/initialize.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/monkey/monkey_island/cc/services/initialize.py b/monkey/monkey_island/cc/services/initialize.py index b1caab657..5f5fc76c1 100644 --- a/monkey/monkey_island/cc/services/initialize.py +++ b/monkey/monkey_island/cc/services/initialize.py @@ -14,6 +14,7 @@ from common.aws import AWSInstance from common.common_consts.telem_categories import TelemCategoryEnum from common.event_queue import IAgentEventQueue, PyPubSubAgentEventQueue from common.utils.file_utils import get_binary_io_sha256_hash +from monkey.common.event_queue import IIslandEventQueue, PyPubSubIslandEventQueue from monkey_island.cc.repository import ( AgentBinaryRepository, FileAgentConfigurationRepository, @@ -64,6 +65,7 @@ def initialize_services(container: DIContainer, data_dir: Path): ) container.register(Publisher, Publisher) container.register_instance(IAgentEventQueue, container.resolve(PyPubSubAgentEventQueue)) + container.register_instance(IIslandEventQueue, container.resolve(PyPubSubIslandEventQueue)) _register_repositories(container, data_dir) _register_services(container)