UT: Use BatchingAgentEventForwarder instead of AgentEventsToIslandSender

This commit is contained in:
Shreya Malviya 2022-09-14 19:52:33 +05:30
parent c24fd9ddab
commit 7fed97530f
1 changed files with 2 additions and 2 deletions

View File

@ -3,14 +3,14 @@ import time
import pytest import pytest
import requests_mock 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" SERVER = "1.1.1.1:9999"
@pytest.fixture @pytest.fixture
def event_sender(): def event_sender():
return AgentEventsToIslandSender(SERVER, time_period=0.001) return BatchingAgentEventForwarder(SERVER, time_period=0.001)
def test_send_events(event_sender): def test_send_events(event_sender):