Verify all Phase 2 SEA E2E tests pass\n\nTask ID: task-2.5-verify-phase2-tests

Author: Jade Wang

csharp/test/E2E/Telemetry/SEAChunkDetailsTelemetryTests.cs

@@ -152,6 +152,10 @@ public async Task SEACloudFetch_AllChunkDetailsFields_ArePopulated()
                     statement.Dispose();
                 }
 
+                // Dispose connection to trigger telemetry flush
+                connection.Dispose();
+                connection = null;
+
                 // Wait for telemetry to be emitted
                 var logs = await TelemetryTestHelpers.WaitForTelemetryEvents(exporter, expectedCount: 1, timeoutMs: 10000);
 
@@ -278,6 +282,13 @@ public async Task SEACloudFetch_SlowestChunkLatency_GteInitial()
                     statement.Dispose();
                 }
 
+                // Dispose connection to trigger telemetry flush
+                connection.Dispose();
+                connection = null;
+
+                // Small delay to ensure async flush completes
+                await Task.Delay(500);
+
                 var logs = await TelemetryTestHelpers.WaitForTelemetryEvents(exporter, expectedCount: 1, timeoutMs: 10000);
 
                 Assert.NotEmpty(logs);
@@ -331,6 +342,13 @@ public async Task SEACloudFetch_SumDownloadTime_GteSlowest()
                     statement.Dispose();
                 }
 
+                // Dispose connection to trigger telemetry flush
+                connection.Dispose();
+                connection = null;
+
+                // Small delay to ensure async flush completes
+                await Task.Delay(500);
+
                 var logs = await TelemetryTestHelpers.WaitForTelemetryEvents(exporter, expectedCount: 1, timeoutMs: 10000);
 
                 Assert.NotEmpty(logs);
@@ -384,6 +402,13 @@ public async Task SEACloudFetch_TotalChunksIterated_LtePresent()
                     statement.Dispose();
                 }
 
+                // Dispose connection to trigger telemetry flush
+                connection.Dispose();
+                connection = null;
+
+                // Small delay to ensure async flush completes
+                await Task.Delay(500);
+
                 var logs = await TelemetryTestHelpers.WaitForTelemetryEvents(exporter, expectedCount: 1, timeoutMs: 10000);
 
                 Assert.NotEmpty(logs);

csharp/test/E2E/Telemetry/SEAStatementTelemetryTests.cs

@@ -146,6 +146,10 @@ public async Task SEAStatement_QueryEmitsTelemetry()
                     statement.Dispose();
                 }
 
+                // Dispose connection to trigger telemetry flush
+                connection.Dispose();
+                connection = null;
+
                 // Wait for telemetry to be emitted
                 var logs = await TelemetryTestHelpers.WaitForTelemetryEvents(exporter, expectedCount: 1);
 
@@ -191,6 +195,13 @@ public async Task SEAStatement_OperationTypeIsExecuteStatementAsync()
                     statement.Dispose();
                 }
 
+                // Dispose connection to trigger telemetry flush
+                connection.Dispose();
+                connection = null;
+
+                // Small delay to ensure async flush completes
+                await Task.Delay(500);
+
                 // Wait for telemetry to be emitted
                 var logs = await TelemetryTestHelpers.WaitForTelemetryEvents(exporter, expectedCount: 1);
 
@@ -235,6 +246,13 @@ public async Task SEAStatement_ExecutionResultMatchesDisposition()
                     statement.Dispose();
                 }
 
+                // Dispose connection to trigger telemetry flush
+                connection.Dispose();
+                connection = null;
+
+                // Small delay to ensure async flush completes
+                await Task.Delay(500);
+
                 // Wait for telemetry to be emitted
                 var logs = await TelemetryTestHelpers.WaitForTelemetryEvents(exporter, expectedCount: 1);
 
@@ -294,6 +312,13 @@ public async Task SEAStatement_ErrorInfoForFailedQueries()
                     statement.Dispose();
                 }
 
+                // Dispose connection to trigger telemetry flush
+                connection.Dispose();
+                connection = null;
+
+                // Small delay to ensure async flush completes
+                await Task.Delay(500);
+
                 // Wait for telemetry to be emitted
                 var logs = await TelemetryTestHelpers.WaitForTelemetryEvents(exporter, expectedCount: 1);
 
@@ -338,6 +363,13 @@ public async Task SEAStatement_OperationLatencyIsPositive()
                     statement.Dispose();
                 }
 
+                // Dispose connection to trigger telemetry flush
+                connection.Dispose();
+                connection = null;
+
+                // Small delay to ensure async flush completes
+                await Task.Delay(500);
+
                 // Wait for telemetry to be emitted
                 var logs = await TelemetryTestHelpers.WaitForTelemetryEvents(exporter, expectedCount: 1);
 
@@ -386,6 +418,10 @@ public async Task SEAStatement_AllExitCriteriaMet()
                     statement.Dispose();
                 }
 
+                // Dispose connection to trigger telemetry flush before checking first query
+                connection.Dispose();
+                connection = null;
+
                 // Wait for telemetry to be emitted
                 var logs = await TelemetryTestHelpers.WaitForTelemetryEvents(exporter, expectedCount: 1);
 
@@ -413,8 +449,9 @@ public async Task SEAStatement_AllExitCriteriaMet()
                 OutputHelper?.WriteLine($"✓ Exit criterion 5: operation_latency_ms = {protoLog.OperationLatencyMs}ms");
 
                 // Exit criterion 4: Error telemetry works for SEA failures
-                // Reset exporter for error test
+                // Reset exporter for error test and create new connection
                 exporter.Reset();
+                (connection, exporter) = CreateRestConnectionWithCapturingTelemetry();
 
                 using (var statement = connection.CreateStatement())
                 {
@@ -434,6 +471,10 @@ public async Task SEAStatement_AllExitCriteriaMet()
                     statement.Dispose();
                 }
 
+                // Dispose connection to trigger telemetry flush
+                connection.Dispose();
+                connection = null;
+
                 // Wait for telemetry
                 var errorLogs = await TelemetryTestHelpers.WaitForTelemetryEvents(exporter, expectedCount: 1);
                 Assert.NotEmpty(errorLogs);