tests: integration: cover idle timeout recovery

Signed-off-by: Eduardo Silva <eduardo@chronosphere.io>

Author: edsiper

tests/integration/scenarios/in_http_max_connections/config/in_http_max_connections.yaml

@@ -10,6 +10,7 @@ pipeline:
     - name: http
       listen: 127.0.0.1
       port: ${FLUENT_BIT_TEST_LISTENER_PORT}
+      http_server.idle_timeout: 2s
       http_server.max_connections: 1
 
   outputs:

tests/integration/scenarios/in_http_max_connections/tests/test_in_http_max_connections_001.py

@@ -1,12 +1,13 @@
 import os
 import socket
+import time
 
 import pytest
 import requests
 
 from server.http_server import data_storage, http_server_run
 from utils.fluent_bit_manager import FluentBitStartupError
-from utils.http_matrix import run_curl_request
+from utils.http_matrix import curl_supports_http2, run_curl_request
 from utils.test_service import FluentBitTestService
 
 
@@ -93,3 +94,101 @@ def test_in_http_max_connections_blocks_and_recovers():
 
     assert accepted["status_code"] == 201
     assert forwarded_payloads[0][0]["message"] == "max-connections"
+
+
+def test_in_http_idle_timeout_evicts_partial_request_connection():
+    service = Service()
+    service.start()
+
+    held_connection = None
+    try:
+        held_connection = socket.create_connection(("127.0.0.1", service.flb_listener_port), timeout=2)
+        held_connection.settimeout(5)
+        held_connection.sendall(b"POST / HTTP/1.1\r\n")
+
+        overflow_rejected = False
+        try:
+            response = run_curl_request(
+                f"http://127.0.0.1:{service.flb_listener_port}/",
+                payload='{"message":"idle-timeout-blocked"}',
+                headers=["Content-Type: application/json"],
+                http_mode="http1.1",
+            )
+            overflow_rejected = response["status_code"] != 201
+        except Exception:
+            overflow_rejected = True
+
+        assert overflow_rejected
+
+        time.sleep(3)
+
+        response = run_curl_request(
+            f"http://127.0.0.1:{service.flb_listener_port}/",
+            payload='{"message":"idle-timeout-recovered"}',
+            headers=["Content-Type: application/json"],
+            http_mode="http1.1",
+        )
+        forwarded_payloads = service.service.wait_for_condition(
+            lambda: data_storage["payloads"] if data_storage["payloads"] else None,
+            timeout=10,
+            interval=0.5,
+            description="forwarded idle-timeout payload",
+        )
+    finally:
+        if held_connection:
+            held_connection.close()
+        service.stop()
+
+    assert response["status_code"] == 201
+    assert forwarded_payloads[0][0]["message"] == "idle-timeout-recovered"
+
+
+def test_in_http_idle_timeout_evicts_partial_http2_preface_connection():
+    if not curl_supports_http2():
+        pytest.skip("curl was built without HTTP/2 support")
+
+    service = Service()
+    service.start()
+
+    held_connection = None
+    try:
+        held_connection = socket.create_connection(("127.0.0.1", service.flb_listener_port), timeout=2)
+        held_connection.settimeout(5)
+        held_connection.sendall(b"PRI * HTTP/2.0\r\n\r\nSM\r\n")
+
+        overflow_rejected = False
+        try:
+            response = run_curl_request(
+                f"http://127.0.0.1:{service.flb_listener_port}/",
+                payload='{"message":"idle-timeout-http2-blocked"}',
+                headers=["Content-Type: application/json"],
+                http_mode="http2-prior-knowledge",
+            )
+            overflow_rejected = response["status_code"] != 201
+        except Exception:
+            overflow_rejected = True
+
+        assert overflow_rejected
+
+        time.sleep(3)
+
+        response = run_curl_request(
+            f"http://127.0.0.1:{service.flb_listener_port}/",
+            payload='{"message":"idle-timeout-http2-recovered"}',
+            headers=["Content-Type: application/json"],
+            http_mode="http2-prior-knowledge",
+        )
+        forwarded_payloads = service.service.wait_for_condition(
+            lambda: data_storage["payloads"] if data_storage["payloads"] else None,
+            timeout=10,
+            interval=0.5,
+            description="forwarded idle-timeout http2 payload",
+        )
+    finally:
+        if held_connection:
+            held_connection.close()
+        service.stop()
+
+    assert response["status_code"] == 201
+    assert response["http_version"] == "2"
+    assert forwarded_payloads[0][0]["message"] == "idle-timeout-http2-recovered"