Method CheckDoPut

cpp/src/arrow/flight/test_definitions.cc:721–753

  }
}
void DoPutTest::CheckDoPut(const FlightDescriptor& descr,
                           const std::shared_ptr<Schema>& schema,
                           const RecordBatchVector& batches) {
  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);

  // Ensure that the reader can be used independently of the writer
  std::thread reader_thread([&reader, &batches]() {
    for (size_t i = 0; i < batches.size(); i++) {
      std::shared_ptr<Buffer> out;
      ASSERT_OK(reader->ReadMetadata(&out));
    }
  });

  int64_t counter = 0;
  for (const auto& batch : batches) {
    if (counter % 2 == 0) {
      ASSERT_OK(writer->WriteRecordBatch(*batch));
    } else {
      auto buffer = Buffer::FromString(std::to_string(counter));
      ASSERT_OK(writer->WriteWithMetadata(*batch, std::move(buffer)));
    }
    counter++;
  }
  // Write a metadata-only message
  ASSERT_OK(writer->WriteMetadata(Buffer::FromString(kExpectedMetadata)));
  ASSERT_OK(writer->DoneWriting());
  reader_thread.join();
  ASSERT_OK(writer->Close());

  CheckBatches(descr, batches);
}

void DoPutTest::TestInts() {
  auto descr = FlightDescriptor::Path({"ints"});