[sinttest] Add AbstractSmackIntTest.assertResult()

Flowdalic · Apr 4, 2024 · dc96484 · dc96484
1 parent 7139a43
commit dc96484
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/...integration-test/src/main/java/org/igniterealtime/smack/inttest/AbstractSmackIntTest.java b/...integration-test/src/main/java/org/igniterealtime/smack/inttest/AbstractSmackIntTest.java
@@ -1,6 +1,6 @@
 /**
  *
- * Copyright 2015-2020 Florian Schmaus
+ * Copyright 2015-2024 Florian Schmaus
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -33,6 +33,10 @@
 import org.jivesoftware.smack.XMPPException.XMPPErrorException;
 import org.jivesoftware.smack.filter.StanzaFilter;
 
+import org.igniterealtime.smack.inttest.util.ResultSyncPoint;
+
+import org.opentest4j.AssertionFailedError;
+
 public abstract class AbstractSmackIntTest {
 
     protected static final Logger LOGGER = Logger.getLogger(AbstractSmackIntTest.class.getName());
@@ -90,4 +94,18 @@ protected HttpURLConnection getHttpUrlConnectionFor(URL url) throws IOException
         }
         return urlConnection;
     }
+
+    public <R> R assertResult(ResultSyncPoint<R, ?> syncPoint, String message) throws InterruptedException, TimeoutException, AssertionFailedError {
+        return assertResult(syncPoint, timeout, message);
+    }
+
+    public static <R> R assertResult(ResultSyncPoint<R, ?> syncPoint, long timeout, String message) throws InterruptedException, TimeoutException, AssertionFailedError {
+        try {
+            return syncPoint.waitForResult(timeout);
+        } catch (InterruptedException | TimeoutException e) {
+            throw e;
+        } catch (Exception e) {
+            throw new AssertionFailedError(message, e);
+        }
+    }
 }