Index: chrome/test/android/javatests/src/org/chromium/chrome/test/MultiActivityTestCommon.java |
diff --git a/chrome/test/android/javatests/src/org/chromium/chrome/test/MultiActivityTestCommon.java b/chrome/test/android/javatests/src/org/chromium/chrome/test/MultiActivityTestCommon.java |
new file mode 100644 |
index 0000000000000000000000000000000000000000..576777e99369433a407a0bc24045c410a22ee739 |
--- /dev/null |
+++ b/chrome/test/android/javatests/src/org/chromium/chrome/test/MultiActivityTestCommon.java |
@@ -0,0 +1,68 @@ |
+// Copyright 2017 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; |
+ |
+import android.app.Instrumentation; |
+import android.content.Context; |
+import android.text.TextUtils; |
+ |
+import org.chromium.base.metrics.RecordHistogram; |
+import org.chromium.chrome.browser.ChromeActivity; |
+import org.chromium.chrome.browser.tab.Tab; |
+import org.chromium.chrome.browser.tabmodel.document.DocumentTabModelSelector; |
+import org.chromium.chrome.test.util.ApplicationTestUtils; |
+import org.chromium.chrome.test.util.browser.tabmodel.document.MockStorageDelegate; |
+import org.chromium.content.browser.test.util.Criteria; |
+import org.chromium.content.browser.test.util.CriteriaHelper; |
+ |
+// TODO(yolandyan): move this class to its test rule once JUnit4 migration is over |
+final class MultiActivityTestCommon { |
+ private final MultiActivityTestCommonCallback mCallback; |
+ MockStorageDelegate mStorageDelegate; |
+ Context mContext; |
+ |
+ MultiActivityTestCommon(MultiActivityTestCommonCallback callback) { |
+ mCallback = callback; |
+ } |
+ |
+ void setUp() throws Exception { |
+ RecordHistogram.setDisabledForTests(true); |
+ mContext = mCallback.getInstrumentation().getTargetContext(); |
+ ApplicationTestUtils.setUp(mContext, true); |
+ |
+ // Make the DocumentTabModelSelector use a mocked out directory so that test runs don't |
+ // interfere with each other. |
+ mStorageDelegate = new MockStorageDelegate(mContext.getCacheDir()); |
+ DocumentTabModelSelector.setStorageDelegateForTests(mStorageDelegate); |
+ } |
+ |
+ void tearDown() throws Exception { |
+ mStorageDelegate.ensureDirectoryDestroyed(); |
+ ApplicationTestUtils.tearDown(mContext); |
+ RecordHistogram.setDisabledForTests(false); |
+ } |
+ |
+ void waitForFullLoad(final ChromeActivity activity, final String expectedTitle) { |
+ waitForFullLoad(activity, expectedTitle, false); |
+ } |
+ |
+ void waitForFullLoad( |
+ final ChromeActivity activity, final String expectedTitle, boolean waitLongerForLoad) { |
+ ApplicationTestUtils.assertWaitForPageScaleFactorMatch(activity, 0.5f, waitLongerForLoad); |
+ final Tab tab = activity.getActivityTab(); |
+ assert tab != null; |
+ |
+ CriteriaHelper.pollUiThread(new Criteria() { |
+ @Override |
+ public boolean isSatisfied() { |
+ if (!tab.isLoadingAndRenderingDone()) return false; |
+ if (!TextUtils.equals(expectedTitle, tab.getTitle())) return false; |
+ return true; |
+ } |
+ }); |
+ } |
+ |
+ public interface MultiActivityTestCommonCallback { Instrumentation getInstrumentation(); } |
+} |