Add assertConsistently

trinodb · Dec 12, 2024 · 9b24b74 · 9b24b74
1 parent 9499dc8
commit 9b24b74
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/testing/trino-testing-services/src/main/java/io/trino/testing/assertions/Assert.java b/testing/trino-testing-services/src/main/java/io/trino/testing/assertions/Assert.java
@@ -58,6 +58,27 @@ public static <E extends Exception> void assertEventually(Duration timeout, Dura
         }
     }
 
+    public static <E extends Exception> void assertConsistently(Duration timeout, Duration retryFrequency, Assert.CheckedRunnable<E> assertion)
+            throws E
+    {
+        long start = System.nanoTime();
+        while (!Thread.currentThread().isInterrupted()) {
+            assertion.run();
+
+            if (Duration.nanosSince(start).compareTo(timeout) > 0) {
+                return;
+            }
+
+            try {
+                Thread.sleep(retryFrequency.toMillis());
+            }
+            catch (InterruptedException e) {
+                Thread.currentThread().interrupt();
+                throw new RuntimeException(e);
+            }
+        }
+    }
+
     public interface CheckedRunnable<E extends Exception>
     {
         void run()