From 2e0ef1865b4b262bdb6e128a90bdf97c9304534f Mon Sep 17 00:00:00 2001 From: Kekoa Kaaikala Date: Fri, 7 Oct 2022 20:34:35 +0000 Subject: [PATCH] UT: Add dummy timestamp to zerologon tests --- .../unit_tests/infection_monkey/exploit/test_zerologon.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/tests/unit_tests/infection_monkey/exploit/test_zerologon.py b/monkey/tests/unit_tests/infection_monkey/exploit/test_zerologon.py index 6bae3746e..76da8570c 100644 --- a/monkey/tests/unit_tests/infection_monkey/exploit/test_zerologon.py +++ b/monkey/tests/unit_tests/infection_monkey/exploit/test_zerologon.py @@ -37,7 +37,7 @@ def zerologon_exploiter_object(monkeypatch, mock_agent_event_queue): @pytest.mark.slow def test_assess_exploit_attempt_result_no_error(zerologon_exploiter_object, mock_agent_event_queue): dummy_exploit_attempt_result = {"ErrorCode": 0} - assert zerologon_exploiter_object.assess_exploit_attempt_result(dummy_exploit_attempt_result) + assert zerologon_exploiter_object.assess_exploit_attempt_result(dummy_exploit_attempt_result, 0) assert mock_agent_event_queue.publish.call_count == 1 assert mock_agent_event_queue.publish.call_args[0][0].__class__ is ExploitationEvent assert mock_agent_event_queue.publish.call_args[0][0].success @@ -49,7 +49,7 @@ def test_assess_exploit_attempt_result_with_error( ): dummy_exploit_attempt_result = {"ErrorCode": 1} assert not zerologon_exploiter_object.assess_exploit_attempt_result( - dummy_exploit_attempt_result + dummy_exploit_attempt_result, 0 ) assert mock_agent_event_queue.publish.call_count == 1 assert mock_agent_event_queue.publish.call_args[0][0].__class__ is ExploitationEvent