[Android] Allow setting recently visited search engines as default search engine

chrome/android/javatests/src/org/chromium/chrome/browser/preferences/PreferencesTest.java

 // Copyright 2014 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.preferences;
 
 import android.app.Activity;
 import android.app.Instrumentation;
 import android.content.Context;
 import android.content.Intent;
@@ skipping 66 matching lines @@
     /**
      * Change search engine and make sure it works correctly.
      */
     @SmallTest
     @Feature({"Preferences"})
     @DisableIf.Build(hardware_is = "sprout", message = "crashes on android-one: crbug.com/540720")
     @RetryOnFailure
     public void testSearchEnginePreference() throws Exception {
         ensureTemplateUrlServiceLoaded();
 
+        final Preferences prefActivity =
+                startPreferences(getInstrumentation(), SearchEnginePreference.class.getName());
+
         // Set the second search engine as the default using TemplateUrlService.
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
-                TemplateUrlService.getInstance().setSearchEngine(1);
+                SearchEnginePreference pref =
+                        (SearchEnginePreference) prefActivity.getFragmentForTest();
+                pref.setValueForTesting("1");
             }
         });
 
-        final Preferences prefActivity =
-                startPreferences(getInstrumentation(), SearchEnginePreference.class.getName());
 
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
                 // Ensure that the second search engine in the list is selected.
                 SearchEnginePreference pref =
                         (SearchEnginePreference) prefActivity.getFragmentForTest();
                 assertNotNull(pref);
                 assertEquals("1", pref.getValueForTesting());
 
                 // Simulate selecting the third search engine, ensure that TemplateUrlService is
                 // updated, but location permission not granted for the new engine.
-                pref.setValueForTesting("2");
+                String keyword2 = pref.setValueForTesting("2");
                 TemplateUrlService templateUrlService = TemplateUrlService.getInstance();
-                assertEquals(2, templateUrlService.getDefaultSearchEngineIndex());
-                assertEquals(ContentSetting.ASK, locationPermissionForSearchEngine(2));
+                assertEquals(keyword2,
+                        templateUrlService.getDefaultSearchEngineTemplateUrl().getKeyword());
+                assertEquals(ContentSetting.ASK, locationPermissionForSearchEngine(keyword2));
 
                 // Simulate selecting the fourth search engine and but set a blocked permission
                 // first and ensure that location permission is NOT granted.
-                String url = templateUrlService.getSearchEngineUrlFromTemplateUrl(3);
+                String keyword3 = pref.getKeywordFromIndexForTesting(3);
+                String url = templateUrlService.getSearchEngineUrlFromTemplateUrl(keyword3);
                 WebsitePreferenceBridge.nativeSetGeolocationSettingForOrigin(
                         url, url, ContentSetting.BLOCK.toInt(), false);
-                pref.setValueForTesting("3");
-                assertEquals(3, TemplateUrlService.getInstance().getDefaultSearchEngineIndex());
-                assertEquals(ContentSetting.BLOCK, locationPermissionForSearchEngine(3));
-                assertEquals(ContentSetting.ASK, locationPermissionForSearchEngine(2));
+                keyword3 = pref.setValueForTesting("3");
+                assertEquals(keyword3, TemplateUrlService.getInstance()
+                                               .getDefaultSearchEngineTemplateUrl()
+                                               .getKeyword());
+                assertEquals(ContentSetting.BLOCK, locationPermissionForSearchEngine(keyword3));
+                assertEquals(ContentSetting.ASK, locationPermissionForSearchEngine(keyword2));
 
                 // Make sure a pre-existing ALLOW value does not get deleted when switching away
                 // from a search engine.
-                url = templateUrlService.getSearchEngineUrlFromTemplateUrl(4);
+                String keyword4 = pref.getKeywordFromIndexForTesting(4);
+                url = templateUrlService.getSearchEngineUrlFromTemplateUrl(keyword4);
                 WebsitePreferenceBridge.nativeSetGeolocationSettingForOrigin(
                         url, url, ContentSetting.ALLOW.toInt(), false);
-                pref.setValueForTesting("4");
-                assertEquals(4, TemplateUrlService.getInstance().getDefaultSearchEngineIndex());
-                assertEquals(ContentSetting.ALLOW, locationPermissionForSearchEngine(4));
+                keyword4 = pref.setValueForTesting("4");
+                assertEquals(keyword4, TemplateUrlService.getInstance()
+                                               .getDefaultSearchEngineTemplateUrl()
+                                               .getKeyword());
+                assertEquals(ContentSetting.ALLOW, locationPermissionForSearchEngine(keyword4));
                 pref.setValueForTesting("3");
-                assertEquals(ContentSetting.ALLOW, locationPermissionForSearchEngine(4));
+                assertEquals(ContentSetting.ALLOW, locationPermissionForSearchEngine(keyword4));
             }
         });
     }
 
     /**
      * Make sure that when a user switches to a search engine that uses HTTP, the location
      * permission is not added.
      */
     /*
      * @SmallTest
      * @Feature({"Preferences"})
      * BUG=crbug.com/540706
      */
     @FlakyTest
     @DisableIf.Build(hardware_is = "sprout", message = "fails on android-one: crbug.com/540706")
     public void testSearchEnginePreferenceHttp() throws Exception {
         ensureTemplateUrlServiceLoaded();
 
+        final Preferences prefActivity =
+                startPreferences(getInstrumentation(), SearchEnginePreference.class.getName());
+
         // Set the first search engine as the default using TemplateUrlService.
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
-                TemplateUrlService.getInstance().setSearchEngine(0);
+                SearchEnginePreference pref =
+                        (SearchEnginePreference) prefActivity.getFragmentForTest();
+                pref.setValueForTesting("0");
             }
         });
 
-        final Preferences prefActivity =
-                startPreferences(getInstrumentation(), SearchEnginePreference.class.getName());
-
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
                 // Ensure that the first search engine in the list is selected.
                 SearchEnginePreference pref =
                         (SearchEnginePreference) prefActivity.getFragmentForTest();
                 assertNotNull(pref);
                 assertEquals("0", pref.getValueForTesting());
 
                 // Simulate selecting a search engine that uses HTTP.
-                int index = indexOfFirstHttpSearchEngine();
-                pref.setValueForTesting(Integer.toString(index));
+                int index = indexOfFirstHttpSearchEngine(pref);
+                String keyword = pref.setValueForTesting(Integer.toString(index));
 
                 TemplateUrlService templateUrlService = TemplateUrlService.getInstance();
-                assertEquals(index, templateUrlService.getDefaultSearchEngineIndex());
-                assertEquals(ContentSetting.ASK, locationPermissionForSearchEngine(index));
+                assertEquals(keyword,
+                        templateUrlService.getDefaultSearchEngineTemplateUrl().getKeyword());
+                assertEquals(ContentSetting.ASK, locationPermissionForSearchEngine(keyword));
             }
         });
     }
 
-    private int indexOfFirstHttpSearchEngine() {
+    private int indexOfFirstHttpSearchEngine(SearchEnginePreference pref) {
         TemplateUrlService templateUrlService = TemplateUrlService.getInstance();
-        List<TemplateUrl> urls = templateUrlService.getLocalizedSearchEngines();
+        List<TemplateUrl> urls = templateUrlService.getSearchEngines();
         int index;
         for (index = 0; index < urls.size(); ++index) {
-            String url = templateUrlService.getSearchEngineUrlFromTemplateUrl(index);
+            String keyword = pref.getKeywordFromIndexForTesting(index);
+            String url = templateUrlService.getSearchEngineUrlFromTemplateUrl(keyword);
             if (url.startsWith("http:")) {
                 return index;
             }
         }
         fail();
         return index;
     }
 
     private void ensureTemplateUrlServiceLoaded() throws Exception {
         // Make sure the template_url_service is loaded.
@@ skipping 10 matching lines @@
                             onTemplateUrlServiceLoadedHelper.notifyCalled();
                         }
                     });
                     TemplateUrlService.getInstance().load();
                 }
             }
         });
         onTemplateUrlServiceLoadedHelper.waitForCallback(0);
     }
 
-    private ContentSetting locationPermissionForSearchEngine(int index) {
-        String url = TemplateUrlService.getInstance().getSearchEngineUrlFromTemplateUrl(index);
+    private ContentSetting locationPermissionForSearchEngine(String keyword) {
+        String url = TemplateUrlService.getInstance().getSearchEngineUrlFromTemplateUrl(keyword);
         GeolocationInfo locationSettings = new GeolocationInfo(url, null, false);
         ContentSetting locationPermission = locationSettings.getContentSetting();
         return locationPermission;
     }
 
     /**
      * Tests setting FontScaleFactor and ForceEnableZoom in AccessibilityPreferences and ensures
      * that ForceEnableZoom changes corresponding to FontScaleFactor.
      */
     @SmallTest
@@ skipping 69 matching lines @@
     private static void userSetForceEnableZoom(final AccessibilityPreferences accessibilityPref,
             final SeekBarLinkedCheckBoxPreference forceEnableZoomPref, final boolean enabled) {
         ThreadUtils.runOnUiThread(new Runnable() {
             @Override
             public void run() {
                 accessibilityPref.onPreferenceChange(forceEnableZoomPref, enabled);
             }
         });
     }
 }