From 311807cf310bb2be9a5a215518a9cbfac12f48ab Mon Sep 17 00:00:00 2001 From: Kekoa Kaaikala Date: Tue, 6 Sep 2022 12:48:19 +0000 Subject: [PATCH] Agent: Update RelayUserHandler docstrings --- monkey/infection_monkey/network/relay/relay_user_handler.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/monkey/infection_monkey/network/relay/relay_user_handler.py b/monkey/infection_monkey/network/relay/relay_user_handler.py index bce43cc4e..8db32a981 100644 --- a/monkey/infection_monkey/network/relay/relay_user_handler.py +++ b/monkey/infection_monkey/network/relay/relay_user_handler.py @@ -14,6 +14,8 @@ class RelayUser: class RelayUserHandler: + """Manages membership to a network relay.""" + def __init__(self, new_client_timeout: float = DEFAULT_NEW_CLIENT_TIMEOUT): self._new_client_timeout = new_client_timeout self._relay_users: Dict[IPv4Address, RelayUser] = {} @@ -37,7 +39,7 @@ class RelayUserHandler: def add_potential_user(self, user_address: IPv4Address): """ - Notify TCPRelay that a new user may try and connect + Notify RelayUserHandler that a new user may try and connect. :param user_address: An address defining potential RelayUser that tries to connect to the relay