1 files changed, 62 insertions, 0 deletions
diff --git a/tests/auto/integration/qwindowcapturebackend/grabber.cpp b/tests/auto/integration/qwindowcapturebackend/grabber.cpp
new file mode 100644
index 000000000..a7b72aeef
--- /dev/null
+++ b/tests/auto/integration/qwindowcapturebackend/grabber.cpp
@@ -0,0 +1,62 @@
+// Copyright (C) 2023 The Qt Company Ltd.
+// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only
+
+#include "grabber.h"
+#include "fixture.h"
+
+#include <qtest.h>
+#include <qvideoframe.h>
+
+FrameGrabber::FrameGrabber()
+{
+    const auto copyFrame = [this](const QVideoFrame &frame) { m_frames.push_back(frame); };
+
+    connect(this, &QVideoSink::videoFrameChanged, this, copyFrame, Qt::DirectConnection);
+}
+
+const std::vector<QVideoFrame> &FrameGrabber::getFrames() const
+{
+    return m_frames;
+}
+
+std::vector<QVideoFrame> FrameGrabber::waitAndTakeFrames(size_t minCount, qint64 noOlderThanTime)
+{
+    m_frames.clear();
+
+    const auto enoughFramesOrStopped = [this, minCount, noOlderThanTime]() -> bool {
+        if (m_stopped)
+            return true; // Stop waiting
+
+        if (noOlderThanTime > 0) {
+            // Reject frames older than noOlderThanTime
+            const auto newEnd = std::remove_if(m_frames.begin(), m_frames.end(),
+                                               [noOlderThanTime](const QVideoFrame &frame) {
+                                                   return frame.startTime() <= noOlderThanTime;
+                                               });
+            m_frames.erase(newEnd, m_frames.end());
+        }
+
+        return m_frames.size() >= minCount;
+    };
+
+    if (!QTest::qWaitFor(enoughFramesOrStopped, s_testTimeout))
+        return {};
+
+    if (m_stopped)
+        return {};
+
+    return std::exchange(m_frames, {});
+}
+
+bool FrameGrabber::isStopped() const
+{
+    return m_stopped;
+}
+
+void FrameGrabber::stop()
+{
+    qWarning() << "Stopping grabber";
+    m_stopped = true;
+}
+
+#include "moc_grabber.cpp"