From a002c96bc63bb0bf342319908329287eaafd20f0 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Thu, 17 Mar 2022 10:45:56 -0400 Subject: [PATCH] Agent: Add interrupt to powershell tests --- .../unit_tests/infection_monkey/exploit/test_powershell.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/monkey/tests/unit_tests/infection_monkey/exploit/test_powershell.py b/monkey/tests/unit_tests/infection_monkey/exploit/test_powershell.py index f75c57f17..21a0bdeb3 100644 --- a/monkey/tests/unit_tests/infection_monkey/exploit/test_powershell.py +++ b/monkey/tests/unit_tests/infection_monkey/exploit/test_powershell.py @@ -1,3 +1,4 @@ +import threading from io import BytesIO from unittest.mock import MagicMock @@ -43,6 +44,7 @@ def powershell_arguments(): "current_depth": 2, "telemetry_messenger": MagicMock(), "agent_repository": mock_agent_repository, + "interrupt": threading.Event(), } return arguments