Fix SitePerProcessTextInputManagerTest.StopTrackingCrashedChildFrame on CFI Bots.

content/public/test/text_input_test_utils.cc

 // Copyright 2016 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.
 
 #include "content/public/test/text_input_test_utils.h"
 
+#include <unordered_set>
+
 #include "content/browser/renderer_host/render_widget_host_impl.h"
 #include "content/browser/renderer_host/render_widget_host_view_aura.h"
 #include "content/browser/renderer_host/render_widget_host_view_base.h"
 #include "content/browser/renderer_host/render_widget_host_view_base_observer.h"
 #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/public/browser/render_widget_host_view.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_observer.h"
@@ skipping 172 matching lines @@
                              RenderWidgetHostView* view,
                              ui::TextInputType* type) {
   TextInputManager* manager =
       static_cast<WebContentsImpl*>(web_contents)->GetTextInputManager();
 
   RenderWidgetHostViewBase* view_base =
       static_cast<RenderWidgetHostViewBase*>(view);
   if (!manager || !manager->IsRegistered(view_base))
     return false;
 
-  *type = manager->text_input_state_map_[view_base].type;
+  *type = manager->GetTextInputTypeForViewForTesting(view_base);
+
   return true;
 }
 
+size_t GetRegisteredViewsCountFromTextInputManager(WebContents* web_contents) {
+  std::unordered_set<RenderWidgetHostView*> views;
+  TextInputManager* manager =
+      static_cast<WebContentsImpl*>(web_contents)->GetTextInputManager();
+  return !!manager ? manager->GetRegisteredViewsCountForTesting() : 0;
+}
+
 RenderWidgetHostView* GetActiveViewFromWebContents(WebContents* web_contents) {
   return static_cast<WebContentsImpl*>(web_contents)
       ->GetTextInputManager()
       ->active_view_for_testing();
 }
 
 TextInputManagerTester::TextInputManagerTester(WebContents* web_contents)
     : observer_(new InternalObserver(web_contents, this)) {}
 
 TextInputManagerTester::~TextInputManagerTester() {}
@@ skipping 104 matching lines @@
 
 #ifdef USE_AURA
   RenderWidgetHostViewAura* view = static_cast<RenderWidgetHostViewAura*>(
       web_contents->GetRenderWidgetHostView());
   observer.reset(new InputMethodObserverAura(view->GetInputMethod()));
 #endif
   return observer;
 }
 
 }  // namespace content