Function TEST

cpp/src/arrow/csv/writer_test.cc:409–438

#endif

TEST(TestWriteCSV, EmptyBatchShouldNotPolluteOutput) {
  auto schema = arrow::schema({field("col1", utf8())});
  auto empty_batch = RecordBatchFromJSON(schema, "[]");
  auto batch_a = RecordBatchFromJSON(schema, R"([{"col1": "a"}])");
  auto batch_b = RecordBatchFromJSON(schema, R"([{"col1": "b"}])");

  struct TestParam {
    std::shared_ptr<Table> table;
    std::string expected_output;
  };

  std::vector<TestParam> test_params = {
      // Empty batch in the beginning
      {Table::FromRecordBatches(schema, {empty_batch, batch_a, batch_b}).ValueOrDie(),
       "\"col1\"\n\"a\"\n\"b\"\n"},
      // Empty batch in the middle
      {Table::FromRecordBatches(schema, {batch_a, empty_batch, batch_b}).ValueOrDie(),
       "\"col1\"\n\"a\"\n\"b\"\n"},
      // Empty batch in the end
      {Table::FromRecordBatches(schema, {batch_a, batch_b, empty_batch}).ValueOrDie(),
       "\"col1\"\n\"a\"\n\"b\"\n"},
  };

  for (const auto& param : test_params) {
    ASSERT_OK_AND_ASSIGN(auto out, io::BufferOutputStream::Create());
    ASSERT_OK(WriteCSV(*param.table, WriteOptions::Defaults(), out.get()));
    ASSERT_OK_AND_ASSIGN(auto buffer, out->Finish());
    EXPECT_EQ(buffer->ToString(), param.expected_output);
  }
}

}  // namespace csv
}  // namespace arrow