diff --git a/monkey/infection_monkey/credential_collectors/__init__.py b/monkey/infection_monkey/credential_collectors/__init__.py index 76ebc4d87..a9d22a4c4 100644 --- a/monkey/infection_monkey/credential_collectors/__init__.py +++ b/monkey/infection_monkey/credential_collectors/__init__.py @@ -2,3 +2,4 @@ from .credential_components.nt_hash import NTHash from .credential_components.lm_hash import LMHash from .credential_components.password import Password from .credential_components.username import Username +from .mimikatz_collector import MimikatzCredentialCollector diff --git a/monkey/infection_monkey/credential_collectors/mimikatz_collector/__init__.py b/monkey/infection_monkey/credential_collectors/mimikatz_collector/__init__.py index e69de29bb..c6a8f1a91 100644 --- a/monkey/infection_monkey/credential_collectors/mimikatz_collector/__init__.py +++ b/monkey/infection_monkey/credential_collectors/mimikatz_collector/__init__.py @@ -0,0 +1 @@ +from .mimikatz_credential_collector import MimikatzCredentialCollector diff --git a/monkey/tests/unit_tests/infection_monkey/credential_collectors/mimikatz_collector/test_mimikatz_collector.py b/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_mimikatz_collector.py similarity index 94% rename from monkey/tests/unit_tests/infection_monkey/credential_collectors/mimikatz_collector/test_mimikatz_collector.py rename to monkey/tests/unit_tests/infection_monkey/credential_collectors/test_mimikatz_collector.py index 8380229b5..b33d4e097 100644 --- a/monkey/tests/unit_tests/infection_monkey/credential_collectors/mimikatz_collector/test_mimikatz_collector.py +++ b/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_mimikatz_collector.py @@ -2,9 +2,12 @@ from typing import List import pytest -from infection_monkey.credential_collectors import LMHash, NTHash, Password, Username -from infection_monkey.credential_collectors.mimikatz_collector.mimikatz_cred_collector import ( +from infection_monkey.credential_collectors import ( + LMHash, MimikatzCredentialCollector, + NTHash, + Password, + Username, ) from infection_monkey.credential_collectors.mimikatz_collector.windows_credentials import ( WindowsCredentials,