1 files changed, 10 insertions, 0 deletions
diff --git a/tests/auto/client/shared/mocksurface.h b/tests/auto/client/shared/mocksurface.h
index 68fe9b94c..3b0f01fdb 100644
--- a/tests/auto/client/shared/mocksurface.h
+++ b/tests/auto/client/shared/mocksurface.h
@@ -37,6 +37,9 @@
 
 namespace Impl {
 
+class XdgToplevelV6;
+class WlShellSurface;
+
 class Surface : public QtWaylandServer::wl_surface
 {
 public:
@@ -47,6 +50,8 @@ public:
     static Surface *fromResource(struct ::wl_resource *resource);
     void map();
     bool isMapped() const;
+    XdgToplevelV6 *xdgToplevelV6() const { return m_xdgToplevelV6; }
+    WlShellSurface *wlShellSurface() const { return m_wlShellSurface; }
 
     QSharedPointer<MockSurface> mockSurface() const { return m_mockSurface; }
 
@@ -64,11 +69,16 @@ protected:
     void surface_commit(Resource *resource) override;
 private:
     wl_resource *m_buffer = nullptr;
+    XdgToplevelV6 *m_xdgToplevelV6 = nullptr;
+    WlShellSurface *m_wlShellSurface = nullptr;
 
     Compositor *m_compositor = nullptr;
     QSharedPointer<MockSurface> m_mockSurface;
     QList<wl_resource *> m_frameCallbackList;
     bool m_mapped = false;
+
+    friend class XdgToplevelV6;
+    friend class WlShellSurface;
 };
 
 }