diff --git a/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_ssh_credentials_collector.py b/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_ssh_credentials_collector.py index 14a2e320a..ba12e416f 100644 --- a/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_ssh_credentials_collector.py +++ b/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_ssh_credentials_collector.py @@ -1,10 +1,9 @@ from unittest.mock import MagicMock import pytest -from pubsub.core import Publisher from common.credentials import Credentials, SSHKeypair, Username -from common.event_queue import IEventQueue, PyPubSubEventQueue +from common.event_queue import IEventQueue from infection_monkey.credential_collectors import SSHCredentialCollector @@ -13,11 +12,6 @@ def patch_telemetry_messenger(): return MagicMock() -@pytest.fixture -def event_queue() -> IEventQueue: - return PyPubSubEventQueue(Publisher()) - - def patch_ssh_handler(ssh_creds, monkeypatch): monkeypatch.setattr( "infection_monkey.credential_collectors.ssh_collector.ssh_handler.get_ssh_info",