From 441c14f15dd9f4d824aa73e420b02dad1a829593 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Tue, 27 Sep 2022 12:56:10 +0530 Subject: [PATCH] Use SocketAddress.from_string() in network/relay/utils.py --- monkey/infection_monkey/network/relay/utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/monkey/infection_monkey/network/relay/utils.py b/monkey/infection_monkey/network/relay/utils.py index 737a07c9a..0f20a91e3 100644 --- a/monkey/infection_monkey/network/relay/utils.py +++ b/monkey/infection_monkey/network/relay/utils.py @@ -4,7 +4,6 @@ from contextlib import suppress from typing import Dict, Iterable, Iterator, Optional from common.common_consts.timeouts import LONG_REQUEST_TIMEOUT -from common.network.network_utils import address_to_ip_port from common.types import SocketAddress from infection_monkey.island_api_client import ( AbstractIslandAPIClientFactory, @@ -89,8 +88,7 @@ def send_remove_from_waitlist_control_message_to_relays(servers: Iterable[str]): def _send_remove_from_waitlist_control_message_to_relay(server: str): - server_address = SocketAddress(*address_to_ip_port(server)) - notify_disconnect(server_address) + notify_disconnect(SocketAddress.from_string(server)) def notify_disconnect(server_address: SocketAddress):