From 67220574918be9561bd9c35a16e7bda39d08b1eb Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Wed, 17 Aug 2022 18:03:07 +0530 Subject: [PATCH] Agent: Use existing patch function in test_pypykatz_result_parsing_no_secrets --- .../credential_collectors/test_mimikatz_collector.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_mimikatz_collector.py b/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_mimikatz_collector.py index a3777b8e8..74c852437 100644 --- a/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_mimikatz_collector.py +++ b/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_mimikatz_collector.py @@ -1,5 +1,5 @@ from typing import Sequence -from unittest.mock import MagicMock, Mock +from unittest.mock import MagicMock import pytest @@ -121,11 +121,7 @@ def test_pypykatz_result_parsing_no_secrets(monkeypatch): def test_mimikatz_credentials_stolen_event_published(monkeypatch): mock_event_queue = MagicMock(spec=IEventQueue) - mock_pypykatz_handler = Mock() - monkeypatch.setattr( - "infection_monkey.credential_collectors.mimikatz_collector.pypykatz_handler", - mock_pypykatz_handler, - ) + patch_pypykatz([], monkeypatch) mimikatz_credential_collector = MimikatzCredentialCollector(mock_event_queue) collected_credentials = mimikatz_credential_collector.collect_credentials()