Make limitations for input.ime.sendKeyEvents API.

chrome/browser/ui/views/ime/input_ime_apitest_nonchromeos.cc

Index: chrome/browser/ui/views/ime/input_ime_apitest_nonchromeos.cc
diff --git a/chrome/browser/ui/views/ime/input_ime_apitest_nonchromeos.cc b/chrome/browser/ui/views/ime/input_ime_apitest_nonchromeos.cc
index f49c3d778faec7d952ee69cbf7d309fb69c60f04..4fcc232ce932bcb0864106a8d2ef053cccf44125 100644
--- a/chrome/browser/ui/views/ime/input_ime_apitest_nonchromeos.cc
+++ b/chrome/browser/ui/views/ime/input_ime_apitest_nonchromeos.cc
@@ -2,11 +2,14 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
+#include "base/auto_reset.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/extensions/api/input_ime/input_ime_api_nonchromeos.h"
 #include "chrome/browser/extensions/extension_apitest.h"
-#include "chrome/browser/ui/views/frame/browser_view.h"
+#include "chrome/browser/ui/browser_window.h"
 #include "chrome/common/chrome_switches.h"
+#include "chrome/common/url_constants.h"
+#include "chrome/test/base/ui_test_utils.h"
 #include "extensions/test/extension_test_message_listener.h"
 #include "ui/aura/window.h"
 #include "ui/aura/window_tree_host.h"
@@ -26,31 +29,58 @@ class InputImeApiTest : public ExtensionApiTest {
     command_line->AppendSwitch(switches::kEnableInputImeAPI);
   }
 
+  // Sets the private flag of |track_key_events_for_testing_| in InputMethod.
+  void SetTrackKeyEvents(ui::InputMethod* input_method, bool track) {
+    input_method->track_key_events_for_testing_ = track;
+  }
+
+  // Returns true if the key events get from input method equals to the expected
+  // key events.
+  bool CompareKeyEvents(
+      const std::vector<std::unique_ptr<ui::KeyEvent>>& expected_key_events,
+      ui::InputMethod* input_method) {
+    if (expected_key_events.size() != GetKeyEvents(input_method).size())
+      return false;
+    for (size_t i = 0; i < expected_key_events.size(); i++) {
+      ui::KeyEvent* event1 = expected_key_events[i].get();
+      ui::KeyEvent* event2 = GetKeyEvents(input_method)[i].get();
+      if (event1->type() != event2->type() ||
+          event1->key_code() != event2->key_code() ||
+          event1->flags() != event2->flags()) {
+        return false;
+      }
+    }
+    return true;
+  }
+
  private:
+  // Returns the tracked key events of using input.ime.sendKeyEvents API.
+  const std::vector<std::unique_ptr<ui::KeyEvent>>& GetKeyEvents(
+      ui::InputMethod* input_method) {
+    return input_method->GetKeyEventsForTesting();
+  }
+
   DISALLOW_COPY_AND_ASSIGN(InputImeApiTest);
 };
 
 IN_PROC_BROWSER_TEST_F(InputImeApiTest, BasicApiTest) {
   // Manipulates the focused text input client because the follow cursor
   // window requires the text input focus.
-  BrowserView* browser_view = BrowserView::GetBrowserViewForBrowser(browser());
   ui::InputMethod* input_method =
-      browser_view->GetNativeWindow()->GetHost()->GetInputMethod();
+      browser()->window()->GetNativeWindow()->GetHost()->GetInputMethod();
   std::unique_ptr<ui::DummyTextInputClient> client(
       new ui::DummyTextInputClient(ui::TEXT_INPUT_TYPE_TEXT));
   input_method->SetFocusedTextInputClient(client.get());
   ExtensionFunction::ScopedUserGestureForTests scoped_user_gesture;
-  InputImeActivateFunction::disable_bubble_for_testing_ = true;
+  base::AutoReset<bool> auto_reset_disable_bubble(
+      &InputImeActivateFunction::disable_bubble_for_testing_, true);
+  SetTrackKeyEvents(input_method, true);
 
   // Listens for the input.ime.onBlur event.
   ExtensionTestMessageListener blur_listener("get_blur_event", false);
 
   ASSERT_TRUE(RunExtensionTest("input_ime_nonchromeos")) << message_;
 
-  // Test the input.ime.sendKeyEvents API.
-  ASSERT_EQ(1, client->insert_char_count());
-  EXPECT_EQ(L'a', client->last_insert_char());
-
   // Test the input.ime.commitText API.
   ASSERT_EQ(1, client->insert_text_count());
   EXPECT_EQ(base::UTF8ToUTF16("test_commit_text"), client->last_insert_text());
@@ -75,4 +105,64 @@ IN_PROC_BROWSER_TEST_F(InputImeApiTest, BasicApiTest) {
   input_method->DetachTextInputClient(client2.get());
 }
 
+IN_PROC_BROWSER_TEST_F(InputImeApiTest, SendKeyEvntsOnNormalPage) {
+  // Navigates to special page that sendKeyEvents API has limition with.
+  ui_test_utils::NavigateToURL(browser(), GURL(chrome::kChromeUINewTabURL));
+  // Manipulates the focused text input client because the follow cursor
+  // window requires the text input focus.
+  ui::InputMethod* input_method =
+      browser()->window()->GetNativeWindow()->GetHost()->GetInputMethod();
+  std::unique_ptr<ui::DummyTextInputClient> client(
+      new ui::DummyTextInputClient(ui::TEXT_INPUT_TYPE_TEXT));
+  input_method->SetFocusedTextInputClient(client.get());
+  ExtensionFunction::ScopedUserGestureForTests scoped_user_gesture;
+  base::AutoReset<bool> auto_reset_disable_bubble(
+      &InputImeActivateFunction::disable_bubble_for_testing_, true);
+  SetTrackKeyEvents(input_method, true);
+
+  ASSERT_TRUE(RunExtensionTest("input_ime_nonchromeos")) << message_;
+
+  std::vector<std::unique_ptr<ui::KeyEvent>> key_events;
+  key_events.push_back(std::unique_ptr<ui::KeyEvent>(
+      new ui::KeyEvent(ui::ET_KEY_PRESSED, ui::VKEY_A, ui::EF_NONE)));
+  key_events.push_back(std::unique_ptr<ui::KeyEvent>(
+      new ui::KeyEvent(ui::ET_KEY_RELEASED, ui::VKEY_A, ui::EF_NONE)));
+  key_events.push_back(std::unique_ptr<ui::KeyEvent>(
+      new ui::KeyEvent(ui::ET_KEY_PRESSED, ui::VKEY_A, ui::EF_CONTROL_DOWN)));
+  key_events.push_back(std::unique_ptr<ui::KeyEvent>(
+      new ui::KeyEvent(ui::ET_KEY_RELEASED, ui::VKEY_A, ui::EF_CONTROL_DOWN)));
+  key_events.push_back(std::unique_ptr<ui::KeyEvent>(
+      new ui::KeyEvent(ui::ET_KEY_PRESSED, ui::VKEY_TAB, ui::EF_NONE)));
+
+  EXPECT_TRUE(CompareKeyEvents(key_events, input_method));
+
+  input_method->DetachTextInputClient(client.get());
+}
+
+IN_PROC_BROWSER_TEST_F(InputImeApiTest, SendKeyEventsOnSpecialPage) {
+  // Navigates to special page that sendKeyEvents API has limition with.
+  ui_test_utils::NavigateToURL(browser(), GURL("chrome://flags"));
+
+  ui::InputMethod* input_method =
+      browser()->window()->GetNativeWindow()->GetHost()->GetInputMethod();
+  std::unique_ptr<ui::DummyTextInputClient> client(
+      new ui::DummyTextInputClient(ui::TEXT_INPUT_TYPE_TEXT));
+  input_method->SetFocusedTextInputClient(client.get());
+  ExtensionFunction::ScopedUserGestureForTests scoped_user_gesture;
+  base::AutoReset<bool> auto_reset_disable_bubble(
+      &InputImeActivateFunction::disable_bubble_for_testing_, true);
+  SetTrackKeyEvents(input_method, true);
+
+  ASSERT_TRUE(RunExtensionTest("input_ime_nonchromeos")) << message_;
+
+  std::vector<std::unique_ptr<ui::KeyEvent>> key_events;
+  key_events.push_back(std::unique_ptr<ui::KeyEvent>(
+      new ui::KeyEvent(ui::ET_KEY_PRESSED, ui::VKEY_A, ui::EF_NONE)));
+  key_events.push_back(std::unique_ptr<ui::KeyEvent>(
+      new ui::KeyEvent(ui::ET_KEY_RELEASED, ui::VKEY_A, ui::EF_NONE)));
+
+  EXPECT_TRUE(CompareKeyEvents(key_events, input_method));
+  input_method->DetachTextInputClient(client.get());
+}
+
 }  // namespace extensions