[Android] Relanding Smart GO NEXT feature in Android Chrome 2/2

content/public/test/android/javatests/src/org/chromium/content/browser/test/util/TestInputMethodManagerWrapper.java

 // Copyright 2013 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.content.browser.test.util;
 
 import android.os.IBinder;
 import android.os.ResultReceiver;
 import android.util.Pair;
 import android.view.View;
@@ skipping 21 matching lines @@
     private InputConnection mInputConnection;
     private int mRestartInputCounter;
     private int mShowSoftInputCounter;
     private int mHideSoftInputCounter;
     private final Range mSelection = new Range(-1, -1);
     private final Range mComposition = new Range(-1, -1);
     private boolean mIsShowWithoutHideOutstanding;
     private final List<Pair<Range, Range>> mUpdateSelectionList;
     private int mUpdateCursorAnchorInfoCounter;
     private CursorAnchorInfo mLastCursorAnchorInfo;
+    private final ArrayList<EditorInfo> mEditorInfoList = new ArrayList<>();
 
     public TestInputMethodManagerWrapper(ContentViewCore contentViewCore) {
         super(null);
         Log.d(TAG, "TestInputMethodManagerWrapper constructor");
         mContentViewCore = contentViewCore;
         mUpdateSelectionList = new ArrayList<>();
     }
 
     @Override
     public void restartInput(View view) {
         mRestartInputCounter++;
         Log.d(TAG, "restartInput: count [%d]", mRestartInputCounter);
         EditorInfo editorInfo = new EditorInfo();
         mInputConnection = mContentViewCore.onCreateInputConnection(editorInfo);
+        mEditorInfoList.add(editorInfo);
     }
 
     @Override
     public void showSoftInput(View view, int flags, ResultReceiver resultReceiver) {
         mIsShowWithoutHideOutstanding = true;
         mShowSoftInputCounter++;
         Log.d(TAG, "showSoftInput: count [%d]", mShowSoftInputCounter);
         if (mInputConnection != null) return;
         EditorInfo editorInfo = new EditorInfo();
         mInputConnection = mContentViewCore.onCreateInputConnection(editorInfo);
+        mEditorInfoList.add(editorInfo);
     }
 
     @Override
     public boolean isActive(View view) {
         boolean result = mInputConnection != null;
         Log.d(TAG, "isActive: returns [%b]", result);
         return result;
     }
 
     @Override
@@ skipping 55 matching lines @@
     public int getHideSoftInputCounter() {
         return mHideSoftInputCounter;
     }
 
     public void reset() {
         Log.d(TAG, "reset");
         mRestartInputCounter = 0;
         mShowSoftInputCounter = 0;
         mHideSoftInputCounter = 0;
         mUpdateSelectionList.clear();
+        mEditorInfoList.clear();
     }
 
     public InputConnection getInputConnection() {
         return mInputConnection;
     }
 
     public Range getSelection() {
         return mSelection;
     }
 
@@ skipping 10 matching lines @@
     }
 
     public void clearLastCursorAnchorInfo() {
         mLastCursorAnchorInfo = null;
     }
 
     public CursorAnchorInfo getLastCursorAnchorInfo() {
         return mLastCursorAnchorInfo;
     }
 
+    public ArrayList<EditorInfo> getEditorInfoList() {
+        return mEditorInfoList;
+    }
+
     public void onUpdateSelection(Range oldSel, Range oldComp, Range newSel, Range newComp) {}
 
     public void expectsSelectionOutsideComposition() {}
 }