fix fail test

apache · Jan 20, 2025 · 5a675f9 · 5a675f9
1 parent f795ba0
commit 5a675f9
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/...e/src/test/java/org/apache/rocketmq/store/timer/rocksdb/TimerMessageRocksDBStoreTest.java b/...e/src/test/java/org/apache/rocketmq/store/timer/rocksdb/TimerMessageRocksDBStoreTest.java
@@ -282,11 +282,12 @@ public void testDeleteTimerMessage() throws Exception {
         assertEquals(PutMessageStatus.PUT_OK, messageStore.putMessage(delMsg).getPutMessageStatus());
 
         // The first one should have been deleted.
-        ByteBuffer msgBuff = getOneMessage(topic, 0, 0, 3000);
-        assertNotNull(msgBuff);
-        MessageExt msgExt = MessageDecoder.decode(msgBuff);
-        assertNotNull(msgExt);
-
+        for (int i = 1; i < 4; i++) {
+            ByteBuffer msgBuff = getOneMessage(topic, 0, i, 3000);
+            assertNotNull(msgBuff);
+            MessageExt msgExt = MessageDecoder.decode(msgBuff);
+            assertNotNull(msgExt);
+        }
         // The last one should be null.
         assertNull(getOneMessage(topic, 0, 4, 500));
         timerMessageStore.shutdown();