Method AssertSkip

cpp/src/arrow/csv/chunker_test.cc:74–102

  void MakeChunker() { chunker_ = ::arrow::csv::MakeChunker(options_); }

  void AssertSkip(const std::string& str, int64_t count, int64_t rem_count,
                  int64_t rest_size) {
    MakeChunker();
    {
      auto test_count = count;
      auto partial = std::make_shared<Buffer>("");
      auto block = std::make_shared<Buffer>(reinterpret_cast<const uint8_t*>(str.data()),
                                            static_cast<int64_t>(str.size()));
      std::shared_ptr<Buffer> rest;
      ASSERT_OK(chunker_->ProcessSkip(partial, block, true, &test_count, &rest));
      ASSERT_EQ(rem_count, test_count);
      ASSERT_EQ(rest_size, rest->size());
      AssertBufferEqual(*SliceBuffer(block, block->size() - rest_size), *rest);
    }
    {
      auto test_count = count;
      auto split = static_cast<int64_t>(str.find_first_of('\n'));
      auto partial =
          std::make_shared<Buffer>(reinterpret_cast<const uint8_t*>(str.data()), split);
      auto block =
          std::make_shared<Buffer>(reinterpret_cast<const uint8_t*>(str.data() + split),
                                   static_cast<int64_t>(str.size()) - split);
      std::shared_ptr<Buffer> rest;
      ASSERT_OK(chunker_->ProcessSkip(partial, block, true, &test_count, &rest));
      ASSERT_EQ(rem_count, test_count);
      ASSERT_EQ(rest_size, rest->size());
      AssertBufferEqual(*SliceBuffer(block, block->size() - rest_size), *rest);
    }
  }

  ParseOptions options_;
  std::unique_ptr<Chunker> chunker_;