p34709852
/
monkey
forked from p15670423/monkey
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

Agent: Pass island_api_client when constructing AgentEventForwarder

This commit is contained in:
Ilija Lazoroski 2022-09-20 12:52:36 +02:00
parent b320fba2c8
commit 14592d964e
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monkey/infection_monkey/monkey.py
View File

@ -248,7 +248,7 @@ class InfectionMonkey:
) )
event_queue = PyPubSubAgentEventQueue(Publisher()) event_queue = PyPubSubAgentEventQueue(Publisher())
InfectionMonkey._subscribe_events( self._subscribe_events(
event_queue, event_queue,
propagation_credentials_repository, propagation_credentials_repository,
self._control_client.server_address, self._control_client.server_address,
@ -274,8 +274,8 @@ class InfectionMonkey:
propagation_credentials_repository, propagation_credentials_repository,
) )
@staticmethod
def _subscribe_events( def _subscribe_events(
self,
event_queue: IAgentEventQueue, event_queue: IAgentEventQueue,
propagation_credentials_repository: IPropagationCredentialsRepository, propagation_credentials_repository: IPropagationCredentialsRepository,
server_address: str, server_address: str,
@ -288,7 +288,7 @@ class InfectionMonkey:
), ),
) )
event_queue.subscribe_all_events( event_queue.subscribe_all_events(
AgentEventForwarder(server_address, agent_event_serializer_registry).send_event AgentEventForwarder(self.island_api_client, agent_event_serializer_registry).send_event
) )
def _build_puppet( def _build_puppet(