Fixes unittests with --enable-text-input-focus-manager flag enabled.

ui/base/ime/mock_input_method.cc

 // Copyright (c) 2012 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 "ui/base/ime/mock_input_method.h"
 
+#include "ui/base/ime/text_input_focus_manager.h"
+#include "ui/base/ui_base_switches_util.h"
+
 namespace ui {
 
 MockInputMethod::MockInputMethod(internal::InputMethodDelegate* delegate)
     : text_input_client_(NULL) {
 }
 
 MockInputMethod::~MockInputMethod() {
 }
 
 void MockInputMethod::SetDelegate(internal::InputMethodDelegate* delegate) {
 }
 
 void MockInputMethod::SetFocusedTextInputClient(TextInputClient* client) {
+  if (switches::IsTextInputFocusManagerEnabled())
+    return;
+
   if (text_input_client_ == client)
     return;
   text_input_client_ = client;
   if (client)
     OnTextInputTypeChanged(client);
 }
 
 void MockInputMethod::DetachTextInputClient(TextInputClient* client) {
   if (text_input_client_ == client) {
     text_input_client_ = NULL;
   }
 }
 
 TextInputClient* MockInputMethod::GetTextInputClient() const {
+  if (switches::IsTextInputFocusManagerEnabled())
+    return TextInputFocusManager::GetInstance()->GetFocusedTextInputClient();
+
   return text_input_client_;
 }
 
 bool MockInputMethod::DispatchKeyEvent(const ui::KeyEvent& event) {
   return false;
 }
 
 void MockInputMethod::Init(bool focused) {
 }
 
@@ skipping 63 matching lines @@
 
 void MockInputMethod::AddObserver(InputMethodObserver* observer) {
   observer_list_.AddObserver(observer);
 }
 
 void MockInputMethod::RemoveObserver(InputMethodObserver* observer) {
   observer_list_.RemoveObserver(observer);
 }
 
 }  // namespace ui