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

UT: Parametrize test_mimikatz_collector.test_empty_results()

This commit is contained in:
Mike Salvatore 2022-02-15 13:28:38 -05:00
parent ebd5642b52
commit 86f2c7b08c
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
monkey/tests/unit_tests/infection_monkey/credential_collectors/mimikatz_collector/test_mimikatz_collector.py
View File

@ -1,3 +1,5 @@
import pytest
from infection_monkey.credential_collectors import Credentials, LMHash, NTHash, Password, Username from infection_monkey.credential_collectors import Credentials, LMHash, NTHash, Password, Username
from infection_monkey.credential_collectors.mimikatz_collector.mimikatz_cred_collector import ( from infection_monkey.credential_collectors.mimikatz_collector.mimikatz_cred_collector import (
MimikatzCredentialCollector, MimikatzCredentialCollector,
@ -15,15 +17,12 @@ def patch_pypykatz(win_creds: [WindowsCredentials], monkeypatch):
) )
def test_empty_results(monkeypatch): @pytest.mark.parametrize(
win_creds = [WindowsCredentials(username="", password="", ntlm_hash="", lm_hash="")] "win_creds", [([WindowsCredentials(username="", password="", ntlm_hash="", lm_hash="")]), ([])]
)
def test_empty_results(monkeypatch, win_creds):
patch_pypykatz(win_creds, monkeypatch) patch_pypykatz(win_creds, monkeypatch)
expected_credentials = [] collected_credentials = list(MimikatzCredentialCollector().collect_credentials())
collected_credentials = MimikatzCredentialCollector().collect_credentials()
assert expected_credentials == collected_credentials
patch_pypykatz([], monkeypatch)
collected_credentials = MimikatzCredentialCollector().collect_credentials()
assert not collected_credentials assert not collected_credentials