Use hamcrest for asserting order on expectation.

apache · Jun 3, 2024 · d2d1c2d · d2d1c2d
1 parent da73aa6
commit d2d1c2d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/server/src/test/java/org/apache/druid/segment/loading/SegmentLocalCacheManagerTest.java b/server/src/test/java/org/apache/druid/segment/loading/SegmentLocalCacheManagerTest.java
@@ -54,6 +54,10 @@
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
+import java.util.Map;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.containsInAnyOrder;
 
 public class SegmentLocalCacheManagerTest
 {
@@ -207,7 +211,7 @@ public void testGetCachedSegments() throws IOException
     manager.storeInfoFile(segment2);
 
     Assert.assertTrue(manager.canHandleSegments());
-    Assert.assertEquals(ImmutableList.of(segment2, segment1), manager.getCachedSegments());
+    assertThat(manager.getCachedSegments(), containsInAnyOrder(segment1, segment2));
   }
 
   @Test
@@ -255,7 +259,7 @@ public void testGetCachedSegmentsWithMissingSegmentFile() throws IOException
     Assert.assertTrue(segment3InfoFile.exists());
 
     Assert.assertTrue(manager.canHandleSegments());
-    Assert.assertEquals(ImmutableList.of(segment2, segment1), manager.getCachedSegments());
+    assertThat(manager.getCachedSegments(), containsInAnyOrder(segment1, segment2));
     Assert.assertFalse(segment3InfoFile.exists());
   }