p59342861
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

UT: Simplify test_mimikatz_credentials_stolen_event_published

This commit is contained in:
Shreya Malviya 2022-08-17 17:31:06 +05:30
parent 2c3b29493f
commit 4334740002
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
monkey/tests/unit_tests/infection_monkey/credential_collectors/test_mimikatz_collector.py
View File

@ -121,15 +121,12 @@ def test_pypykatz_result_parsing_no_secrets(monkeypatch):
def test_mimikatz_credentials_stolen_event_published(monkeypatch): def test_mimikatz_credentials_stolen_event_published(monkeypatch):
mock_event_queue = MagicMock(spec=IEventQueue) mock_event_queue = MagicMock(spec=IEventQueue)
mock_subscriber = mock_pypykatz_handler = Mock() mock_pypykatz_handler = Mock()
monkeypatch.setattr( monkeypatch.setattr(
"infection_monkey.credential_collectors.mimikatz_collector.pypykatz_handler", "infection_monkey.credential_collectors.mimikatz_collector.pypykatz_handler",
mock_pypykatz_handler, mock_pypykatz_handler,
) )
for event_tag in MIMIKATZ_EVENT_TAGS:
mock_event_queue.subscribe_tag(event_tag, mock_subscriber)
mimikatz_credential_collector = MimikatzCredentialCollector(mock_event_queue) mimikatz_credential_collector = MimikatzCredentialCollector(mock_event_queue)
collected_credentials = mimikatz_credential_collector.collect_credentials() collected_credentials = mimikatz_credential_collector.collect_credentials()