From a093a3e52751938287001fcb13880336f97299ae Mon Sep 17 00:00:00 2001
From: Shreya Malviya <shreya.malviya@gmail.com>
Date: Mon, 26 Sep 2022 16:49:42 +0530
Subject: [PATCH] Agent: Use default str representation of SocketAddress in log
 message in notify_disconnect()

---
 monkey/infection_monkey/network/relay/utils.py | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/monkey/infection_monkey/network/relay/utils.py b/monkey/infection_monkey/network/relay/utils.py
index f09a076f3..3eca4ac01 100644
--- a/monkey/infection_monkey/network/relay/utils.py
+++ b/monkey/infection_monkey/network/relay/utils.py
@@ -107,11 +107,6 @@ def notify_disconnect(server_address: SocketAddress):
         try:
             d_socket.connect((str(server_address.ip), server_address.port))
             d_socket.sendall(RELAY_CONTROL_MESSAGE_REMOVE_FROM_WAITLIST)
-            logger.info(
-                "Control message was sent to the server/relay "
-                f"{server_address.ip}:{server_address.port}"
-            )
+            logger.info(f"Control message was sent to the server/relay {server_address}")
         except OSError as err:
-            logger.error(
-                f"Error connecting to socket {server_address.ip}:{server_address.port}: {err}"
-            )
+            logger.error(f"Error connecting to socket {server_address}: {err}")