[Multi-Instance] Add method to detect Multi-Instance

This change provides a simple method to detect if Chrome is in
multi-instance mode.

Bug: 1071925
Change-Id: I3b82e5fe7a97e923f67f0ca49735c724f9ed4f17
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2155017
Commit-Queue: Calder Kitagawa <ckitagawa@chromium.org>
Reviewed-by: Theresa  <twellington@chromium.org>
Cr-Commit-Position: refs/heads/master@{#762344}

chrome/android/java/src/org/chromium/chrome/browser/multiwindow/MultiInstanceManager.java

@@ -247,14 +247,8 @@ public class MultiInstanceManager
         List<ActivityManager.AppTask> appTasks = activityManager.getAppTasks();
         ActivityManager.AppTask otherActivityTask = null;
         for (ActivityManager.AppTask task : appTasks) {
-            if (task.getTaskInfo() == null || task.getTaskInfo().baseActivity == null) continue;
-            String baseActivity = task.getTaskInfo().baseActivity.getClassName();
-            // Contrary to the documentation task.getTaskInfo().baseActivity for the .LauncherMain
-            // activity alias is the alias itself, and not the implementation. Filed b/66729258;
-            // for now translate the alias manually.
-            if (baseActivity.equals(ChromeTabbedActivity.MAIN_LAUNCHER_ACTIVITY_NAME)) {
-                baseActivity = ChromeTabbedActivity.class.getName();
-            }
+            String baseActivity = MultiWindowUtils.getActivityNameFromTask(task);
+
             if (baseActivity.equals(otherWindowActivityClass.getName())) {
                 otherActivityTask = task;
             }

chrome/android/java/src/org/chromium/chrome/browser/multiwindow/MultiWindowUtils.java

@@ -191,6 +191,61 @@ public class MultiWindowUtils implements ActivityStateListener {
         }
     }
 
+    /**
+     * Determines the name of an activity from its {@link AppTask}.
+     * @param task The AppTask to get the name of.
+     */
+    public static String getActivityNameFromTask(AppTask task) {
+        if (task.getTaskInfo() == null || task.getTaskInfo().baseActivity == null) return "";
+
+        String baseActivity = task.getTaskInfo().baseActivity.getClassName();
+        // Contrary to the documentation task.getTaskInfo().baseActivity for the .LauncherMain
+        // activity alias is the alias itself, and not the implementation. Filed b/66729258;
+        // for now translate the alias manually.
+        if (TextUtils.equals(baseActivity, ChromeTabbedActivity.MAIN_LAUNCHER_ACTIVITY_NAME)) {
+            baseActivity = ChromeTabbedActivity.class.getName();
+        }
+        return baseActivity;
+    }
+
+    /**
+     * Determines if multiple instances of Chrome are running.
+     * @param context The current Context, used to retrieve the ActivityManager system service.
+     * @return True if multiple instances of Chrome are running.
+     */
+    public boolean areMultipleChromeInstancesRunning(Context context) {
+        // Exit early if multi-window isn't supported.
+        if (Build.VERSION.SDK_INT <= Build.VERSION_CODES.M) return false;
+
+        // Check if both tasks are running.
+        boolean tabbedTaskRunning = false;
+        boolean tabbed2TaskRunning = false;
+        for (Activity activity : ApplicationStatus.getRunningActivities()) {
+            if (activity.getClass().equals(ChromeTabbedActivity.class)) {
+                tabbedTaskRunning = true;
+            } else if (activity.getClass().equals(ChromeTabbedActivity2.class)) {
+                tabbed2TaskRunning = true;
+            }
+        }
+        if (tabbedTaskRunning && tabbed2TaskRunning) return true;
+
+        // If a task isn't running check if it is in recents since another instance could be
+        // recovered from there.
+        ActivityManager activityManager =
+                (ActivityManager) context.getSystemService(Context.ACTIVITY_SERVICE);
+        List<AppTask> appTasks = activityManager.getAppTasks();
+        for (AppTask task : appTasks) {
+            String baseActivity = getActivityNameFromTask(task);
+
+            if (TextUtils.equals(baseActivity, ChromeTabbedActivity.class.getName())) {
+                tabbedTaskRunning = true;
+            } else if (TextUtils.equals(baseActivity, ChromeTabbedActivity2.class.getName())) {
+                tabbed2TaskRunning = true;
+            }
+        }
+        return tabbedTaskRunning && tabbed2TaskRunning;
+    }
+
     /**
      * Determines the correct ChromeTabbedActivity class to use for an incoming intent.
      * @param intent The incoming intent that is starting ChromeTabbedActivity.
@@ -284,12 +339,7 @@ public class MultiWindowUtils implements ActivityStateListener {
                 context.getSystemService(Context.ACTIVITY_SERVICE);
         List<AppTask> appTasks = activityManager.getAppTasks();
         for (AppTask task : appTasks) {
-            if (task.getTaskInfo() == null || task.getTaskInfo().baseActivity == null) continue;
-            String baseActivity = task.getTaskInfo().baseActivity.getClassName();
-
-            if (TextUtils.equals(baseActivity, ChromeTabbedActivity.MAIN_LAUNCHER_ACTIVITY_NAME)) {
-                baseActivity = ChromeTabbedActivity.class.getName();
-            }
+            String baseActivity = getActivityNameFromTask(task);
 
             if (TextUtils.equals(baseActivity, className)) return true;
         }

chrome/android/javatests/src/org/chromium/chrome/browser/multiwindow/MultiWindowUtilsTest.java

@@ -7,10 +7,12 @@ package org.chromium.chrome.browser.multiwindow;
 import static org.chromium.chrome.browser.multiwindow.MultiWindowTestHelper.createSecondChromeTabbedActivity;
 
 import android.annotation.TargetApi;
+import android.app.Activity;
 import android.content.Intent;
 import android.os.Build;
 import android.support.test.filters.SmallTest;
 
+import org.junit.After;
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Rule;
@@ -19,6 +21,7 @@ import org.junit.runner.RunWith;
 
 import org.chromium.base.ActivityState;
 import org.chromium.base.ApplicationStatus;
+import org.chromium.base.test.util.CallbackHelper;
 import org.chromium.base.test.util.CommandLineFlags;
 import org.chromium.base.test.util.Feature;
 import org.chromium.base.test.util.MinAndroidSdkLevel;
@@ -33,6 +36,7 @@ import org.chromium.content_public.browser.test.util.Criteria;
 import org.chromium.content_public.browser.test.util.CriteriaHelper;
 
 import java.util.concurrent.Callable;
+import java.util.concurrent.TimeoutException;
 
 /**
  * Class for testing MultiWindowUtils.
@@ -49,6 +53,11 @@ public class MultiWindowUtilsTest {
         mActivityTestRule.startMainActivityOnBlankPage();
     }
 
+    @After
+    public void teardown() {
+        MultiWindowUtils.getInstance().setIsInMultiWindowModeForTesting(false);
+    }
+
     /**
      * Tests that ChromeTabbedActivity2 is used for intents when EXTRA_WINDOW_ID is set to 2.
      */
@@ -177,4 +186,134 @@ public class MultiWindowUtilsTest {
                 mActivityTestRule.getActivity().getIntent(), mActivityTestRule.getActivity());
         Assert.assertFalse(MultiWindowUtils.getInstance().getTabbedActivity2TaskRunning());
     }
+
+    /**
+     * Tests that {@link MultiWindowUtils#areMultipleChromeInstancesRunning} behaves correctly in
+     * the case the second instance is killed first.
+     */
+    @Test
+    @SmallTest
+    @Feature("MultiWindow")
+    public void testAreMultipleChromeInstancesRunningSecondInstanceKilledFirst()
+            throws TimeoutException {
+        ChromeTabbedActivity activity1 = mActivityTestRule.getActivity();
+        MultiWindowUtils.getInstance().setIsInMultiWindowModeForTesting(true);
+        Assert.assertFalse("Only a single instance should be running at the start.",
+                MultiWindowUtils.getInstance().areMultipleChromeInstancesRunning(activity1));
+
+        CallbackHelper activity1StoppedCallback = new CallbackHelper();
+        CallbackHelper activity1DestroyedCallback = new CallbackHelper();
+        ApplicationStatus.ActivityStateListener activity1StateListener =
+                new ApplicationStatus.ActivityStateListener() {
+                    @Override
+                    public void onActivityStateChange(Activity activity, int newState) {
+                        switch (newState) {
+                            case ActivityState.STOPPED:
+                                activity1StoppedCallback.notifyCalled();
+                                break;
+                            case ActivityState.DESTROYED:
+                                activity1DestroyedCallback.notifyCalled();
+                                break;
+                        }
+                    }
+                };
+        ApplicationStatus.registerStateListenerForActivity(activity1StateListener, activity1);
+
+        int callCount = activity1StoppedCallback.getCallCount();
+        ChromeTabbedActivity activity2 = createSecondChromeTabbedActivity(activity1);
+        activity1StoppedCallback.waitForCallback(callCount);
+        Assert.assertTrue("Both instances should be running now that the second has started.",
+                MultiWindowUtils.getInstance().areMultipleChromeInstancesRunning(activity1));
+
+        CallbackHelper activity2DestroyedCallback = new CallbackHelper();
+        ApplicationStatus.ActivityStateListener activity2StateListener =
+                new ApplicationStatus.ActivityStateListener() {
+                    @Override
+                    public void onActivityStateChange(Activity activity, int newState) {
+                        switch (newState) {
+                            case ActivityState.DESTROYED:
+                                activity2DestroyedCallback.notifyCalled();
+                                break;
+                        }
+                    }
+                };
+        ApplicationStatus.registerStateListenerForActivity(activity2StateListener, activity2);
+
+        callCount = activity2DestroyedCallback.getCallCount();
+        activity2.finishAndRemoveTask();
+        activity2DestroyedCallback.waitForCallback(callCount);
+        Assert.assertFalse("Only a single instance should be running after the second is killed.",
+                MultiWindowUtils.getInstance().areMultipleChromeInstancesRunning(activity1));
+
+        callCount = activity1DestroyedCallback.getCallCount();
+        activity1.finishAndRemoveTask();
+        activity1DestroyedCallback.waitForCallback(callCount);
+        Assert.assertFalse("No instances should be running as all instances are killed.",
+                MultiWindowUtils.getInstance().areMultipleChromeInstancesRunning(activity1));
+    }
+
+    /**
+     * Tests that {@link MultiWindowUtils#areMultipleChromeInstancesRunning} behaves correctly in
+     * the case the first instance is killed first.
+     */
+    @Test
+    @SmallTest
+    @Feature("MultiWindow")
+    public void testAreMultipleChromeInstancesRunningFirstInstanceKilledFirst()
+            throws TimeoutException {
+        ChromeTabbedActivity activity1 = mActivityTestRule.getActivity();
+        MultiWindowUtils.getInstance().setIsInMultiWindowModeForTesting(true);
+        Assert.assertFalse("Only a single instance should be running at the start.",
+                MultiWindowUtils.getInstance().areMultipleChromeInstancesRunning(activity1));
+
+        CallbackHelper activity1StoppedCallback = new CallbackHelper();
+        CallbackHelper activity1DestroyedCallback = new CallbackHelper();
+        ApplicationStatus.ActivityStateListener activity1StateListener =
+                new ApplicationStatus.ActivityStateListener() {
+                    @Override
+                    public void onActivityStateChange(Activity activity, int newState) {
+                        switch (newState) {
+                            case ActivityState.STOPPED:
+                                activity1StoppedCallback.notifyCalled();
+                                break;
+                            case ActivityState.DESTROYED:
+                                activity1DestroyedCallback.notifyCalled();
+                                break;
+                        }
+                    }
+                };
+        ApplicationStatus.registerStateListenerForActivity(activity1StateListener, activity1);
+
+        int callCount = activity1StoppedCallback.getCallCount();
+        ChromeTabbedActivity activity2 = createSecondChromeTabbedActivity(activity1);
+        activity1StoppedCallback.waitForCallback(callCount);
+        Assert.assertTrue("Both instances should be running now that the second has started.",
+                MultiWindowUtils.getInstance().areMultipleChromeInstancesRunning(activity1));
+
+        CallbackHelper activity2DestroyedCallback = new CallbackHelper();
+        ApplicationStatus.ActivityStateListener activity2StateListener =
+                new ApplicationStatus.ActivityStateListener() {
+                    @Override
+                    public void onActivityStateChange(Activity activity, int newState) {
+                        switch (newState) {
+                            case ActivityState.DESTROYED:
+                                activity2DestroyedCallback.notifyCalled();
+                                break;
+                        }
+                    }
+                };
+        ApplicationStatus.registerStateListenerForActivity(activity2StateListener, activity2);
+
+        callCount = activity1DestroyedCallback.getCallCount();
+        activity1.finishAndRemoveTask();
+        activity1DestroyedCallback.waitForCallback(callCount);
+        Assert.assertFalse("Only a single instance should be running after the first is killed.",
+                MultiWindowUtils.getInstance().areMultipleChromeInstancesRunning(activity1));
+
+        callCount = activity2DestroyedCallback.getCallCount();
+        activity2.finishAndRemoveTask();
+        activity2DestroyedCallback.waitForCallback(callCount);
+        Assert.assertFalse("No instances should be running as all instances are killed.",
+                MultiWindowUtils.getInstance().areMultipleChromeInstancesRunning(activity1));
+    }
 }