Upstream oodles of Chrome for Android code into Chromium.

chrome/test/android/javatests_staging/src/org/chromium/chrome/test/util/OmniboxTestUtils.java

Index: chrome/test/android/javatests_staging/src/org/chromium/chrome/test/util/OmniboxTestUtils.java
diff --git a/chrome/test/android/javatests_staging/src/org/chromium/chrome/test/util/OmniboxTestUtils.java b/chrome/test/android/javatests_staging/src/org/chromium/chrome/test/util/OmniboxTestUtils.java
new file mode 100644
index 0000000000000000000000000000000000000000..90a84546215b44220efe824d867db8694e364c57
--- /dev/null
+++ b/chrome/test/android/javatests_staging/src/org/chromium/chrome/test/util/OmniboxTestUtils.java
@@ -0,0 +1,327 @@
+// Copyright 2015 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.test.util;
+
+import android.content.Context;
+import android.view.View;
+import android.view.inputmethod.InputMethodManager;
+
+import junit.framework.Assert;
+
+import org.chromium.base.ThreadUtils;
+import org.chromium.chrome.browser.omnibox.AutocompleteController;
+import org.chromium.chrome.browser.omnibox.AutocompleteController.OnSuggestionsReceivedListener;
+import org.chromium.chrome.browser.omnibox.LocationBarLayout;
+import org.chromium.chrome.browser.omnibox.OmniboxSuggestion;
+import org.chromium.chrome.browser.omnibox.UrlBar;
+import org.chromium.chrome.browser.profiles.Profile;
+import org.chromium.content.browser.test.util.Criteria;
+import org.chromium.content.browser.test.util.CriteriaHelper;
+import org.chromium.content.browser.test.util.TouchCommon;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.concurrent.Callable;
+import java.util.concurrent.ExecutionException;
+
+/**
+ * Utility methods and classes for testing the Omnibox.
+ */
+public class OmniboxTestUtils {
+    private OmniboxTestUtils() {}
+
+    /**
+     * Builder for the data structure that describes a set of omnibox results for a given
+     * query.
+     */
+    public static class TestSuggestionResultsBuilder {
+        private final List<SuggestionsResultBuilder> mSuggestionBuilders =
+                new ArrayList<SuggestionsResultBuilder>();
+        private String mTextShownFor;
+
+        public TestSuggestionResultsBuilder addSuggestions(SuggestionsResultBuilder suggestions) {
+            mSuggestionBuilders.add(suggestions);
+            return this;
+        }
+
+        public TestSuggestionResultsBuilder setTextShownFor(String text) {
+            mTextShownFor = text;
+            return this;
+        }
+
+        private List<SuggestionsResult> buildSuggestionsList() {
+            ArrayList<SuggestionsResult> suggestions = new ArrayList<SuggestionsResult>();
+            for (int i = 0; i < mSuggestionBuilders.size(); i++) {
+                suggestions.add(mSuggestionBuilders.get(i).build());
+            }
+            return suggestions;
+        }
+    }
+
+    /**
+     * Builder for {@link SuggestionsResult}.
+     */
+    public static class SuggestionsResultBuilder {
+        private final List<OmniboxSuggestion> mSuggestions = new ArrayList<OmniboxSuggestion>();
+        private String mAutocompleteText;
+
+        public SuggestionsResultBuilder addGeneratedSuggestion(
+                OmniboxSuggestion.Type type, String text, String url) {
+            mSuggestions.add(new OmniboxSuggestion(
+                    type.nativeType(), 0, 0, text, null, null, null, "", url, url, false, false));
+            return this;
+        }
+
+        public SuggestionsResultBuilder addSuggestion(OmniboxSuggestion suggestion) {
+            mSuggestions.add(suggestion);
+            return this;
+        }
+
+        public SuggestionsResultBuilder setAutocompleteText(String autocompleteText) {
+            mAutocompleteText = autocompleteText;
+            return this;
+        }
+
+        private SuggestionsResult build() {
+            return new SuggestionsResult(mSuggestions, mAutocompleteText);
+        }
+    }
+
+    /**
+     * Data structure that contains the test data to be sent to
+     * {@link OnSuggestionsReceivedListener#onSuggestionsReceived}.
+     */
+    public static class SuggestionsResult {
+        private final List<OmniboxSuggestion> mSuggestions;
+        private final String mAutocompleteText;
+
+        public SuggestionsResult(List<OmniboxSuggestion> suggestions, String autocompleteText) {
+            mSuggestions = suggestions;
+            mAutocompleteText = autocompleteText;
+        }
+    }
+
+    /**
+     * Builds the necessary suggestion input for a TestAutocompleteController.
+     */
+    public static Map<String, List<SuggestionsResult>> buildSuggestionMap(
+            TestSuggestionResultsBuilder... builders) {
+        Map<String, List<SuggestionsResult>> suggestionMap =
+                new HashMap<String, List<SuggestionsResult>>();
+        for (TestSuggestionResultsBuilder builder : builders) {
+            suggestionMap.put(builder.mTextShownFor, builder.buildSuggestionsList());
+        }
+        return suggestionMap;
+    }
+
+    /**
+     * AutocompleteController instance that allows for easy testing.
+     */
+    public static class TestAutocompleteController extends AutocompleteController {
+        private final View mView;
+        private final Map<String, List<SuggestionsResult>> mSuggestions;
+        private Runnable mSuggestionsDispatcher;
+        private int mZeroSuggestCalledCount;
+
+        public TestAutocompleteController(
+                View view,
+                OnSuggestionsReceivedListener listener,
+                Map<String, List<SuggestionsResult>> suggestions) {
+            super(listener);
+            mView = view;
+            mSuggestions = suggestions;
+        }
+
+        @Override
+        public void start(
+                Profile profile, String url ,
+                final String text, boolean preventInlineAutocomplete) {
+            mSuggestionsDispatcher = new Runnable() {
+                @Override
+                public void run() {
+                    List<SuggestionsResult> suggestions = mSuggestions.get(text.toLowerCase());
+                    if (suggestions == null) return;
+
+                    for (int i = 0; i < suggestions.size(); i++) {
+                        onSuggestionsReceived(
+                                suggestions.get(i).mSuggestions,
+                                suggestions.get(i).mAutocompleteText,
+                                0);
+                    }
+                }
+            };
+            mView.post(mSuggestionsDispatcher);
+        }
+
+        @Override
+        public void startZeroSuggest(Profile profile, String omniboxText, String url,
+                boolean isQueryInOmnibox, boolean focusedFromFakebox) {
+            mZeroSuggestCalledCount++;
+        }
+
+        public int numZeroSuggestRequests() {
+            return mZeroSuggestCalledCount;
+        }
+
+        @Override
+        public void stop(boolean clear) {
+            if (mSuggestionsDispatcher != null) mView.removeCallbacks(mSuggestionsDispatcher);
+            mSuggestionsDispatcher = null;
+        }
+
+        @Override
+        protected long nativeInit(Profile profile) {
+            return 1;
+        }
+
+        @Override
+        public void setProfile(Profile profile) {}
+    }
+
+    /**
+     * AutocompleteController instance that will trigger no suggestions.
+     */
+    public static class StubAutocompleteController extends AutocompleteController {
+        public StubAutocompleteController() {
+            super(new OnSuggestionsReceivedListener() {
+                @Override
+                public void onSuggestionsReceived(List<OmniboxSuggestion> suggestions,
+                        String inlineAutocompleteText) {
+                    Assert.fail("No autocomplete suggestions should be received");
+                }
+            });
+        }
+
+        @Override
+        public void start(
+                Profile profile, String url ,
+                final String text, boolean preventInlineAutocomplete) {}
+
+        @Override
+        public void startZeroSuggest(Profile profile, String omniboxText, String url,
+                boolean isQueryInOmnibox, boolean focusedFromFakebox) {
+        }
+
+        @Override
+        public void stop(boolean clear) {}
+
+        @Override
+        protected long nativeInit(Profile profile) {
+            return 1;
+        }
+
+        @Override
+        public void setProfile(Profile profile) {}
+    }
+
+    /**
+     * Checks and verifies that the URL bar can request and release focus X times without issue.
+     * @param urlBar The view to focus.
+     * @param times The number of times focus should be requested and released.
+     * @throws InterruptedException
+     */
+    public static void checkUrlBarRefocus(UrlBar urlBar, int times)
+            throws InterruptedException {
+        for (int i = 0; i < times; i++) {
+            toggleUrlBarFocus(urlBar, true);
+            Assert.assertTrue(waitForFocusAndKeyboardActive(urlBar, true));
+            toggleUrlBarFocus(urlBar, false);
+            Assert.assertTrue(waitForFocusAndKeyboardActive(urlBar, false));
+        }
+    }
+
+    /**
+     * Determines whether the UrlBar currently has focus.
+     * @param urlBar The view to check focus on.
+     * @return Whether the UrlBar has focus.
+     */
+    public static boolean doesUrlBarHaveFocus(final UrlBar urlBar) {
+        return ThreadUtils.runOnUiThreadBlockingNoException(new Callable<Boolean>() {
+            @Override
+            public Boolean call() throws Exception {
+                return urlBar.hasFocus();
+            }
+        });
+    }
+
+    private static boolean isKeyboardActiveForView(final View view) {
+        return ThreadUtils.runOnUiThreadBlockingNoException(new Callable<Boolean>() {
+            @Override
+            public Boolean call() throws Exception {
+                InputMethodManager imm =
+                        (InputMethodManager) view.getContext().getSystemService(
+                                Context.INPUT_METHOD_SERVICE);
+                return imm.isActive(view);
+            }
+        });
+    }
+
+    /**
+     * Toggles the focus state for the passed in UrlBar.
+     * @param urlBar The UrlBar whose focus is being changed.
+     * @param gainFocus Whether focus should be requested or cleared.
+     */
+    public static void toggleUrlBarFocus(final UrlBar urlBar, boolean gainFocus) {
+        if (gainFocus) {
+            TouchCommon.singleClickView(urlBar);
+        } else {
+            ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+                @Override
+                public void run() {
+                    urlBar.clearFocus();
+                }
+            });
+        }
+    }
+
+    /**
+     * Waits for the UrlBar to have the expected focus state.
+     *
+     * @param urlBar The UrlBar whose focus is being inspected.
+     * @param active Whether the UrlBar is expected to have focus or not.
+     * @return Whether the UrlBar had the requested focus state.
+     */
+    public static boolean waitForFocusAndKeyboardActive(final UrlBar urlBar, final boolean active)
+            throws InterruptedException {
+        return CriteriaHelper.pollForCriteria(new Criteria() {
+            @Override
+            public boolean isSatisfied() {
+                return (doesUrlBarHaveFocus(urlBar) == active)
+                        && (isKeyboardActiveForView(urlBar) == active);
+            }
+        });
+    }
+
+    /**
+     * Waits for a non-empty list of omnibox suggestions is shown.
+     *
+     * @param locationBar The LocationBar who owns the suggestions.
+     * @return Whether the suggestions were shown.
+     */
+    public static boolean waitForOmniboxSuggestions(final LocationBarLayout locationBar)
+            throws InterruptedException {
+        return CriteriaHelper.pollForCriteria(new Criteria() {
+            @Override
+            public boolean isSatisfied() {
+                try {
+                    return ThreadUtils.runOnUiThreadBlocking(new Callable<Boolean>() {
+                        @Override
+                        public Boolean call() throws Exception {
+                            LocationBarLayout.OmniboxSuggestionsList suggestionsList =
+                                    locationBar.getSuggestionList();
+                            return suggestionsList != null
+                                    && suggestionsList.isShown()
+                                    && suggestionsList.getCount() >= 1;
+                        }
+                    });
+                } catch (ExecutionException e) {
+                    return false;
+                }
+            }
+        });
+    }
+}