Function TEST

cpp/fory/row/row_test.cc:32–99

namespace row {

TEST(RowTest, write) {
  auto f1 = field("f1", utf8());
  auto f2 = field("f2", int32());
  auto arr_type = list(int32());
  auto f3 = field("f3", arr_type);
  auto map_type = map(utf8(), float32());
  auto f4 = field("f4", map_type);
  auto struct_type = std::dynamic_pointer_cast<StructType>(
      struct_({field("n1", utf8()), field("n2", int32())}));
  auto f5 = field("f5", struct_type);
  std::vector<FieldPtr> fields = {f1, f2, f3, f4, f5};
  auto s = schema(fields);

  RowWriter row_writer(s);
  row_writer.reset();
  row_writer.write_string(0, std::string("str"));
  row_writer.write(1, static_cast<int32_t>(1));

  // array
  row_writer.set_not_null_at(2);
  int start = row_writer.cursor();
  ArrayWriter array_writer(arr_type, &row_writer);
  array_writer.reset(2);
  array_writer.write(0, static_cast<int32_t>(2));
  array_writer.write(1, static_cast<int32_t>(2));
  EXPECT_EQ(array_writer.copy_to_array_data()->to_string(),
            std::string("[2, 2]"));
  row_writer.set_offset_and_size(2, start, row_writer.cursor() - start);

  // map
  row_writer.set_not_null_at(3);
  int offset = row_writer.cursor();
  row_writer.write_directly(-1);
  ArrayWriter key_array_writer(list(utf8()), &row_writer);
  key_array_writer.reset(2);
  key_array_writer.write_string(0, "key1");
  key_array_writer.write_string(1, "key2");
  EXPECT_EQ(key_array_writer.copy_to_array_data()->to_string(),
            std::string("[key1, key2]"));
  row_writer.write_directly(offset, key_array_writer.size());
  ArrayWriter value_array_writer(list(float32()), &row_writer);
  value_array_writer.reset(2);
  value_array_writer.write(0, 1.0f);
  value_array_writer.write(1, 1.0f);
  EXPECT_EQ(value_array_writer.copy_to_array_data()->to_string(),
            std::string("[1, 1]"));
  int size = row_writer.cursor() - offset;
  row_writer.set_offset_and_size(3, offset, size);

  // struct
  RowWriter struct_writer(schema(struct_type->fields()), &row_writer);
  row_writer.set_not_null_at(4);
  offset = row_writer.cursor();
  struct_writer.reset();
  struct_writer.write_string(0, "str");
  struct_writer.write(1, 1);
  size = row_writer.cursor() - offset;
  row_writer.set_offset_and_size(4, offset, size);