fix some test using new api

apache · Jan 4, 2024 · 32cc331 · 32cc331
1 parent fb4abba
commit 32cc331
Showing 1 changed file with 5 additions and 11 deletions.
diff --git a/core/src/test/java/org/apache/iceberg/TestBase.java b/core/src/test/java/org/apache/iceberg/TestBase.java
@@ -399,9 +399,9 @@ void validateSnapshot(Snapshot old, Snapshot snap, Long sequenceNumber, DataFile
           .isTrue();
     }
 
-    Assertions.assertThat(newManifests.size())
+    Assertions.assertThat(newManifests)
         .as("Should create 1 new manifest and reuse old manifests")
-        .isEqualTo(1);
+        .hasSize(1);
     ManifestFile manifest = newManifests.get(0);
 
     long id = snap.snapshotId();
@@ -542,9 +542,7 @@ void validateManifest(
       }
     }
 
-    Assertions.assertThat(expectedFiles.hasNext())
-        .as("Should find all files in the manifest")
-        .isFalse();
+    Assertions.assertThat(expectedFiles).as("Should find all files in the manifest").isExhausted();
   }
 
   void validateDeleteManifest(
@@ -572,9 +570,7 @@ void validateDeleteManifest(
           .isEqualTo(statuses.next());
     }
 
-    Assertions.assertThat(expectedFiles.hasNext())
-        .as("Should find all files in the manifest")
-        .isFalse();
+    Assertions.assertThat(expectedFiles).as("Should find all files in the manifest").isExhausted();
   }
 
   private <T extends ContentFile<T>> void validateManifestSequenceNumbers(
@@ -697,9 +693,7 @@ static void validateManifestEntries(
           .isEqualTo(expectedStatus);
     }
 
-    Assertions.assertThat(expectedFiles.hasNext())
-        .as("Should find all files in the manifest")
-        .isFalse();
+    Assertions.assertThat(expectedFiles).as("Should find all files in the manifest").isExhausted();
   }
 
   static Iterator<ManifestEntry.Status> statuses(ManifestEntry.Status... statuses) {