Function TestStraddling

cpp/src/arrow/csv/reader_test.cc:109–134

}

void TestStraddling(TableReaderFactory reader_factory) {
  // The largest line is 25 bytes, it is surrounded by many valid lines
  constexpr std::string_view kData =
      "a,b\n1,2\n3,4\n5,6\n7,8\n111111111111,22222222222\n1,2\n3,4\n5,6\n7,8\n";
  constexpr int kHeaderLineLen = 4;
  constexpr int kLargestLineLen = 25;

  auto data = std::make_shared<Buffer>(kData);
  auto input = std::make_shared<io::BufferReader>(data);
  // Sanity check that a large enough block size reads the data successfully
  ASSERT_OK_AND_ASSIGN(
      auto reader, reader_factory(input, ParseOptions::Defaults(), /*block_size=*/{}));
  ASSERT_FINISHES_OK(reader->ReadAsync());

  for (int block_size = kHeaderLineLen; block_size < kLargestLineLen / 2; ++block_size) {
    input = std::make_shared<io::BufferReader>(data);
    ASSERT_OK_AND_ASSIGN(auto reader,
                         reader_factory(input, ParseOptions::Defaults(), block_size));
    ASSERT_FINISHES_AND_RAISES(Invalid, reader->ReadAsync());
    reader.reset();
  }
  // This could deadlock in the TableReader destructor if it is not able to recover
  // gracefully after an error (GH-48741). Alternatively, the destructor could
  // deadlock at process shutdown when the thread pool is reaped.
  internal::GetCpuThreadPool()->WaitForIdle();
}

void StressTableReader(TableReaderFactory reader_factory) {
#ifdef ARROW_VALGRIND