commit 7a417fe3204e812f9f41f9ac325140c5e134d9d1
author TreeHugger Robot <treehugger-gerrit@google.com> Tue Mar 06 00:39:32 2018 +0000
committer Android (Google) Code Review <android-gerrit@google.com> Tue Mar 06 00:39:32 2018 +0000
tree b44d8c79ff0296b0967c03c505d1173eae355335
parent cfdd49487ca4ae3cfa3abf83fe1bb6d01313e81c
parent e8bcc9e034aabf37ec8a591fe53469839e7b0f9b

Merge "Add back de-flake workaround" into pi-dev

tests/tests/uirendering/src/android/uirendering/cts/testinfrastructure/DrawActivity.java

diff --git a/tests/tests/uirendering/src/android/uirendering/cts/testinfrastructure/DrawActivity.java b/tests/tests/uirendering/src/android/uirendering/cts/testinfrastructure/DrawActivity.java
index 54ce087..b8ba73b 100644
--- a/tests/tests/uirendering/src/android/uirendering/cts/testinfrastructure/DrawActivity.java
+++ b/tests/tests/uirendering/src/android/uirendering/cts/testinfrastructure/DrawActivity.java
@@ -189,8 +189,16 @@
     }
 
     private class DrawCounterListener implements ViewTreeObserver.OnDrawListener {
+        private static final int DEBUG_REQUIRE_EXTRA_FRAMES = 1;
+        private int mDrawCount = 0;
+
         @Override
         public void onDraw() {
+            if (++mDrawCount <= DEBUG_REQUIRE_EXTRA_FRAMES) {
+                mView.postInvalidate();
+                return;
+            }
+
             long vsyncMillis = mView.getDrawingTime();
 
             mView.post(() -> mView.getViewTreeObserver().removeOnDrawListener(this));