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

UT: Simplify MimikatzCredentialCollector's event publishing test

This commit is contained in:
Shreya Malviya 2022-08-16 19:09:50 +05:30
parent 95a3be0273
commit b7ada959fa
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -8,11 +8,8 @@ from common.credentials import Credentials, LMHash, NTHash, Password, Username
from common.event_queue import IEventQueue, PyPubSubEventQueue from common.event_queue import IEventQueue, PyPubSubEventQueue
from common.events import AbstractEvent from common.events import AbstractEvent
from infection_monkey.credential_collectors import MimikatzCredentialCollector from infection_monkey.credential_collectors import MimikatzCredentialCollector
from infection_monkey.credential_collectors.mimikatz_collector.mimikatz_credential_collector import ( from infection_monkey.credential_collectors.mimikatz_collector.mimikatz_credential_collector import ( # noqa: E501
MIMIKATZ_CREDENTIAL_COLLECTOR_TAG,
MIMIKATZ_EVENT_TAGS, MIMIKATZ_EVENT_TAGS,
T1003_ATTACK_TECHNIQUE_TAG,
T1005_ATTACK_TECHNIQUE_TAG,
) )
from infection_monkey.credential_collectors.mimikatz_collector.windows_credentials import ( from infection_monkey.credential_collectors.mimikatz_collector.windows_credentials import (
WindowsCredentials, WindowsCredentials,
@ -136,9 +133,8 @@ def test_pypykatz_credentials_stolen_event_published(monkeypatch, event_queue):
subscriber.call_count = 0 subscriber.call_count = 0
subscriber.call_tags = set() subscriber.call_tags = set()
event_queue.subscribe_tag(MIMIKATZ_CREDENTIAL_COLLECTOR_TAG, subscriber) for event_tag in MIMIKATZ_EVENT_TAGS:
event_queue.subscribe_tag(T1003_ATTACK_TECHNIQUE_TAG, subscriber) event_queue.subscribe_tag(event_tag, subscriber)
event_queue.subscribe_tag(T1005_ATTACK_TECHNIQUE_TAG, subscriber)
mimikatz_credential_collector = MimikatzCredentialCollector(event_queue) mimikatz_credential_collector = MimikatzCredentialCollector(event_queue)
monkeypatch.setattr( monkeypatch.setattr(