[improve][test][branch-3.0] Improve ManagedLedgerTest.testGetNumberOf…

…EntriesInStorage - improve the test case based on suggestion from @poorbarcode
datastax · Mar 28, 2024 · 8f17446 · 8f17446
1 parent d3f0a4b
commit 8f17446
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/managed-ledger/src/test/java/org/apache/bookkeeper/mledger/impl/ManagedLedgerTest.java b/managed-ledger/src/test/java/org/apache/bookkeeper/mledger/impl/ManagedLedgerTest.java
@@ -2640,10 +2640,10 @@ public void testGetNumberOfEntriesInStorage() throws Exception {
         Awaitility.await().untilAsserted(() -> {
            assertEquals(managedLedger.getState(), ManagedLedgerImpl.State.ClosedLedger);
         });
-        managedLedger.rollCurrentLedgerIfFull();
+        managedLedger.createLedgerAfterClosed();
         Awaitility.await().untilAsserted(() -> {
-            assertEquals(managedLedger.getLedgersInfo().size(), 2);
-            assertEquals(managedLedger.getState(), ManagedLedgerImpl.State.ClosedLedger);
+            assertEquals(managedLedger.getLedgersInfo().size(), 3);
+            assertEquals(managedLedger.getState(), ManagedLedgerImpl.State.LedgerOpened);
         });
         assertEquals(5, managedLedger.getLedgersInfoAsList().get(0).getEntries());
         assertEquals(5, managedLedger.getLedgersInfoAsList().get(1).getEntries());