UT: Add another test for AgentEventsToIslandSender

monkey/tests/unit_tests/infection_monkey/test_send_all_events_to_island.py

@@ -10,10 +10,9 @@ SERVER = "1.1.1.1:9999"
 
 @pytest.fixture
 def event_sender():
-    return AgentEventsToIslandSender(SERVER)
+    return AgentEventsToIslandSender(SERVER, time_period=0.001)
 
 
-# @pytest.mark.skipif(os.name != "posix", reason="This test is racey on Windows")
 def test_send_events(event_sender):
     with requests_mock.Mocker() as mock:
         mock.post(EVENTS_API_URL % SERVER)
@@ -22,11 +21,27 @@ def test_send_events(event_sender):
 
         for _ in range(5):
             event_sender.add_event_to_queue({})
-        time.sleep(1)
-        assert mock.call_count == 5
+        time.sleep(0.1)
+        assert mock.call_count == 1
 
         event_sender.add_event_to_queue({})
-        time.sleep(1)
-        assert mock.call_count == 6
+        time.sleep(0.1)
+        assert mock.call_count == 2
 
         event_sender.stop()
+
+
+def test_send_remaining_events(event_sender):
+    with requests_mock.Mocker() as mock:
+        mock.post(EVENTS_API_URL % SERVER)
+
+        event_sender.start()
+
+        for _ in range(5):
+            event_sender.add_event_to_queue({})
+        time.sleep(0.1)
+        assert mock.call_count == 1
+
+        event_sender.add_event_to_queue({})
+        event_sender.stop()
+        assert mock.call_count == 2