🔍 More SearchActivity tests

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

Index: chrome/android/javatests/src/org/chromium/chrome/browser/locale/DefaultSearchEngineDialogHelperUtils.java
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/locale/DefaultSearchEngineDialogHelperUtils.java b/chrome/android/javatests/src/org/chromium/chrome/browser/locale/DefaultSearchEngineDialogHelperUtils.java
new file mode 100644
index 0000000000000000000000000000000000000000..a7d1ba723f41d5c8b2055e82fd908896e89b76c6
--- /dev/null
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/locale/DefaultSearchEngineDialogHelperUtils.java
@@ -0,0 +1,78 @@
+// 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.view.View;
+import android.view.ViewGroup;
+
+import org.junit.Assert;
+
+import org.chromium.base.ThreadUtils;
+import org.chromium.chrome.R;
+import org.chromium.chrome.browser.search_engines.TemplateUrlService;
+import org.chromium.content.browser.test.util.Criteria;
+import org.chromium.content.browser.test.util.CriteriaHelper;
+
+/**
+ * Utilities for interacting with a {@link DefaultSearchEngineDialogHelper}.
+ */
+public class DefaultSearchEngineDialogHelperUtils {
+    private static final int OPTION_LAYOUT_ID = R.id.default_search_engine_dialog_options;
+    private static final int OK_BUTTON_ID = R.id.button_primary;
+
+    private static String sSelectedEngine;
+
+    /** Clicks on the first search engine option available. */
+    public static void clickOnFirstEngine(final View rootView) {
+        // Wait for the options to appear.
+        CriteriaHelper.pollUiThread(new Criteria() {
+            @Override
+            public boolean isSatisfied() {
+                ViewGroup options = (ViewGroup) rootView.findViewById(OPTION_LAYOUT_ID);
+                return options.getChildCount() > 0;
+            }
+        });
+
+        // Click on the first search engine option available.
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                ViewGroup options = (ViewGroup) rootView.findViewById(OPTION_LAYOUT_ID);
+                options.getChildAt(0).performClick();
+                sSelectedEngine = (String) (options.getChildAt(0).getTag());
+            }
+        });
+
+        // Wait for the OK button to be clicakble.
+        CriteriaHelper.pollUiThread(new Criteria() {
+            @Override
+            public boolean isSatisfied() {
+                View view = rootView.findViewById(OK_BUTTON_ID);
+                return view != null && view.isEnabled();
+            }
+        });
+
+        // Click on the OK button.
+        ThreadUtils.runOnUiThread(new Runnable() {
+            @Override
+            public void run() {
+                View view = rootView.findViewById(OK_BUTTON_ID);
+                view.performClick();
+            }
+        });
+
+        // Confirm the engine was set appropriately.
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                Assert.assertEquals("Search engine wasn't set",
+                        TemplateUrlService.getInstance()
+                                .getDefaultSearchEngineTemplateUrl()
+                                .getKeyword(),
+                        sSelectedEngine);
+            }
+        });
+    }
+}