Method TestAddAndGetZeroSizedRows

be/src/runtime/buffered-tuple-stream-test.cc:2297–2343

}

void StreamStateTest::TestAddAndGetZeroSizedRows() {
  Init(BUFFER_POOL_LIMIT);
  BufferedTupleStream stream(
      runtime_state_, zero_sized_row_desc_, &client_, PAGE_LEN, PAGE_LEN);
  ASSERT_OK(stream.Init("StreamStateTest::TestAddAndGetZeroSizedRows", false));
  bool got_reservation = false;
  ASSERT_OK(stream.PrepareForReadWrite(true, &got_reservation));
  ASSERT_TRUE(got_reservation);

  RowBatch write_batch(zero_sized_row_desc_, 1, &tracker_);
  write_batch.CommitRows(1);

  // Adding 1 row to initialize the row counters
  TestAddOneZeroSizedRow(stream, write_batch);

  // Set the row counters to mock a stream with INT_MAX rows
  stream.num_rows_ = INT_MAX;
  stream.write_page_->num_rows = INT_MAX;

  // Test if the stream can hold more than INT_MAX rows.
  TestAddOneZeroSizedRow(stream, write_batch);

  // Set the row counters to mock a stream with UINT_MAX rows
  stream.num_rows_ = UINT_MAX;
  stream.write_page_->num_rows = UINT_MAX;

  // Test if the stream can hold more than UINT_MAX rows.
  TestAddOneZeroSizedRow(stream, write_batch);

  stream.DoneWriting();

  // Test if we can get 1 row after getting 0 rows.
  TestGetOneZeroSizedRow(stream);

  // Test if we can get 1 row after getting INT_MAX rows.
  stream.read_it_.IncrRowsReturned(INT_MAX - 1);
  ASSERT_EQ(stream.read_it_.read_page_rows_returned_, INT_MAX);
  TestGetOneZeroSizedRow(stream);

  // Test if we can get 1 row after getting UINT_MAX rows.
  stream.read_it_.IncrRowsReturned(UINT_MAX - INT_MAX - 1);
  ASSERT_EQ(stream.read_it_.read_page_rows_returned_, UINT_MAX);
  TestGetOneZeroSizedRow(stream);

  ASSERT_EQ(stream.read_it_.GetRowsLeftInPage(), 0);
  stream.Close(nullptr, RowBatch::FlushMode::NO_FLUSH_RESOURCES);
}

TEST_F(StreamStateTest, AddAndGetZeroSizedRows) {
  TestAddAndGetZeroSizedRows();