fix unit tests

Author: HuaHuaY

test/parquet_test.cc

@@ -123,6 +123,31 @@ class ParquetReaderTest : public ::testing::Test {
         array, {.path = temp_parquet_file_, .schema = schema, .io = file_io_}));
   }
 
+  void CreateSplitParquetFile() {
+    const std::string kParquetFieldIdKey = "PARQUET:field_id";
+    auto arrow_schema = ::arrow::schema(
+        {::arrow::field("id", ::arrow::int32(), /*nullable=*/false,
+                        ::arrow::KeyValueMetadata::Make({kParquetFieldIdKey}, {"1"})),
+         ::arrow::field("name", ::arrow::utf8(), /*nullable=*/true,
+                        ::arrow::KeyValueMetadata::Make({kParquetFieldIdKey}, {"2"}))});
+    auto table = ::arrow::Table::FromRecordBatches(
+                     arrow_schema, {::arrow::RecordBatch::FromStructArray(
+                                        ::arrow::json::ArrayFromJSONString(
+                                            ::arrow::struct_(arrow_schema->fields()),
+                                            R"([[1, "Foo"], [2, "Bar"], [3, "Baz"]])")
+                                            .ValueOrDie())
+                                        .ValueOrDie()})
+                     .ValueOrDie();
+
+    auto io = internal::checked_cast<arrow::ArrowFileSystemFileIO&>(*file_io_);
+    auto outfile = io.fs()->OpenOutputStream(temp_parquet_file_).ValueOrDie();
+
+    ASSERT_TRUE(::parquet::arrow::WriteTable(*table, ::arrow::default_memory_pool(),
+                                             outfile, /*chunk_size=*/2)
+                    .ok());
+    ASSERT_TRUE(outfile->Close().ok());
+  }
+
   void VerifyNextBatch(Reader& reader, std::string_view expected_json) {
     // Boilerplate to get Arrow schema
     auto schema_result = reader.Schema();
@@ -213,7 +238,7 @@ TEST_F(ParquetReaderTest, ReadWithBatchSize) {
 }
 
 TEST_F(ParquetReaderTest, ReadSplit) {
-  CreateSimpleParquetFile();
+  CreateSplitParquetFile();
 
   // Read split offsets
   auto io = internal::checked_cast<arrow::ArrowFileSystemFileIO&>(*file_io_);