diff --git a/tools/ledger/src/test/java/org/apache/bookkeeper/tools/cli/commands/client/SimpleTestCommandTest.java b/tools/ledger/src/test/java/org/apache/bookkeeper/tools/cli/commands/client/SimpleTestCommandTest.java index 70ffa46f00d..24a10a9ffea 100644 --- a/tools/ledger/src/test/java/org/apache/bookkeeper/tools/cli/commands/client/SimpleTestCommandTest.java +++ b/tools/ledger/src/test/java/org/apache/bookkeeper/tools/cli/commands/client/SimpleTestCommandTest.java @@ -149,7 +149,13 @@ public void testCommand(String... args) throws Exception { // verify appends verify(wh, times(10)).append(eq(data)); - verify(rh, times(1)).read(anyLong(), anyLong()); + // verify close write handle. + verify(wh, times(1)).close(); + // verify close the read handle. + verify(rh, times(2)).close(); + // verify read entry 0-9 + verify(rh, times(1)).readUnconfirmed(0, 9); + verify(rh, times(1)).read(0, 9); } }