diff --git a/monkey/infection_monkey/credential_collectors/__init__.py b/monkey/infection_monkey/credential_collectors/__init__.py index a5d48e466..1f259949d 100644 --- a/monkey/infection_monkey/credential_collectors/__init__.py +++ b/monkey/infection_monkey/credential_collectors/__init__.py @@ -4,3 +4,4 @@ from .credential_components.password import Password from .credential_components.username import Username from .credential_components.ssh_keypair import SSHKeypair from .mimikatz_collector import MimikatzCredentialCollector +from .ssh_collector import SSHCredentialCollector diff --git a/monkey/tests/unit_tests/infection_monkey/credential_collectors/linux_credentials_collector/test_ssh_credentials_collector.py b/monkey/tests/unit_tests/infection_monkey/credential_collectors/linux_credentials_collector/test_ssh_credentials_collector.py index 3727f8698..4b344d9b0 100644 --- a/monkey/tests/unit_tests/infection_monkey/credential_collectors/linux_credentials_collector/test_ssh_credentials_collector.py +++ b/monkey/tests/unit_tests/infection_monkey/credential_collectors/linux_credentials_collector/test_ssh_credentials_collector.py @@ -2,8 +2,7 @@ from unittest.mock import MagicMock import pytest -from infection_monkey.credential_collectors import SSHKeypair, Username -from infection_monkey.credential_collectors.ssh_collector import SSHCredentialCollector +from infection_monkey.credential_collectors import SSHCredentialCollector, SSHKeypair, Username from infection_monkey.i_puppet.credential_collection import Credentials