1 files changed, 13 insertions, 0 deletions
diff --git a/tests/util.h b/tests/util.h
index bb2628f7..e0d30aa5 100644
--- a/tests/util.h
+++ b/tests/util.h
@@ -8,6 +8,7 @@
 #define TESTS_UTIL_H
 
 #include "QXmppPasswordChecker.h"
+#include "QXmppTask.h"
 
 #include <memory>
 #include <variant>
@@ -96,6 +97,18 @@ T expectFutureVariant(const QFuture<Input> &future)
     return expectVariant<T>(future.result());
 }
 
+template<typename T, typename Input>
+T expectFutureVariant(QXmppTask<Input> &future)
+{
+#define return \
+    return     \
+    {          \
+    }
+    QVERIFY(future.isFinished());
+#undef return
+    return expectVariant<T>(future.result());
+}
+
 template<typename T>
 T wait(const QFuture<T> &future)
 {