1 files changed, 18 insertions, 0 deletions
diff --git a/libs/hwui/tests/common/TestContext.h b/libs/hwui/tests/common/TestContext.h
index 2bbe5dffd9b8..312988b968de 100644
--- a/libs/hwui/tests/common/TestContext.h
+++ b/libs/hwui/tests/common/TestContext.h
@@ -19,12 +19,16 @@
 
 #include <gui/DisplayEventReceiver.h>
 #include <gui/ISurfaceComposer.h>
+#include <gui/BufferItemConsumer.h>
 #include <gui/SurfaceComposerClient.h>
 #include <gui/SurfaceControl.h>
 #include <gui/Surface.h>
 #include <ui/DisplayInfo.h>
 #include <utils/Looper.h>
 
+#include <thread>
+#include <atomic>
+
 namespace android {
 namespace uirenderer {
 namespace test {
@@ -39,15 +43,29 @@ public:
     TestContext();
     ~TestContext();
 
+    // Must be called before surface();
+    void setRenderOffscreen(bool renderOffscreen) {
+        LOG_ALWAYS_FATAL_IF(mSurface.get(),
+                "Must be called before surface is created");
+        mRenderOffscreen = renderOffscreen;
+    }
+
     sp<Surface> surface();
 
     void waitForVsync();
 
 private:
+    void createSurface();
+    void createWindowSurface();
+    void createOffscreenSurface();
+
     sp<SurfaceComposerClient> mSurfaceComposerClient;
     sp<SurfaceControl> mSurfaceControl;
+    sp<BufferItemConsumer> mConsumer;
     DisplayEventReceiver mDisplayEventReceiver;
     sp<Looper> mLooper;
+    sp<Surface> mSurface;
+    bool mRenderOffscreen;
 };
 
 } // namespace test