Method TestUndrained

cpp/src/arrow/flight/test_definitions.cc:865–885

  CheckBatches(descr, {batch1, batch2});
}
void DoPutTest::TestUndrained() {
  // Ensure if the application doesn't drain all messages, that the
  // server/transport does

  auto descr = FlightDescriptor::Command("TestUndrained");
  auto schema = arrow::schema({arrow::field("ints", int64())});
  ASSERT_OK_AND_ASSIGN(auto do_put_result, client_->DoPut(descr, schema));
  std::unique_ptr<FlightStreamWriter> writer = std::move(do_put_result.writer);
  std::unique_ptr<FlightMetadataReader> reader = std::move(do_put_result.reader);
  auto batch = RecordBatchFromJSON(schema, "[[1], [2], [3], [4]]");
  // These calls may or may not fail depending on how quickly the
  // transport reacts, whether it batches, writes, etc.
  ARROW_UNUSED(writer->WriteRecordBatch(*batch));
  ARROW_UNUSED(writer->WriteRecordBatch(*batch));
  ARROW_UNUSED(writer->WriteRecordBatch(*batch));
  ARROW_UNUSED(writer->WriteRecordBatch(*batch));
  ASSERT_OK(writer->Close());

  // We should be able to make another call
  CheckDoPut(FlightDescriptor::Command("foo"), schema, {batch, batch});
}

//------------------------------------------------------------
// Test app_metadata in data plane methods