From ac633a6e75850eb80fda6497d1efbd3fc5e46611 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Mon, 26 Sep 2022 17:16:17 +0530 Subject: [PATCH] Agent: Change argument in InfectionMonkey._register_agent() to type SocketAddress --- monkey/infection_monkey/monkey.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/infection_monkey/monkey.py b/monkey/infection_monkey/monkey.py index e86bdbee1..071f742b1 100644 --- a/monkey/infection_monkey/monkey.py +++ b/monkey/infection_monkey/monkey.py @@ -131,7 +131,7 @@ class InfectionMonkey: server_address=server, island_api_client=self._island_api_client ) self._control_channel = ControlChannel(server, get_agent_id(), self._island_api_client) - self._register_agent(server) + self._register_agent(self._island_address) # TODO Refactor the telemetry messengers to accept control client # and remove control_client_object @@ -180,7 +180,7 @@ class InfectionMonkey: return server, island_api_client - def _register_agent(self, server: str): + def _register_agent(self, server: SocketAddress): agent_registration_data = AgentRegistrationData( id=get_agent_id(), machine_hardware_id=get_machine_id(),