Fix non-Latin key event handling for menu mnemonics

ui/views/controls/menu/menu_runner_unittest.cc

 // Copyright 2015 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/views/controls/menu/menu_runner.h"
 
 #include <stdint.h>
 
 #include "base/macros.h"
 #include "base/memory/scoped_ptr.h"
+#include "base/strings/utf_string_conversions.h"
 #include "ui/base/ui_base_types.h"
 #include "ui/events/test/event_generator.h"
 #include "ui/views/controls/menu/menu_delegate.h"
 #include "ui/views/controls/menu/menu_item_view.h"
 #include "ui/views/controls/menu/menu_types.h"
+#include "ui/views/controls/menu/submenu_view.h"
 #include "ui/views/test/views_test_base.h"
 #include "ui/views/widget/widget.h"
 
 namespace views {
 namespace test {
 
 // Implementation of MenuDelegate that only reports the values of calls to
-// OnMenuClosed.
+// OnMenuClosed and ExecuteCommand.
 class TestMenuDelegate : public MenuDelegate {
  public:
   TestMenuDelegate();
   ~TestMenuDelegate() override;
 
-  int on_menu_closed_called() { return on_menu_closed_called_; }
-  MenuItemView* on_menu_closed_menu() { return on_menu_closed_menu_; }
-  MenuRunner::RunResult on_menu_closed_run_result() {
+  int execute_command_id() const { return execute_command_id_; }
+
+  int on_menu_closed_called() const { return on_menu_closed_called_; }
+  MenuItemView* on_menu_closed_menu() const { return on_menu_closed_menu_; }
+  MenuRunner::RunResult on_menu_closed_run_result() const {
     return on_menu_closed_run_result_;
   }
 
   // MenuDelegate:
+  void ExecuteCommand(int id) override;
   void OnMenuClosed(MenuItemView* menu, MenuRunner::RunResult result) override;
 
  private:
+  // ID of last executed command.
+  int execute_command_id_;
+
   // The number of times OnMenuClosed was called.
   int on_menu_closed_called_;
 
   // The values of the last call to OnMenuClosed.
   MenuItemView* on_menu_closed_menu_;
   MenuRunner::RunResult on_menu_closed_run_result_;
 
   DISALLOW_COPY_AND_ASSIGN(TestMenuDelegate);
 };
 
 TestMenuDelegate::TestMenuDelegate()
-    : on_menu_closed_called_(0),
+    : execute_command_id_(0),
+      on_menu_closed_called_(0),
       on_menu_closed_menu_(nullptr),
       on_menu_closed_run_result_(MenuRunner::MENU_DELETED) {}
 
 TestMenuDelegate::~TestMenuDelegate() {}
 
+void TestMenuDelegate::ExecuteCommand(int id) {
+  execute_command_id_ = id;
+}
+
 void TestMenuDelegate::OnMenuClosed(MenuItemView* menu,
                                     MenuRunner::RunResult result) {
   on_menu_closed_called_++;
   on_menu_closed_menu_ = menu;
   on_menu_closed_run_result_ = result;
 }
 
 class MenuRunnerTest : public ViewsTestBase {
  public:
   MenuRunnerTest();
@@ skipping 28 matching lines @@
 MenuRunnerTest::~MenuRunnerTest() {}
 
 void MenuRunnerTest::InitMenuRunner(int32_t run_types) {
   menu_runner_.reset(new MenuRunner(menu_item_view_, run_types));
 }
 
 void MenuRunnerTest::SetUp() {
   ViewsTestBase::SetUp();
   menu_delegate_.reset(new TestMenuDelegate);
   menu_item_view_ = new MenuItemView(menu_delegate_.get());
+  menu_item_view_->AppendMenuItemWithLabel(1, base::ASCIIToUTF16("One"));
+  menu_item_view_->AppendMenuItemWithLabel(2,
+                                           base::WideToUTF16(L"\x062f\x0648"));
 
   owner_.reset(new Widget);
   Widget::InitParams params = CreateParams(Widget::InitParams::TYPE_POPUP);
   params.ownership = Widget::InitParams::WIDGET_OWNS_NATIVE_WIDGET;
   owner_->Init(params);
   owner_->Show();
 }
 
 void MenuRunnerTest::TearDown() {
   owner_->CloseNow();
@@ skipping 31 matching lines @@
 
   ui::test::EventGenerator generator(GetContext(), owner()->GetNativeWindow());
   generator.PressKey(ui::VKEY_ESCAPE, 0);
   EXPECT_FALSE(runner->IsRunning());
   TestMenuDelegate* delegate = menu_delegate();
   EXPECT_EQ(1, delegate->on_menu_closed_called());
   EXPECT_EQ(nullptr, delegate->on_menu_closed_menu());
   EXPECT_EQ(MenuRunner::NORMAL_EXIT, delegate->on_menu_closed_run_result());
 }
 
+// Tests that a key press on a US keyboard layout activates the correct menu
+// item.
+TEST_F(MenuRunnerTest, LatinMnemonic) {
+  InitMenuRunner(MenuRunner::ASYNC);
+  MenuRunner* runner = menu_runner();
+  MenuRunner::RunResult result = runner->RunMenuAt(
+      owner(), nullptr, gfx::Rect(), MENU_ANCHOR_TOPLEFT, ui::MENU_SOURCE_NONE);
+  EXPECT_EQ(MenuRunner::NORMAL_EXIT, result);
+  EXPECT_TRUE(runner->IsRunning());
+
+  ui::test::EventGenerator generator(GetContext(), owner()->GetNativeWindow());
+  generator.PressKey(ui::VKEY_O, 0);
+  EXPECT_FALSE(runner->IsRunning());
+  TestMenuDelegate* delegate = menu_delegate();
+  EXPECT_EQ(1, delegate->execute_command_id());
+  EXPECT_EQ(1, delegate->on_menu_closed_called());
+  EXPECT_NE(nullptr, delegate->on_menu_closed_menu());
+  EXPECT_EQ(MenuRunner::NORMAL_EXIT, delegate->on_menu_closed_run_result());
+}
+
+// Tests that a key press on a non-US keyboard layout activates the correct menu
+// item.
+TEST_F(MenuRunnerTest, NonLatinMnemonic) {
+  InitMenuRunner(MenuRunner::ASYNC);
+  MenuRunner* runner = menu_runner();
+  MenuRunner::RunResult result = runner->RunMenuAt(
+      owner(), nullptr, gfx::Rect(), MENU_ANCHOR_TOPLEFT, ui::MENU_SOURCE_NONE);
+  EXPECT_EQ(MenuRunner::NORMAL_EXIT, result);
+  EXPECT_TRUE(runner->IsRunning());
+
+  ui::test::EventGenerator generator(GetContext(), owner()->GetNativeWindow());
+  ui::KeyEvent key_press(0x062f, ui::VKEY_N, 0);
+  generator.Dispatch(&key_press);
+  EXPECT_FALSE(runner->IsRunning());
+  TestMenuDelegate* delegate = menu_delegate();
+  EXPECT_EQ(2, delegate->execute_command_id());
+  EXPECT_EQ(1, delegate->on_menu_closed_called());
+  EXPECT_NE(nullptr, delegate->on_menu_closed_menu());
+  EXPECT_EQ(MenuRunner::NORMAL_EXIT, delegate->on_menu_closed_run_result());
+}
+
 }  // namespace test
 }  // namespace views