1 files changed, 9 insertions, 1 deletions
diff --git a/tests/pysidetest/testview.h b/tests/pysidetest/testview.h
index 1a65ea459..76a6f00f5 100644
--- a/tests/pysidetest/testview.h
+++ b/tests/pysidetest/testview.h
@@ -2,12 +2,15 @@
 #define TESTVIEW_H
 
 #include <QObject>
-#include <QAbstractListModel>
 #ifdef pysidetest_EXPORTS
 #define PYSIDE_EXPORTS 1
 #endif
 #include "pysidemacros.h"
 
+class QWidget;
+class QAbstractListModel;
+class QAbstractItemDelegate;
+
 class PYSIDE_API TestView : public QObject
 {
     Q_OBJECT
@@ -15,8 +18,13 @@ public:
     TestView(QAbstractListModel* model, QObject* parent = 0) : QObject(parent), m_model(model) {}
     QAbstractListModel* model() { return m_model; }
     QVariant getData();
+
+    void setItemDelegate(QAbstractItemDelegate* delegate) { m_delegate = delegate; }
+    QWidget* getEditorWidgetFromItemDelegate() const;
+
 private:
     QAbstractListModel* m_model;
+    QAbstractItemDelegate* m_delegate;
 };
 
 #endif // TESTVIEW_H