| 385 | class MockFragment : public Fragment { |
| 386 | public: |
| 387 | explicit MockFragment(uint32_t start, int64_t rows_per_batch, int num_batches, |
| 388 | const std::shared_ptr<Schema>& schema) |
| 389 | : Fragment(compute::literal(true), schema), |
| 390 | start_(start), |
| 391 | rows_per_batch_(rows_per_batch), |
| 392 | num_batches_(num_batches) {} |
| 393 | |
| 394 | Result<RecordBatchGenerator> ScanBatchesAsync( |
| 395 | const std::shared_ptr<ScanOptions>& options) override { |