1 files changed, 39 insertions, 1 deletions
diff --git a/src/qmltest/quicktestresult.cpp b/src/qmltest/quicktestresult.cpp
index 8c62196128..90d11e9734 100644
--- a/src/qmltest/quicktestresult.cpp
+++ b/src/qmltest/quicktestresult.cpp
@@ -1,5 +1,6 @@
 /****************************************************************************
 **
+** Copyright (C) 2017 Crimson AS <info@crimson.no>
 ** Copyright (C) 2016 The Qt Company Ltd.
 ** Contact: https://www.qt.io/licensing/
 **
@@ -61,7 +62,10 @@
 #include <QtCore/QDir>
 #include <QtQuick/qquickwindow.h>
 #include <QtGui/qvector3d.h>
+#include <QtGui/qimagewriter.h>
 #include <QtQml/private/qqmlglobal_p.h>
+#include <QtQml/QQmlEngine>
+#include <QtQml/QQmlContext>
 #include <private/qv4qobjectwrapper_p.h>
 
 #include <algorithm>
@@ -77,6 +81,11 @@ extern bool qWaitForSignal(QObject *obj, const char* signal, int timeout = 5000)
 class Q_QUICK_TEST_EXPORT QuickTestImageObject : public QObject
 {
     Q_OBJECT
+
+    Q_PROPERTY(int width READ width CONSTANT)
+    Q_PROPERTY(int height READ height CONSTANT)
+    Q_PROPERTY(QSize size READ size CONSTANT)
+
 public:
     QuickTestImageObject(const QImage& img, QObject *parent = 0)
         : QObject(parent)
@@ -127,6 +136,33 @@ public Q_SLOTS:
 
         return m_image == other->m_image;
     }
+
+    void save(const QString &filePath)
+    {
+        QImageWriter writer(filePath);
+        if (!writer.write(m_image)) {
+            QQmlEngine *engine = qmlContext(this)->engine();
+            QV4::ExecutionEngine *v4 = QV8Engine::getV4(engine->handle());
+            v4->throwError(QStringLiteral("Can't save to %1: %2").arg(filePath, writer.errorString()));
+        }
+    }
+
+public:
+    int width() const
+    {
+        return m_image.width();
+    }
+
+    int height() const
+    {
+        return m_image.height();
+    }
+
+    QSize size() const
+    {
+        return m_image.size();
+    }
+
 private:
     QImage m_image;
 };
@@ -718,7 +754,9 @@ QObject *QuickTestResult::grabImage(QQuickItem *item)
         QImage grabbed = window->grabWindow();
         QRectF rf(item->x(), item->y(), item->width(), item->height());
         rf = rf.intersected(QRectF(0, 0, grabbed.width(), grabbed.height()));
-        return new QuickTestImageObject(grabbed.copy(rf.toAlignedRect()));
+        QObject *o = new QuickTestImageObject(grabbed.copy(rf.toAlignedRect()));
+        QQmlEngine::setContextForObject(o, qmlContext(this));
+        return o;
     }
     return 0;
 }