Tracking SelectionBounds for all RenderWidgets on the Browser Side (Aura Only)

content/public/test/text_input_test_utils.cc

Index: content/public/test/text_input_test_utils.cc
diff --git a/content/public/test/text_input_test_utils.cc b/content/public/test/text_input_test_utils.cc
index ed3e7483070dd4393ea62bfc5c8fa6e7fcfa2068..1a78378ba989762980d274309da76d26225b3171 100644
--- a/content/public/test/text_input_test_utils.cc
+++ b/content/public/test/text_input_test_utils.cc
@@ -13,6 +13,7 @@
 #include "content/browser/renderer_host/text_input_manager.h"
 #include "content/browser/web_contents/web_contents_impl.h"
 #include "content/common/text_input_state.h"
+#include "content/common/view_messages.h"
 #include "content/public/browser/render_widget_host_view.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_observer.h"
@@ -26,6 +27,12 @@ class TextInputClient;
 
 namespace content {
 
+namespace {
+void DefaultTextInputManagerObserverCallback() {}
+TextInputManagerTester::Callback default_text_input_manager_observer_callback =
+    base::Bind(&DefaultTextInputManagerObserverCallback);
+}  // namespace
+
 // This class is an observer of TextInputManager associated with the provided
 // WebContents. An instance of this class is used in TextInputManagerTester to
 // expose the required API for testing outside of content/.
@@ -35,9 +42,12 @@ class TextInputManagerTester::InternalObserver
  public:
   InternalObserver(WebContents* web_contents, TextInputManagerTester* tester)
       : WebContentsObserver(web_contents),
-        tester_(tester),
         updated_view_(nullptr),
-        text_input_state_changed_(false) {
+        text_input_state_changed_(false),
+        update_text_input_state_callback_(
+            default_text_input_manager_observer_callback),
+        on_selection_bounds_changed_callback_(
+            default_text_input_manager_observer_callback) {
     text_input_manager_ =
         static_cast<WebContentsImpl*>(web_contents)->GetTextInputManager();
     DCHECK(!!text_input_manager_);
@@ -54,6 +64,11 @@ class TextInputManagerTester::InternalObserver
     update_text_input_state_callback_ = callback;
   }
 
+  void set_on_selection_bounds_changed_callback(
+      const TextInputManagerTester::Callback& callback) {
+    on_selection_bounds_changed_callback_ = callback;
+  }
+
   const RenderWidgetHostView* GetUpdatedView() const { return updated_view_; }
 
   bool text_input_state_changed() const { return text_input_state_changed_; }
@@ -68,18 +83,25 @@ class TextInputManagerTester::InternalObserver
       return;
     text_input_state_changed_ = did_change_state;
     updated_view_ = updated_view;
-    update_text_input_state_callback_.Run(tester_);
+    update_text_input_state_callback_.Run();
+  }
+
+  void OnSelectionBoundsChanged(
+      TextInputManager* text_input_manager_,
+      RenderWidgetHostViewBase* updated_view) override {
+    updated_view_ = updated_view;
+    on_selection_bounds_changed_callback_.Run();
   }
 
   // WebContentsObserver implementation.
   void WebContentsDestroyed() override { text_input_manager_ = nullptr; }
 
  private:
-  TextInputManagerTester* tester_;
   TextInputManager* text_input_manager_;
   RenderWidgetHostViewBase* updated_view_;
   bool text_input_state_changed_;
   TextInputManagerTester::Callback update_text_input_state_callback_;
+  TextInputManagerTester::Callback on_selection_bounds_changed_callback_;
 
   DISALLOW_COPY_AND_ASSIGN(InternalObserver);
 };
@@ -226,6 +248,11 @@ void TextInputManagerTester::SetUpdateTextInputStateCalledCallback(
   observer_->set_update_text_input_state_called_callback(callback);
 }
 
+void TextInputManagerTester::SetOnSelectionBoundsChangedCallback(
+    const Callback& callback) {
+  observer_->set_on_selection_bounds_changed_callback(callback);
+}
+
 bool TextInputManagerTester::GetTextInputType(ui::TextInputType* type) {
   DCHECK(observer_->text_input_manager());
   const TextInputState* state =