Ensure only one live search engine promo.

chrome/android/javatests/src/org/chromium/chrome/browser/locale/DefaultSearchEnginePromoDialogTest.java

Index: chrome/android/javatests/src/org/chromium/chrome/browser/locale/DefaultSearchEnginePromoDialogTest.java
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/locale/DefaultSearchEnginePromoDialogTest.java b/chrome/android/javatests/src/org/chromium/chrome/browser/locale/DefaultSearchEnginePromoDialogTest.java
new file mode 100644
index 0000000000000000000000000000000000000000..e5302c10018f0e5238738986b5949083dfb6ae2d
--- /dev/null
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/locale/DefaultSearchEnginePromoDialogTest.java
@@ -0,0 +1,123 @@
+// 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.locale;
+
+import android.app.Activity;
+import android.support.test.InstrumentationRegistry;
+import android.support.test.filters.LargeTest;
+
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+
+import org.chromium.base.ThreadUtils;
+import org.chromium.base.library_loader.ProcessInitException;
+import org.chromium.base.test.util.CallbackHelper;
+import org.chromium.base.test.util.CommandLineFlags;
+import org.chromium.chrome.browser.ChromeSwitches;
+import org.chromium.chrome.browser.ChromeTabbedActivity;
+import org.chromium.chrome.browser.init.ChromeBrowserInitializer;
+import org.chromium.chrome.browser.search_engines.TemplateUrlService;
+import org.chromium.chrome.browser.searchwidget.SearchActivity;
+import org.chromium.chrome.test.ChromeJUnit4ClassRunner;
+import org.chromium.chrome.test.util.ActivityUtils;
+import org.chromium.content.browser.test.util.Criteria;
+import org.chromium.content.browser.test.util.CriteriaHelper;
+
+import java.util.concurrent.Callable;
+import java.util.concurrent.ExecutionException;
+
+/**
+ * Integration tests for the {@link DefaultSearchEnginePromoDialog}.
+ */
+@RunWith(ChromeJUnit4ClassRunner.class)
+@CommandLineFlags.Add({ChromeSwitches.DISABLE_FIRST_RUN_EXPERIENCE})
+public class DefaultSearchEnginePromoDialogTest {
+    @Before
+    public void setUp() throws Exception {
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                try {
+                    ChromeBrowserInitializer.getInstance(InstrumentationRegistry.getTargetContext())
+                            .handleSynchronousStartup();
+                } catch (ProcessInitException e) {
+                    Assert.fail("Failed to initialize Chrome process");
+                }
+            }
+        });
+    }
+
+    @Test
+    @LargeTest
+    public void testOnlyOneLiveDialog() throws Exception {
+        final CallbackHelper templateUrlServiceInit = new CallbackHelper();
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                TemplateUrlService.getInstance().registerLoadListener(
+                        new TemplateUrlService.LoadListener() {
+                            @Override
+                            public void onTemplateUrlServiceLoaded() {
+                                TemplateUrlService.getInstance().unregisterLoadListener(this);
+                                templateUrlServiceInit.notifyCalled();
+                            }
+                        });
+            }
+        });
+        templateUrlServiceInit.waitForCallback(0);
+
+        final SearchActivity searchActivity = ActivityUtils.waitForActivity(
+                InstrumentationRegistry.getInstrumentation(), SearchActivity.class);
+        final DefaultSearchEnginePromoDialog searchDialog = showDialog(searchActivity);
+        Assert.assertEquals(searchDialog, DefaultSearchEnginePromoDialog.getCurrentDialog());
+
+        ChromeTabbedActivity tabbedActivity = ActivityUtils.waitForActivity(
+                InstrumentationRegistry.getInstrumentation(), ChromeTabbedActivity.class);
+        final DefaultSearchEnginePromoDialog tabbedDialog = showDialog(tabbedActivity);
+        Assert.assertEquals(tabbedDialog, DefaultSearchEnginePromoDialog.getCurrentDialog());
+
+        CriteriaHelper.pollUiThread(Criteria.equals(false, new Callable<Boolean>() {
+            @Override
+            public Boolean call() throws Exception {
+                return searchDialog.isShowing();
+            }
+        }));
+
+        CriteriaHelper.pollUiThread(Criteria.equals(true, new Callable<Boolean>() {
+            @Override
+            public Boolean call() throws Exception {
+                return searchActivity.isFinishing();
+            }
+        }));
+
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                tabbedDialog.dismiss();
+            }
+        });
+        CriteriaHelper.pollUiThread(new Criteria() {
+            @Override
+            public boolean isSatisfied() {
+                return DefaultSearchEnginePromoDialog.getCurrentDialog() == null;
+            }
+        });
+    }
+
+    private DefaultSearchEnginePromoDialog showDialog(final Activity activity)
+            throws ExecutionException {
+        return ThreadUtils.runOnUiThreadBlocking(new Callable<DefaultSearchEnginePromoDialog>() {
+            @Override
+            public DefaultSearchEnginePromoDialog call() throws Exception {
+                DefaultSearchEnginePromoDialog dialog = new DefaultSearchEnginePromoDialog(
+                        activity, LocaleManager.SEARCH_ENGINE_PROMO_SHOW_EXISTING, null);
+                dialog.show();
+                return dialog;
+            }
+        });
+    }
+}