Determine whether to show "Add to Homescreen" dialog or WebAPK infobar

chrome/android/javatests/src/org/chromium/chrome/browser/webapps/AddToHomescreenProcessTest.java

Index: chrome/android/javatests/src/org/chromium/chrome/browser/webapps/AddToHomescreenProcessTest.java
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/webapps/AddToHomescreenDialogHelperTest.java b/chrome/android/javatests/src/org/chromium/chrome/browser/webapps/AddToHomescreenProcessTest.java
similarity index 87%
copy from chrome/android/javatests/src/org/chromium/chrome/browser/webapps/AddToHomescreenDialogHelperTest.java
copy to chrome/android/javatests/src/org/chromium/chrome/browser/webapps/AddToHomescreenProcessTest.java
index 6a3538845f67cd8da28537e28af30316de6e85b3..570edc613448ea17237901642a850ae3275f6d37 100644
--- a/chrome/android/javatests/src/org/chromium/chrome/browser/webapps/AddToHomescreenDialogHelperTest.java
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/webapps/AddToHomescreenProcessTest.java
@@ -4,6 +4,7 @@
 
 package org.chromium.chrome.browser.webapps;
 
+import android.app.Activity;
 import android.content.Context;
 import android.content.Intent;
 import android.graphics.Bitmap;
@@ -30,9 +31,9 @@ import org.chromium.net.test.EmbeddedTestServer;
 import java.util.concurrent.Callable;
 
 /**
- * Tests org.chromium.chrome.browser.webapps.AddToHomescreenDialogHelper and its C++ counterpart.
+ * Tests org.chromium.chrome.browser.webapps.AddToHomescreenProcess and its C++ counterpart.
  */
-public class AddToHomescreenDialogHelperTest extends ChromeActivityTestCaseBase<ChromeActivity> {
+public class AddToHomescreenProcessTest extends ChromeActivityTestCaseBase<ChromeActivity> {
     private static final String WEBAPP_ACTION_NAME = "WEBAPP_ACTION";
 
     private static final String WEBAPP_TITLE = "Webapp shortcut";
@@ -108,11 +109,43 @@ public class AddToHomescreenDialogHelperTest extends ChromeActivityTestCaseBase<
         }
     }
 
+    /**
+     * Test AddToHomescreenProcess subclass which mocks showing the add-to-homescreen dialog and
+     * adds the shortcut to the home screen once it is ready.
+     */
+    private static class TestAddToHomescreenProcess extends AddToHomescreenProcess {
+        private String mTitle;
+
+        /**
+         * Creates an instance of {@link TestAddToHomescreenProcess}.
+         * @param title The title that the user entered into the add-to-homescreen dialog.
+         */
+        public TestAddToHomescreenProcess(Activity activity, Tab tab, String title) {
+            super(activity, tab);
+            mTitle = title;
+        }
+
+        @Override
+        public void showDialog() {
+            AddToHomescreenProcess.Ui ui = new AddToHomescreenProcess.Ui() {
+                @Override
+                public void onUserTitleAvailable(String title) {}
+
+                @Override
+                public void onReadyToAdd(Bitmap icon) {
+                    TestAddToHomescreenProcess.this.addShortcut(mTitle);
+                }
+            };
+
+            setUi(ui);
+        }
+    }
+
     private ChromeActivity mActivity;
     private Tab mTab;
     private TestShortcutHelperDelegate mShortcutHelperDelegate;
 
-    public AddToHomescreenDialogHelperTest() {
+    public AddToHomescreenProcessTest() {
         super(ChromeActivity.class);
     }
 
@@ -249,27 +282,17 @@ public class AddToHomescreenDialogHelperTest extends ChromeActivityTestCaseBase<
 
     private void addShortcutToTab(final Tab tab, final String title) throws Exception {
         // Add the shortcut.
-        Callable<AddToHomescreenDialogHelper> callable =
-                new Callable<AddToHomescreenDialogHelper>() {
+        Callable<AddToHomescreenProcess> callable =
+                new Callable<AddToHomescreenProcess>() {
             @Override
-            public AddToHomescreenDialogHelper call() {
-                final AddToHomescreenDialogHelper helper =
-                        new AddToHomescreenDialogHelper(mActivity.getApplicationContext(), tab);
-                // Calling initialize() isn't strictly required but it is testing this code path.
-                helper.initialize(new AddToHomescreenDialogHelper.Observer() {
-                    @Override
-                    public void onUserTitleAvailable(String t) {
-                    }
-
-                    @Override
-                    public void onIconAvailable(Bitmap icon) {
-                        helper.addShortcut(title);
-                    }
-                });
-                return helper;
+            public AddToHomescreenProcess call() {
+                AddToHomescreenProcess process =
+                        new TestAddToHomescreenProcess(mActivity, tab, title);
+                process.start();
+                return process;
             }
         };
-        final AddToHomescreenDialogHelper helper =
+        final AddToHomescreenProcess process =
                 ThreadUtils.runOnUiThreadBlockingNoException(callable);
 
         // Make sure that the shortcut was added.
@@ -283,7 +306,7 @@ public class AddToHomescreenDialogHelperTest extends ChromeActivityTestCaseBase<
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
-                helper.destroy();
+                process.destroy();
             }
         });
     }