diff --git a/monkey/infection_monkey/master/mock_master.py b/monkey/infection_monkey/master/mock_master.py index fa4087e82..5c522a565 100644 --- a/monkey/infection_monkey/master/mock_master.py +++ b/monkey/infection_monkey/master/mock_master.py @@ -34,11 +34,11 @@ class MockMaster(IMaster): def _run_credential_collectors(self): logger.info("Running credential collectors") - windows_credentials = self._puppet.run_credential_collector("MimikatzCredentialCollector") + windows_credentials = self._puppet.run_credential_collector("MimikatzCollector") if windows_credentials: self._telemetry_messenger.send_telemetry(CredentialsTelem(windows_credentials)) - ssh_credentials = self._puppet.run_sys_info_collector("SSHCredentialCollector") + ssh_credentials = self._puppet.run_sys_info_collector("SSHCollector") if ssh_credentials: self._telemetry_messenger.send_telemetry(CredentialsTelem(ssh_credentials)) diff --git a/monkey/infection_monkey/puppet/mock_puppet.py b/monkey/infection_monkey/puppet/mock_puppet.py index a30cbf353..8b76d175a 100644 --- a/monkey/infection_monkey/puppet/mock_puppet.py +++ b/monkey/infection_monkey/puppet/mock_puppet.py @@ -30,7 +30,7 @@ class MockPuppet(IPuppet): def run_credential_collector(self, name: str, options: Dict) -> Sequence[Credentials]: logger.debug(f"run_credential_collector({name})") - if name == "SSHCredentialCollector": + if name == "SSHCollector": # TODO: Replace Passwords with SSHKeypair after it is implemented ssh_credentials = Credentials( [Username("m0nk3y")],