fix logging OOM issue during e2e tests

Author: sohankshirsagar

drift/instrumentation/e2e_common/base_runner.py

@@ -150,8 +150,8 @@ def record_traces(self) -> bool:
         self.app_process = subprocess.Popen(
             ["python", "src/app.py"],
             env={**os.environ, **env},
-            stdout=subprocess.PIPE,
-            stderr=subprocess.STDOUT,
+            stdout=subprocess.DEVNULL,
+            stderr=subprocess.DEVNULL,
             text=True,
         )
 

drift/instrumentation/psycopg/e2e-tests/src/app.py

@@ -723,14 +723,6 @@ def test_binary_format():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-
-# =====================================================================
-# BUG-EXPOSING TEST ENDPOINTS
-# These endpoints expose confirmed bugs in the psycopg instrumentation.
-# See BUG_TRACKING.md for detailed analysis.
-# =====================================================================
-
-
 @app.route("/test/null-values")
 def test_null_values():
     """Test handling of NULL values in results."""

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

@@ -75,8 +75,8 @@
     # Test: Transaction context manager
     make_request("GET", "/test/transaction-context")
 
-    # These still need investigation:
-    # make_request("GET", "/test/json-jsonb")
-    # make_request("GET", "/test/array-types")
+    # JSON/JSONB and array types tests
+    make_request("GET", "/test/json-jsonb")
+    make_request("GET", "/test/array-types")
 
     print_request_summary()

drift/instrumentation/psycopg/instrumentation.py

@@ -1202,7 +1202,6 @@ def _mock_execute_with_data(self, cursor: Any, mock_data: dict[str, Any]) -> Non
         # The SDK communicator already extracts response.body from the CLI's MockInteraction structure
         # So mock_data should already contain: {"rowcount": ..., "description": [...], "rows": [...]}
         actual_data = mock_data
-        logger.debug(f"[MOCK_DATA] mock_data: {mock_data}")
 
         try:
             cursor._rowcount = actual_data.get("rowcount", -1)