Agent: move mimikatz collector to credential collectors

monkey/infection_monkey/credential_collectors/mimikatz_collector/mimikatz_cred_collector.py

@@ -1,5 +1,7 @@
 from typing import List
 
+import pypykatz_handler
+
 from infection_monkey.credential_collectors import (
     Credentials,
     ICredentialCollector,
@@ -7,10 +9,8 @@ from infection_monkey.credential_collectors import (
     Password,
     Username,
 )
-from infection_monkey.system_info.windows_cred_collector import pypykatz_handler
-from infection_monkey.system_info.windows_cred_collector.windows_credentials import (
-    WindowsCredentials,
-)
+
+from .windows_credentials import WindowsCredentials
 
 
 class MimikatzCredentialCollector(ICredentialCollector):

monkey/infection_monkey/credential_collectors/mimikatz_collector/pypykatz_handler.py

@@ -3,9 +3,7 @@ from typing import Any, Dict, List, NewType
 
 from pypykatz.pypykatz import pypykatz
 
-from infection_monkey.system_info.windows_cred_collector.windows_credentials import (
-    WindowsCredentials,
-)
+from .windows_credentials import WindowsCredentials
 
 CREDENTIAL_TYPES = [
     "msv_creds",

monkey/infection_monkey/system_info/windows_info_collector.py

@@ -2,7 +2,7 @@ import logging
 import sys
 
 from common.common_consts.system_info_collectors_names import MIMIKATZ_COLLECTOR
-from infection_monkey.system_info.windows_cred_collector.mimikatz_cred_collector import (
+from infection_monkey.credential_collectors.windows_cred_collector.mimikatz_cred_collector import (
     MimikatzCredentialCollector,
 )
 

monkey/tests/unit_tests/infection_monkey/system_info/windows_cred_collector/test_pypykatz_handler.py

@@ -1,6 +1,6 @@
 from unittest import TestCase
 
-from infection_monkey.system_info.windows_cred_collector.pypykatz_handler import (
+from infection_monkey.credential_collectors.mimikatz_collector.pypykatz_handler import (
     _get_creds_from_pypykatz_session,
 )