Revert of chromeos: Remove SystemTrayDelegate::GetIMEManagedMessage()

ash/system/ime_menu/ime_menu_tray_unittest.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 "ash/system/ime_menu/ime_menu_tray.h"
 
 #include "ash/accelerators/accelerator_controller.h"
 #include "ash/accessibility_delegate.h"
 #include "ash/shell.h"
 #include "ash/system/ime_menu/ime_list_view.h"
 #include "ash/system/status_area_widget.h"
 #include "ash/system/tray/ime_info.h"
 #include "ash/system/tray/system_tray_notifier.h"
 #include "ash/test/ash_test_base.h"
 #include "ash/test/status_area_widget_test_helper.h"
 #include "ash/test/test_system_tray_delegate.h"
 #include "base/run_loop.h"
 #include "base/strings/utf_string_conversions.h"
 #include "ui/accessibility/ax_node_data.h"
 #include "ui/base/ime/chromeos/input_method_manager.h"
 #include "ui/base/ime/chromeos/mock_input_method_manager.h"
 #include "ui/base/ime/ime_bridge.h"
 #include "ui/base/ime/text_input_flags.h"
 #include "ui/events/event.h"
 #include "ui/views/controls/label.h"
 
 using base::UTF8ToUTF16;
-using chromeos::input_method::InputMethodManager;
-using chromeos::input_method::MockInputMethodManager;
 
 namespace ash {
 
 ImeMenuTray* GetTray() {
   return StatusAreaWidgetTestHelper::GetStatusAreaWidget()->ime_menu_tray();
 }
 
-// An InputMethodManager that always returns an active IME state.
-class TestInputMethodManager : public MockInputMethodManager {
- public:
-  TestInputMethodManager() : state_(new MockInputMethodManager::State) {}
-  ~TestInputMethodManager() override = default;
-
-  // MockInputMethodManager:
-  scoped_refptr<InputMethodManager::State> GetActiveIMEState() override {
-    return state_;
-  }
-
-  scoped_refptr<MockInputMethodManager::State> state_;
-
- private:
-  DISALLOW_COPY_AND_ASSIGN(TestInputMethodManager);
-};
-
 class ImeMenuTrayTest : public test::AshTestBase {
  public:
-  ImeMenuTrayTest() = default;
-  ~ImeMenuTrayTest() override = default;
-
-  // test::AshTestBase:
-  void SetUp() override {
-    test::AshTestBase::SetUp();
-    // Takes ownership.
-    InputMethodManager::Initialize(new TestInputMethodManager);
-  }
-  void TearDown() override {
-    InputMethodManager::Shutdown();
-    test::AshTestBase::TearDown();
-  }
+  ImeMenuTrayTest() {}
+  ~ImeMenuTrayTest() override {}
 
  protected:
   // Returns true if the IME menu tray is visible.
   bool IsVisible() { return GetTray()->visible(); }
 
   // Returns the label text of the tray.
   const base::string16& GetTrayText() { return GetTray()->label_->text(); }
 
   // Returns true if the background color of the tray is active.
   bool IsTrayBackgroundActive() { return GetTray()->is_active(); }
@@ skipping 248 matching lines @@
   EXPECT_TRUE(accessibility_delegate->IsVirtualKeyboardEnabled());
 
   // Hides the keyboard.
   GetTray()->OnKeyboardHidden();
   // The keyboard should still be disabled, which is a posted task.
   base::RunLoop().RunUntilIdle();
   EXPECT_FALSE(accessibility_delegate->IsVirtualKeyboardEnabled());
 }
 
 TEST_F(ImeMenuTrayTest, ShowEmojiHandwritingVoiceButtons) {
-  InputMethodManager* input_method_manager = InputMethodManager::Get();
-  ASSERT_TRUE(input_method_manager);
-  // Feature is enabled.
-  ASSERT_TRUE(input_method_manager->IsEmojiHandwritingVoiceOnImeMenuEnabled());
+  FocusInInputContext(ui::TEXT_INPUT_TYPE_TEXT);
+  EXPECT_FALSE(GetTray()->ShouldShowEmojiHandwritingVoiceButtons());
 
-  // Text fields should show the buttons.
-  FocusInInputContext(ui::TEXT_INPUT_TYPE_TEXT);
+  chromeos::input_method::InputMethodManager* input_method_manager =
+      chromeos::input_method::InputMethodManager::Get();
+  EXPECT_FALSE(input_method_manager);
+  chromeos::input_method::InputMethodManager::Initialize(
+      new chromeos::input_method::MockInputMethodManager);
+  input_method_manager = chromeos::input_method::InputMethodManager::Get();
+  EXPECT_TRUE(input_method_manager &&
+              input_method_manager->IsEmojiHandwritingVoiceOnImeMenuEnabled());
   EXPECT_TRUE(GetTray()->ShouldShowEmojiHandwritingVoiceButtons());
 
-  // Password fields should not show the buttons.
   FocusInInputContext(ui::TEXT_INPUT_TYPE_PASSWORD);
   EXPECT_FALSE(GetTray()->ShouldShowEmojiHandwritingVoiceButtons());
-
-  // Lock screen should not show the buttons.
-  BlockUserSession(BLOCKED_BY_LOCK_SCREEN);
-  FocusInInputContext(ui::TEXT_INPUT_TYPE_TEXT);
-  EXPECT_FALSE(GetTray()->ShouldShowEmojiHandwritingVoiceButtons());
 }
 
 }  // namespace ash