fix: `DataFileCollectionCompactionTest.testMergeUpdateSnapshotRestore…

…` flake (#10577) Signed-off-by: Ivan Malygin <ivan@swirldslabs.com>
hashgraph · Dec 20, 2023 · 4b2a1af · 4b2a1af
1 parent 0cdcec3
commit 4b2a1af
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...s-jasperdb/src/test/java/com/swirlds/merkledb/files/DataFileCollectionCompactionTest.java b/...s-jasperdb/src/test/java/com/swirlds/merkledb/files/DataFileCollectionCompactionTest.java
@@ -467,7 +467,8 @@ void testMergeUpdateSnapshotRestore(final int testParam) throws Throwable {
         final Future<?> f = exec.submit(() -> {
             try {
                 final List<DataFileReader<?>> filesToMerge = getFilesToMerge(store);
-                assertEquals(numFiles, filesToMerge.size());
+                // Data file collection may create a new file before the compaction starts
+                assertTrue(filesToMerge.size() == numFiles || filesToMerge.size() == numFiles + 1);
                 compactor.compactFiles(index, filesToMerge, 1);
                 // Wait for the new file to be available. Without this wait, there
                 // may be 1 or 2