From 8c47d113c3322783b0e39baa76a2f1c10e78000f Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Wed, 8 Dec 2021 15:45:46 -0500 Subject: [PATCH] Agent: Add "options" parameter to IPuppet.ping() --- monkey/infection_monkey/i_puppet.py | 2 +- monkey/infection_monkey/puppet/mock_puppet.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/infection_monkey/i_puppet.py b/monkey/infection_monkey/i_puppet.py index d9d225b7b..03ce3999f 100644 --- a/monkey/infection_monkey/i_puppet.py +++ b/monkey/infection_monkey/i_puppet.py @@ -35,7 +35,7 @@ class IPuppet(metaclass=abc.ABCMeta): """ @abc.abstractmethod - def ping(self, host: str) -> Tuple[bool, Optional[str]]: + def ping(self, host: str, options: Dict) -> Tuple[bool, Optional[str]]: """ Sends a ping (ICMP packet) to a remote host :param str host: The domain name or IP address of a host diff --git a/monkey/infection_monkey/puppet/mock_puppet.py b/monkey/infection_monkey/puppet/mock_puppet.py index 3a32f3718..a606e7043 100644 --- a/monkey/infection_monkey/puppet/mock_puppet.py +++ b/monkey/infection_monkey/puppet/mock_puppet.py @@ -155,7 +155,7 @@ class MockPuppet(IPuppet): else: return PostBreachData("pba command 2", ["pba result 2", False]) - def ping(self, host: str) -> Tuple[bool, Optional[str]]: + def ping(self, host: str, options: Dict) -> Tuple[bool, Optional[str]]: logger.debug(f"run_ping({host})") if host == DOT_1: return (True, "windows")