Ensure FlightDataStream/RecordBatchStream::Close errors are propagated
| 284 | } |
| 285 | // Ensure FlightDataStream/RecordBatchStream::Close errors are propagated |
| 286 | void DataTest::TestFlightDataStreamError() { |
| 287 | Ticket ticket{"ticket-stream-error"}; |
| 288 | |
| 289 | ASSERT_OK_AND_ASSIGN(auto stream, client_->DoGet(ticket)); |
| 290 | Status status; |
| 291 | while (true) { |
| 292 | FlightStreamChunk chunk; |
| 293 | status = stream->Next().Value(&chunk); |
| 294 | if (!chunk.data) break; |
| 295 | if (!status.ok()) break; |
| 296 | } |
| 297 | EXPECT_RAISES_WITH_MESSAGE_THAT(IOError, ::testing::HasSubstr("Expected error"), |
| 298 | status); |
| 299 | } |
| 300 | void DataTest::TestOverflowServerBatch() { |
| 301 | // Regression test for ARROW-13253 |
| 302 | // N.B. this is rather a slow and memory-hungry test |