Function TestStringDictionaryMakeBuilder

cpp/src/arrow/array/array_dict_test.cc:489–509

template <typename BuilderType>
void TestStringDictionaryMakeBuilder(const std::shared_ptr<DataType>& value_type) {
  auto dict_array = ArrayFromJSON(value_type, R"(["test", "test2"])");
  auto dict_type = dictionary(int8(), value_type);
  auto int_array = ArrayFromJSON(int8(), "[0, 1, 0]");
  std::unique_ptr<ArrayBuilder> boxed_builder;
  ASSERT_OK(MakeBuilder(default_memory_pool(), dict_type, &boxed_builder));
  auto& builder = checked_cast<BuilderType&>(*boxed_builder);

  // Build the dictionary Array
  ASSERT_OK(builder.Append("test"));
  ASSERT_OK(builder.Append("test2"));
  ASSERT_OK(builder.Append("test"));

  std::shared_ptr<Array> result;
  ASSERT_OK(builder.Finish(&result));

  // Build expected data
  DictionaryArray expected(dict_type, int_array, dict_array);

  AssertArraysEqual(expected, *result);
}

TEST(TestStringDictionaryBuilder, MakeBuilder) {
  TestStringDictionaryMakeBuilder<DictionaryBuilder<StringType>>(utf8());