UT: Simplify variable logic in MimikatzCredentialCollector's event publishing test

This commit is contained in:
Shreya Malviya 2022-08-17 14:44:43 +05:30
parent eb17b20625
commit 11901b1835
1 changed files with 3 additions and 2 deletions

View File

@ -121,9 +121,10 @@ def test_pypykatz_result_parsing_no_secrets(monkeypatch):
def test_mimikatz_credentials_stolen_event_published(monkeypatch):
mock_event_queue = MagicMock(spec=IEventQueue)
mock_subscriber = Mock()
mock_subscriber = mock_pypykatz_handler = Mock()
monkeypatch.setattr(
"infection_monkey.credential_collectors.mimikatz_collector.pypykatz_handler", Mock()
"infection_monkey.credential_collectors.mimikatz_collector.pypykatz_handler",
mock_pypykatz_handler,
)
for event_tag in MIMIKATZ_EVENT_TAGS: