Edit runStopsRetryingAfterTimeout test

redis · May 21, 2023 · 8f068f1 · 8f068f1
1 parent 12b6bbc
commit 8f068f1
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/test/java/redis/clients/jedis/ClusterCommandExecutorTest.java b/src/test/java/redis/clients/jedis/ClusterCommandExecutorTest.java
@@ -322,7 +322,8 @@ protected void sleep(long ignored) {
   public void runStopsRetryingAfterTimeout() {
     ClusterConnectionProvider connectionHandler = mock(ClusterConnectionProvider.class);
 
-    final LongConsumer sleep = mock(LongConsumer.class);
+    //final LongConsumer sleep = mock(LongConsumer.class);
+    final AtomicLong totalSleepMs = new AtomicLong();
     ClusterCommandExecutor testMe = new ClusterCommandExecutor(connectionHandler, 3, Duration.ZERO) {
       @Override
       public <T> T execute(Connection connection, CommandObject<T> commandObject) {
@@ -337,7 +338,8 @@ public <T> T execute(Connection connection, CommandObject<T> commandObject) {
 
       @Override
       protected void sleep(long sleepMillis) {
-        sleep.accept(sleepMillis);
+        //sleep.accept(sleepMillis);
+        totalSleepMs.addAndGet(sleepMillis);
       }
     };
 
@@ -347,8 +349,10 @@ protected void sleep(long sleepMillis) {
     } catch (JedisClusterOperationException e) {
       // expected
     }
-    InOrder inOrder = inOrder(connectionHandler, sleep);
+    //InOrder inOrder = inOrder(connectionHandler, sleep);
+    InOrder inOrder = inOrder(connectionHandler);
     inOrder.verify(connectionHandler).getConnection(STR_COM_OBJECT.getArguments());
     inOrder.verifyNoMoreInteractions();
+    assertEquals(0L, totalSleepMs.get());
   }
 }