refactor: use shared test utils in urllib3 and aiohttp e2e tests (#37)

Author: sohankshirsagar

drift/instrumentation/aiohttp/e2e-tests/src/test_requests.py

@@ -1,24 +1,6 @@
 """Execute test requests against the Flask app to exercise the aiohttp instrumentation."""
 
-import time
-
-import requests
-
-BASE_URL = "http://localhost:8000"
-
-
-def make_request(method, endpoint, **kwargs):
-    """Make HTTP request and log result."""
-    url = f"{BASE_URL}{endpoint}"
-    print(f"-> {method} {endpoint}")
-
-    # Set default timeout if not provided
-    kwargs.setdefault("timeout", 30)
-    response = requests.request(method, url, **kwargs)
-    print(f"   Status: {response.status_code}")
-    time.sleep(0.5)  # Small delay between requests
-    return response
-
+from drift.instrumentation.e2e_common.test_utils import make_request, print_request_summary
 
 if __name__ == "__main__":
     print("Starting test request sequence for aiohttp instrumentation...\n")
@@ -106,4 +88,4 @@ def make_request(method, endpoint, **kwargs):
     # POST with bytes body
     make_request("POST", "/test/post-bytes")
 
-    print("\nAll requests completed successfully")
+    print_request_summary()

drift/instrumentation/urllib3/e2e-tests/src/test_requests.py

@@ -1,24 +1,6 @@
 """Execute test requests against the Flask app to exercise the urllib3 instrumentation."""
 
-import time
-
-import requests
-
-BASE_URL = "http://localhost:8000"
-
-
-def make_request(method, endpoint, **kwargs):
-    """Make HTTP request and log result."""
-    url = f"{BASE_URL}{endpoint}"
-    print(f"-> {method} {endpoint}")
-
-    # Set default timeout if not provided
-    kwargs.setdefault("timeout", 30)
-    response = requests.request(method, url, **kwargs)
-    print(f"   Status: {response.status_code}")
-    time.sleep(0.5)  # Small delay between requests
-    return response
-
+from drift.instrumentation.e2e_common.test_utils import make_request, print_request_summary
 
 if __name__ == "__main__":
     print("Starting test request sequence for urllib3 instrumentation...\n")
@@ -127,4 +109,4 @@ def make_request(method, endpoint, **kwargs):
     #   curl http://localhost:8000/test/bug/streaming-response
     # ==========================================================================
 
-    print("\nAll requests completed successfully")
+    print_request_summary()