test: Add a spy type that waits for one or more signals

[quassel.git] / src / test / util / invocationspy.cpp

diff --git a/src/test/util/invocationspy.cpp b/src/test/util/invocationspy.cpp
index 74bb3bd..da9c4f0 100644 (file)
--- a/src/test/util/invocationspy.cpp
+++ b/src/test/util/invocationspy.cpp
@@ -43,4 +43,16 @@ bool InvocationSpy::wait(std::chrono::milliseconds timeout)
     return result;
 }
 
+// -----------------------------------------------------------------------------------------------------------------------------------------
+
+bool SignalSpy::wait(std::chrono::milliseconds timeout)
+{
+    bool result = InvocationSpy::wait(timeout);
+    for (auto&& connection : _connections) {
+        QObject::disconnect(connection);
+    }
+    _connections.clear();
+    return result;
+}
+
 }  // namespace test