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

Use SocketAddress.from_string() in network/relay/utils.py

This commit is contained in:
Shreya Malviya 2022-09-27 12:56:10 +05:30
parent 1f80eac4b6
commit 441c14f15d
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
monkey/infection_monkey/network/relay/utils.py
View File

@ -4,7 +4,6 @@ from contextlib import suppress
from typing import Dict, Iterable, Iterator, Optional from typing import Dict, Iterable, Iterator, Optional
from common.common_consts.timeouts import LONG_REQUEST_TIMEOUT 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 common.types import SocketAddress
from infection_monkey.island_api_client import ( from infection_monkey.island_api_client import (
AbstractIslandAPIClientFactory, 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): def _send_remove_from_waitlist_control_message_to_relay(server: str):
server_address = SocketAddress(*address_to_ip_port(server)) notify_disconnect(SocketAddress.from_string(server))
notify_disconnect(server_address)
def notify_disconnect(server_address: SocketAddress): def notify_disconnect(server_address: SocketAddress):