| Index: chrome/android/javatests/src/org/chromium/chrome/browser/webapps/TopActivityListener.java
|
| diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/webapps/TopActivityListener.java b/chrome/android/javatests/src/org/chromium/chrome/browser/webapps/TopActivityListener.java
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..e746f7ef9e885cf561e4b790229c33aee216002c
|
| --- /dev/null
|
| +++ b/chrome/android/javatests/src/org/chromium/chrome/browser/webapps/TopActivityListener.java
|
| @@ -0,0 +1,64 @@
|
| +// 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.browser.webapps;
|
| +
|
| +import android.app.Activity;
|
| +
|
| +import org.junit.rules.TestRule;
|
| +import org.junit.runner.Description;
|
| +import org.junit.runners.model.Statement;
|
| +
|
| +import org.chromium.base.ActivityState;
|
| +import org.chromium.base.ApplicationStatus;
|
| +import org.chromium.base.ApplicationStatus.ActivityStateListener;
|
| +import org.chromium.content.browser.test.util.Criteria;
|
| +import org.chromium.content.browser.test.util.CriteriaHelper;
|
| +
|
| +import javax.annotation.Nullable;
|
| +
|
| +/**
|
| + * Test rule tracking which Chrome activity is currently at the top of the task.
|
| + */
|
| +public class TopActivityListener implements TestRule {
|
| + private final ActivityStateListener mListener = new ActivityStateListener() {
|
| + @Override
|
| + public void onActivityStateChange(Activity activity, int newState) {
|
| + if (newState == ActivityState.RESUMED) {
|
| + mMostRecentActivity = activity;
|
| + }
|
| + }
|
| + };
|
| +
|
| + private Activity mMostRecentActivity;
|
| +
|
| + @Override
|
| + public Statement apply(final Statement base, Description description) {
|
| + return new Statement() {
|
| + @Override
|
| + public void evaluate() throws Throwable {
|
| + ApplicationStatus.registerStateListenerForAllActivities(mListener);
|
| + base.evaluate();
|
| + ApplicationStatus.unregisterActivityStateListener(mListener);
|
| + }
|
| + };
|
| + }
|
| +
|
| + @Nullable
|
| + public Activity getMostRecentActivity() {
|
| + return mMostRecentActivity;
|
| + }
|
| +
|
| + @SuppressWarnings("unchecked")
|
| + public <T extends Activity> T waitFor(final Class<T> expectedActivityClass) {
|
| + CriteriaHelper.pollUiThread(new Criteria() {
|
| + @Override
|
| + public boolean isSatisfied() {
|
| + return mMostRecentActivity != null
|
| + && expectedActivityClass.isAssignableFrom(mMostRecentActivity.getClass());
|
| + }
|
| + });
|
| + return (T) mMostRecentActivity;
|
| + }
|
| +}
|
|
|