| 710 | checked_cast<DoPutTestServer*>(server_.get())->batches_.clear(); |
| 711 | } |
| 712 | void DoPutTest::CheckBatches(const FlightDescriptor& expected_descriptor, |
| 713 | const RecordBatchVector& expected_batches) { |
| 714 | auto* do_put_server = static_cast<DoPutTestServer*>(server_.get()); |
| 715 | ASSERT_EQ(do_put_server->descriptor_, expected_descriptor); |
| 716 | ASSERT_EQ(do_put_server->batches_.size(), expected_batches.size()); |
| 717 | for (size_t i = 0; i < expected_batches.size(); ++i) { |
| 718 | ASSERT_BATCHES_EQUAL(*do_put_server->batches_[i], *expected_batches[i]); |
| 719 | } |
| 720 | } |
| 721 | void DoPutTest::CheckDoPut(const FlightDescriptor& descr, |
| 722 | const std::shared_ptr<Schema>& schema, |
| 723 | const RecordBatchVector& batches) { |