| 2214 | } |
| 2215 | |
| 2216 | Status StreamStateTest::ReadOutStream( |
| 2217 | BufferedTupleStream* stream, RowBatch* read_batch, int64_t& num_read) { |
| 2218 | bool eos = false; |
| 2219 | num_read = 0; |
| 2220 | do { |
| 2221 | read_batch->Reset(); |
| 2222 | RETURN_IF_ERROR(stream->GetNext(read_batch, &eos)); |
| 2223 | num_read += read_batch->num_rows(); |
| 2224 | } while (!eos); |
| 2225 | return Status::OK(); |
| 2226 | } |
| 2227 | |
| 2228 | void StreamStateTest::VerifyStreamState(BufferedTupleStream* stream, int num_page, |
| 2229 | int num_pinned_page, int num_unpinned_page, int buffer_size) { |