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

…EntriesInStorage - improve the test case based on suggestion from @poorbarcode
apache · Mar 28, 2024 · 6255b12 · 6255b12
1 parent 9927b55
commit 6255b12
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
@@ -2641,10 +2641,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());