From d2d1c2dabbc40859544a02e1c6c9ad849d864381 Mon Sep 17 00:00:00 2001 From: Abhishek Balaji Radhakrishnan Date: Mon, 3 Jun 2024 09:35:47 -0700 Subject: [PATCH] Use hamcrest for asserting order on expectation. --- .../segment/loading/SegmentLocalCacheManagerTest.java | 8 ++++++-- 1 file changed, 6 insertions(+), 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 index 4117da41831b..0bb377b8046d 100644 --- 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()); }