Allow selection change update before beginBatchEdit

content/public/android/junit/src/org/chromium/content/browser/input/ThreadedInputConnectionTest.java

Index: content/public/android/junit/src/org/chromium/content/browser/input/ThreadedInputConnectionTest.java
diff --git a/content/public/android/junit/src/org/chromium/content/browser/input/ThreadedInputConnectionTest.java b/content/public/android/junit/src/org/chromium/content/browser/input/ThreadedInputConnectionTest.java
index 124e3259f300fd0b7d7993d50870ee54bdd92f72..47b471699f26effdbf6d81b1fa1fbecdacadfc43 100644
--- a/content/public/android/junit/src/org/chromium/content/browser/input/ThreadedInputConnectionTest.java
+++ b/content/public/android/junit/src/org/chromium/content/browser/input/ThreadedInputConnectionTest.java
@@ -70,12 +70,12 @@ public class ThreadedInputConnectionTest {
         mInOrder.verify(mImeAdapter).sendCompositionToNative("hello", 1, false, 0);
 
         // Renderer updates states asynchronously.
-        mConnection.updateStateOnUiThread("hello", 5, 5, 0, 5, true, true);
+        mConnection.updateStateOnUiThread("hello", 5, 5, 0, 5, true, true, false);
         mInOrder.verify(mImeAdapter).updateSelection(5, 5, 0, 5);
         assertEquals(0, mConnection.getQueueForTest().size());
 
         // Prepare to call requestTextInputStateUpdate.
-        mConnection.updateStateOnUiThread("hello", 5, 5, 0, 5, true, false);
+        mConnection.updateStateOnUiThread("hello", 5, 5, 0, 5, true, false, false);
         assertEquals(1, mConnection.getQueueForTest().size());
         when(mImeAdapter.requestTextInputStateUpdate()).thenReturn(true);
 
@@ -85,11 +85,11 @@ public class ThreadedInputConnectionTest {
         // IME app calls finishComposingText().
         mConnection.finishComposingText();
         mInOrder.verify(mImeAdapter).finishComposingText();
-        mConnection.updateStateOnUiThread("hello", 5, 5, -1, -1, true, true);
+        mConnection.updateStateOnUiThread("hello", 5, 5, -1, -1, true, true, false);
         mInOrder.verify(mImeAdapter).updateSelection(5, 5, -1, -1);
 
         // Prepare to call requestTextInputStateUpdate.
-        mConnection.updateStateOnUiThread("hello", 5, 5, -1, -1, true, false);
+        mConnection.updateStateOnUiThread("hello", 5, 5, -1, -1, true, false, false);
         assertEquals(1, mConnection.getQueueForTest().size());
         when(mImeAdapter.requestTextInputStateUpdate()).thenReturn(true);
 
@@ -114,7 +114,7 @@ public class ThreadedInputConnectionTest {
     @Feature({"TextInput"})
     public void testRenderChangeUpdatesSelection() {
         // User moves the cursor.
-        mConnection.updateStateOnUiThread("hello", 4, 4, -1, -1, true, true);
+        mConnection.updateStateOnUiThread("hello", 4, 4, -1, -1, true, true, false);
         mInOrder.verify(mImeAdapter).updateSelection(4, 4, -1, -1);
         assertEquals(0, mConnection.getQueueForTest().size());
     }
@@ -122,15 +122,24 @@ public class ThreadedInputConnectionTest {
     @Test
     @Feature({"TextInput"})
     public void testBatchEdit() {
+        // Type 'hello'.
+        assertTrue(mConnection.commitText("hello ", 1));
         // IME app calls beginBatchEdit().
         assertTrue(mConnection.beginBatchEdit());
-        // Type hello real fast.
-        mConnection.commitText("hello", 1);
-        mInOrder.verify(mImeAdapter).sendCompositionToNative("hello", 1, true, 0);
+        // Type 'world'.
+        assertTrue(mConnection.commitText("world", 1));
+        mInOrder.verify(mImeAdapter).sendCompositionToNative("hello ", 1, true, 0);
+        mInOrder.verify(mImeAdapter).beginBatchEdit();
+        mInOrder.verify(mImeAdapter).sendCompositionToNative("world", 1, true, 0);
 
         // Renderer updates states asynchronously.
-        mConnection.updateStateOnUiThread("hello", 5, 5, -1, -1, true, true);
-        mInOrder.verify(mImeAdapter, never()).updateSelection(5, 5, -1, -1);
+        mConnection.updateStateOnUiThread(
+                "hello ", 6, 6, -1, -1, true, true, false /* batchEdit */);
+        mConnection.updateStateOnUiThread(
+                "world", 11, 11, -1, -1, true, true, true /* batchEdit */);
+        // 'hello ' is called before beginBatchEdit(), so we still need to update it.
+        mInOrder.verify(mImeAdapter).updateSelection(6, 6, -1, -1);
+        mInOrder.verify(mImeAdapter, never()).updateSelection(11, 11, -1, -1);
         assertEquals(0, mConnection.getQueueForTest().size());
 
         {
@@ -140,18 +149,24 @@ public class ThreadedInputConnectionTest {
             mConnection.setSelection(4, 4);
             assertTrue(mConnection.endBatchEdit());
         }
+        mInOrder.verify(mImeAdapter).setEditableSelectionOffsets(4, 4);
+
+        // Renderer updates state for setSelection().
+        mConnection.updateStateOnUiThread(
+                "hello", 4, 4, -1, -1, true, true, true /* batchEdit */);
+
         // We still have one outer batch edit, so should not update selection yet.
         mInOrder.verify(mImeAdapter, never()).updateSelection(4, 4, -1, -1);
 
-        // Prepare to call requestTextInputStateUpdate.
-        mConnection.updateStateOnUiThread("hello", 4, 4, -1, -1, true, false);
-        assertEquals(1, mConnection.getQueueForTest().size());
-        when(mImeAdapter.requestTextInputStateUpdate()).thenReturn(true);
-
         // IME app calls endBatchEdit().
         assertFalse(mConnection.endBatchEdit());
         // Batch edit is finished, now update selection.
+        mConnection.updateStateOnUiThread(
+                "hello", 4, 4, -1, -1, true, true, false /* batchEdit */);
+
+        mInOrder.verify(mImeAdapter).endBatchEdit();
         mInOrder.verify(mImeAdapter).updateSelection(4, 4, -1, -1);
+        mInOrder.verifyNoMoreInteractions();
         assertEquals(0, mConnection.getQueueForTest().size());
     }