Upstream oodles of Chrome for Android code into Chromium.

chrome/test/android/javatests_staging/src/org/chromium/chrome/test/util/NewTabPageTestUtils.java

+// Copyright 2015 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+package org.chromium.chrome.test.util;
+
+import android.annotation.TargetApi;
+import android.os.Build;
+
+import org.chromium.chrome.browser.Tab;
+import org.chromium.chrome.browser.ntp.IncognitoNewTabPage;
+import org.chromium.chrome.browser.ntp.NewTabPage;
+import org.chromium.chrome.browser.util.FeatureUtilities;
+import org.chromium.content.browser.test.util.Criteria;
+import org.chromium.content.browser.test.util.CriteriaHelper;
+
+/**
+ * Utilities for testing the NTP.
+ */
+public class NewTabPageTestUtils {
+
+    /**
+     * Waits for the NTP owned by the passed in tab to be fully loaded.
+     *
+     * @param tab The tab to be monitored for NTP loading.
+     * @return Whether the NTP has fully loaded.
+     */
+    @TargetApi(Build.VERSION_CODES.LOLLIPOP)
+    public static boolean waitForNtpLoaded(final Tab tab) throws InterruptedException {
+        return CriteriaHelper.pollForUIThreadCriteria(new Criteria() {
+            @Override
+            public boolean isSatisfied() {
+                if (!tab.isIncognito()) {
+                    // TODO(tedchoc): Make MostVisitedPage also have a isLoaded() concept.
+                    if (FeatureUtilities.isDocumentMode(
+                            tab.getWindowAndroid().getApplicationContext())) {
+                        return tab.getView().isAttachedToWindow();
+                    }
+                    if (!(tab.getNativePage() instanceof NewTabPage)) {
+                        return false;
+                    }
+                    return ((NewTabPage) tab.getNativePage()).isLoadedForTests();
+                } else {
+                    if (!(tab.getNativePage() instanceof IncognitoNewTabPage)) {
+                        return false;
+                    }
+                    return ((IncognitoNewTabPage) tab.getNativePage()).isLoadedForTests();
+                }
+            }
+        });
+    }
+
+}