| 535 | } |
| 536 | |
| 537 | RecordBatchVector RunNode(acero::Declaration scan_decl, bool ordered, |
| 538 | MockDataset* mock_dataset) { |
| 539 | Future<RecordBatchVector> batches_fut = |
| 540 | acero::DeclarationToBatchesAsync(std::move(scan_decl)); |
| 541 | if (ordered) { |
| 542 | mock_dataset->DeliverBatchesInOrder(GetParam().slow); |
| 543 | } else { |
| 544 | mock_dataset->DeliverBatchesRandomly(GetParam().slow); |
| 545 | } |
| 546 | EXPECT_FINISHES_OK_AND_ASSIGN(RecordBatchVector record_batches, batches_fut); |
| 547 | return record_batches; |
| 548 | } |
| 549 | |
| 550 | void CheckScannedBatches(RecordBatchVector batches) { |
| 551 | ASSERT_OK_AND_ASSIGN(std::shared_ptr<Table> batches_as_table, |
nothing calls this directly
no test coverage detected