From 7fed97530fab311990b54c68726775a6e0928cda Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Wed, 14 Sep 2022 19:52:33 +0530 Subject: [PATCH] UT: Use BatchingAgentEventForwarder instead of AgentEventsToIslandSender --- .../unit_tests/infection_monkey/test_agent_event_forwarder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/tests/unit_tests/infection_monkey/test_agent_event_forwarder.py b/monkey/tests/unit_tests/infection_monkey/test_agent_event_forwarder.py index 0056a0052..5762c456f 100644 --- a/monkey/tests/unit_tests/infection_monkey/test_agent_event_forwarder.py +++ b/monkey/tests/unit_tests/infection_monkey/test_agent_event_forwarder.py @@ -3,14 +3,14 @@ import time import pytest import requests_mock -from infection_monkey.agent_event_forwarder import EVENTS_API_URL, AgentEventsToIslandSender +from infection_monkey.agent_event_forwarder import EVENTS_API_URL, BatchingAgentEventForwarder SERVER = "1.1.1.1:9999" @pytest.fixture def event_sender(): - return AgentEventsToIslandSender(SERVER, time_period=0.001) + return BatchingAgentEventForwarder(SERVER, time_period=0.001) def test_send_events(event_sender):