From 441c14f15dd9f4d824aa73e420b02dad1a829593 Mon Sep 17 00:00:00 2001
From: Shreya Malviya <shreya.malviya@gmail.com>
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):