Function checkVectorFlagsReset

bolt/vector/tests/VectorTestUtils.cpp:269–308

}

void checkVectorFlagsReset(
    const std::function<VectorPtr()>& createVector,
    const std::function<void(VectorPtr&)>& makeMutable,
    const SelectivityVector& mutatedRows) {
  // Vector is multiply-referenced.
  auto vector = createVector();
  auto asciiRows = createAllSelectedAsciiRows(vector);
  // Mutation of dictionary or constant vectors is expected to create a new
  // vector that doesn't keep the asciiness information.
  auto updatedAsciiRows = isFlat(vector)
      ? getUpdatedAsciiRows(asciiRows, mutatedRows)
      : SelectivityVector{};
  checkVectorFlagsSet(*vector, asciiRows);
  auto another = vector;
  makeMutable(vector);
  // When vector is multiply-referenced, mutation will create a new vector and
  // hence not expected to keep the asciiness information.
  checkVectorFlagsCleared(*vector, mutatedRows, {});

  // nulls buffer is multiply-referenced.
  vector = createVector();
  checkVectorFlagsSet(*vector, asciiRows);
  auto nulls = vector->nulls();
  makeMutable(vector);
  checkVectorFlagsCleared(*vector, mutatedRows, updatedAsciiRows);

  // values buffer is multiply-referenced.
  vector = createVector();
  if (vector->isFlatEncoding()) {
    checkVectorFlagsSet(*vector, asciiRows);
    auto values = vector->values();
    makeMutable(vector);
    checkVectorFlagsCleared(*vector, mutatedRows, updatedAsciiRows);
  }

  vector = createVector();
  checkVectorFlagsSet(*vector, asciiRows);
  makeMutable(vector);
  checkVectorFlagsCleared(*vector, mutatedRows, updatedAsciiRows);
}

} // namespace bytedance::bolt::test