Fix race condition

apache · Oct 11, 2024 · ed02950 · ed02950
1 parent f2e34a2
commit ed02950
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/csharp/test/Apache.Arrow.Flight.IntegrationTest/JsonTestScenario.cs b/csharp/test/Apache.Arrow.Flight.IntegrationTest/JsonTestScenario.cs
@@ -133,6 +133,13 @@ private static async Task UploadBatches(FlightClient client, FlightDescriptor de
         {
             await writer.CompleteAsync().ConfigureAwait(false);
         }
+
+        // Drain the response stream to ensure the server has stored the data
+        var hasMore = await putCall.ResponseStream.MoveNext().ConfigureAwait(false);
+        if (hasMore)
+        {
+            throw new Exception("Expected to have reached the end of the response stream");
+        }
     }
 
     private static async Task ConsumeFlightLocation(FlightClient client, FlightTicket ticket, RecordBatch[] batches)