update test partition stat write impl

Author: danielhumanmod

quarkus/service/src/test/java/org/apache/polaris/service/quarkus/task/BatchFileCleanupTaskHandlerTest.java

@@ -128,8 +128,7 @@ public void close() {
       PartitionStatisticsFile partitionStatisticsFile1 =
           TaskTestUtils.writePartitionStatsFile(
               snapshot.snapshotId(),
-              snapshot.sequenceNumber(),
-              "/metadata/" + UUID.randomUUID() + ".stats",
+              "/metadata/" + "partition-stats-" + UUID.randomUUID() + ".parquet",
               fileIO);
       String firstMetadataFile = "v1-295495059.metadata.json";
       TableMetadata firstMetadata =
@@ -162,8 +161,7 @@ public void close() {
       PartitionStatisticsFile partitionStatisticsFile2 =
           TaskTestUtils.writePartitionStatsFile(
               snapshot2.snapshotId(),
-              snapshot2.sequenceNumber(),
-              "/metadata/" + UUID.randomUUID() + ".stats",
+              "/metadata/" + "partition-stats-" + UUID.randomUUID() + ".parquet",
               fileIO);
       String secondMetadataFile = "v1-295495060.metadata.json";
       TableMetadata secondMetadata =
@@ -182,8 +180,6 @@ public void close() {
           Stream.of(
                   secondMetadata.previousFiles().stream().map(TableMetadata.MetadataLogEntry::file),
                   secondMetadata.statisticsFiles().stream().map(StatisticsFile::path),
-                  firstMetadata.partitionStatisticsFiles().stream()
-                      .map(PartitionStatisticsFile::path),
                   secondMetadata.partitionStatisticsFiles().stream()
                       .map(PartitionStatisticsFile::path))
               .flatMap(s -> s)

quarkus/service/src/test/java/org/apache/polaris/service/quarkus/task/TableCleanupTaskHandlerTest.java

@@ -510,8 +510,7 @@ public void testTableCleanupMultipleMetadata() throws IOException {
     PartitionStatisticsFile partitionStatisticsFile1 =
         TaskTestUtils.writePartitionStatsFile(
             snapshot.snapshotId(),
-            snapshot.sequenceNumber(),
-            "/metadata/" + UUID.randomUUID() + ".stats",
+            "/metadata/" + "partition-stats-" + UUID.randomUUID() + ".parquet",
             fileIO);
     String firstMetadataFile = "v1-295495059.metadata.json";
     TableMetadata firstMetadata =
@@ -544,8 +543,7 @@ public void testTableCleanupMultipleMetadata() throws IOException {
     PartitionStatisticsFile partitionStatisticsFile2 =
         TaskTestUtils.writePartitionStatsFile(
             snapshot2.snapshotId(),
-            snapshot2.sequenceNumber(),
-            "/metadata/" + UUID.randomUUID() + ".stats",
+            "/metadata/" + "partition-stats-" + UUID.randomUUID() + ".parquet",
             fileIO);
     String secondMetadataFile = "v1-295495060.metadata.json";
     TaskTestUtils.writeTableMetadata(

quarkus/service/src/test/java/org/apache/polaris/service/quarkus/task/TaskTestUtils.java

@@ -20,6 +20,7 @@
 
 import jakarta.annotation.Nonnull;
 import java.io.IOException;
+import java.io.UncheckedIOException;
 import java.nio.ByteBuffer;
 import java.nio.charset.StandardCharsets;
 import java.util.Arrays;
@@ -181,24 +182,19 @@ public static StatisticsFile writeStatsFile(
   }
 
   public static PartitionStatisticsFile writePartitionStatsFile(
-      long snapshotId, long snapshotSequenceNumber, String statsLocation, FileIO fileIO)
-      throws IOException {
-
-    try (PuffinWriter puffinWriter = Puffin.write(fileIO.newOutputFile(statsLocation)).build()) {
-      puffinWriter.add(
-          new Blob(
-              "some-blob-type",
-              List.of(1),
-              snapshotId,
-              snapshotSequenceNumber,
-              ByteBuffer.wrap("blob content".getBytes(StandardCharsets.UTF_8))));
-      puffinWriter.finish();
-
-      return ImmutableGenericPartitionStatisticsFile.builder()
-          .snapshotId(snapshotId)
-          .path(statsLocation)
-          .fileSizeInBytes(puffinWriter.fileSize())
-          .build();
+      long snapshotId, String statsLocation, FileIO fileIO) throws UncheckedIOException {
+    PositionOutputStream positionOutputStream;
+    try {
+      positionOutputStream = fileIO.newOutputFile(statsLocation).create();
+      positionOutputStream.close();
+    } catch (IOException e) {
+      throw new UncheckedIOException(e);
     }
+
+    return ImmutableGenericPartitionStatisticsFile.builder()
+        .snapshotId(snapshotId)
+        .path(statsLocation)
+        .fileSizeInBytes(42L)
+        .build();
   }
 }