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

chrome/android/javatests/src/org/chromium/chrome/browser/search_engines/TemplateUrlServiceTest.java

Index: chrome/android/javatests/src/org/chromium/chrome/browser/search_engines/TemplateUrlServiceTest.java
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/search_engines/TemplateUrlServiceTest.java b/chrome/android/javatests/src/org/chromium/chrome/browser/search_engines/TemplateUrlServiceTest.java
index 054ddad6130906471473b6478fac711f24b6c575..74b043617720b40b9011b610db0caae4d76aff49 100644
--- a/chrome/android/javatests/src/org/chromium/chrome/browser/search_engines/TemplateUrlServiceTest.java
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/search_engines/TemplateUrlServiceTest.java
@@ -119,15 +119,16 @@ public class TemplateUrlServiceTest extends NativeLibraryTestBase {
     public void testSetAndGetSearchEngine() throws InterruptedException {
         final TemplateUrlService templateUrlService = waitForTemplateUrlServiceToLoad();
 
+        List<TemplateUrl> searchEngines = templateUrlService.getLocalizedSearchEngines();
         // Ensure known state of default search index before running test.
-        int searchEngineIndex = ThreadUtils.runOnUiThreadBlockingNoException(
-                new Callable<Integer>() {
+        String searchEngineKeyword =
+                ThreadUtils.runOnUiThreadBlockingNoException(new Callable<String>() {
                     @Override
-                    public Integer call() throws Exception {
-                        return templateUrlService.getDefaultSearchEngineIndex();
+                    public String call() throws Exception {
+                        return templateUrlService.getDefaultSearchEngineTemplateUrl().getKeyword();
                     }
                 });
-        assertEquals(0, searchEngineIndex);
+        assertEquals(searchEngines.get(0), searchEngineKeyword);
 
         // Set search engine index and verified it stuck.
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
@@ -137,17 +138,16 @@ public class TemplateUrlServiceTest extends NativeLibraryTestBase {
                         templateUrlService.getLocalizedSearchEngines();
                 assertTrue("There must be more than one search engine to change searchEngines",
                         searchEngines.size() > 1);
-                templateUrlService.setSearchEngine(1);
+                templateUrlService.setSearchEngine(searchEngines.get(1).getKeyword());
             }
         });
-        searchEngineIndex = ThreadUtils.runOnUiThreadBlockingNoException(
-                new Callable<Integer>() {
-                    @Override
-                    public Integer call() throws Exception {
-                        return templateUrlService.getDefaultSearchEngineIndex();
-                    }
-                });
-        assertEquals(1, searchEngineIndex);
+        searchEngineKeyword = ThreadUtils.runOnUiThreadBlockingNoException(new Callable<String>() {
+            @Override
+            public String call() throws Exception {
+                return templateUrlService.getDefaultSearchEngineTemplateUrl().getKeyword();
+            }
+        });
+        assertEquals(searchEngines.get(1), searchEngineKeyword);
     }
 
     private TemplateUrlService waitForTemplateUrlServiceToLoad() throws InterruptedException {