| 2049 | : generator(std::move(generator)), next() {} |
| 2050 | |
| 2051 | void AssertCanConsume() { |
| 2052 | if (!next.is_valid()) { |
| 2053 | next = generator(); |
| 2054 | } |
| 2055 | ASSERT_FINISHES_OK(next); |
| 2056 | next = Future<EnumeratedRecordBatch>(); |
| 2057 | } |
| 2058 | |
| 2059 | void AssertCannotConsume() { |
| 2060 | if (!next.is_valid()) { |