commit e34a2bff00c14722e240edce7ec7e6acd2713213
author Ashita Sriraman <ashitas@google.com> Tue Jan 19 12:48:14 2016 -0800
committer The Android Automerger <android-build@google.com> Thu Jan 21 18:14:46 2016 -0800
tree 9021cf16fc143e2a83da5deb06d72cb60b9d6bdf
parent f7d68a1b4ebcd22369c6580d3a4693b6c6519926

YouTube jank test fix

Change-Id: I8f5a21bf8d03c4e0e9327296b4e12cc0fac5dd5d

tests/jank/sysapp/src/com/android/sysapp/janktests/YouTubeJankTests.java

diff --git a/tests/jank/sysapp/src/com/android/sysapp/janktests/YouTubeJankTests.java b/tests/jank/sysapp/src/com/android/sysapp/janktests/YouTubeJankTests.java
index d0614d4..67765f6 100644
--- a/tests/jank/sysapp/src/com/android/sysapp/janktests/YouTubeJankTests.java
+++ b/tests/jank/sysapp/src/com/android/sysapp/janktests/YouTubeJankTests.java
@@ -136,6 +136,11 @@
                     Until.findObject(By.res(PACKAGE_NAME, "ok").text("OK")), LONG_TIMEOUT);
             Assert.assertNotNull("No 'ok' button to bypass music", ok);
             ok.click();
-      }
+        }
+        UiObject2 laterButton = mDevice.wait(
+            Until.findObject(By.res(PACKAGE_NAME, "later_button")), LONG_TIMEOUT);
+        if (laterButton != null) {
+            laterButton.click();
+         }
     }
 }