Agent: Fix capitalization of Dict type hints in IPuppet

monkey/infection_monkey/i_puppet.py

@@ -62,7 +62,7 @@ class IPuppet(metaclass=abc.ABCMeta):
         """
 
     @abc.abstractmethod
-    def exploit_host(self, name: str, host: str, options: dict, interrupt: threading.Event) -> bool:
+    def exploit_host(self, name: str, host: str, options: Dict, interrupt: threading.Event) -> bool:
         """
         Runs an exploiter against a remote host
         :param str name: The name of the exploiter to run
@@ -74,7 +74,7 @@ class IPuppet(metaclass=abc.ABCMeta):
         """
 
     @abc.abstractmethod
-    def run_payload(self, name: str, options: dict, interrupt: threading.Event) -> None:
+    def run_payload(self, name: str, options: Dict, interrupt: threading.Event) -> None:
         """
         Runs a payload
         :param str name: The name of the payload to run

monkey/infection_monkey/puppet/mock_puppet.py

@@ -141,7 +141,7 @@ class MockPuppet(IPuppet):
 
         return {}
 
-    def run_pba(self, name: str, options: dict) -> None:
+    def run_pba(self, name: str, options: Dict) -> None:
         logger.debug(f"run_pba({name}, {options})")
         return None
 
@@ -206,13 +206,13 @@ class MockPuppet(IPuppet):
 
         return {}
 
-    def exploit_host(self, name: str, host: str, options: dict, interrupt: threading.Event) -> bool:
+    def exploit_host(self, name: str, host: str, options: Dict, interrupt: threading.Event) -> bool:
         logger.debug(f"exploit_hosts({name}, {host}, {options})")
         successful_exploiters = {DOT_1: {"PowerShellExploiter"}, DOT_3: {"SSHExploiter"}}
 
         return name in successful_exploiters.get(host, {})
 
-    def run_payload(self, name: str, options: dict, interrupt: threading.Event) -> None:
+    def run_payload(self, name: str, options: Dict, interrupt: threading.Event) -> None:
         logger.debug(f"run_payload({name}, {options})")
         return None