Allow Widget to pass accelerator handling to its parent.

ui/views/widget/native_widget_mac_unittest.mm

 // Copyright 2014 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.
 
 #import "ui/views/widget/native_widget_mac.h"
 
 #import <Cocoa/Cocoa.h>
 
 #import "base/mac/foundation_util.h"
 #include "base/mac/mac_util.h"
 #import "base/mac/scoped_nsautorelease_pool.h"
 #import "base/mac/scoped_nsobject.h"
 #import "base/mac/scoped_objc_class_swizzler.h"
 #include "base/macros.h"
+#include "base/optional.h"
 #include "base/run_loop.h"
 #include "base/strings/sys_string_conversions.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/test/test_timeouts.h"
 #include "base/threading/thread_task_runner_handle.h"
 #import "testing/gtest_mac.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 #include "third_party/skia/include/core/SkCanvas.h"
 #import "ui/base/cocoa/constrained_window/constrained_window_animation.h"
+#import "ui/base/cocoa/user_interface_item_command_handler.h"
 #import "ui/base/cocoa/window_size_constants.h"
 #import "ui/base/test/scoped_fake_full_keyboard_access.h"
 #import "ui/events/test/cocoa_test_event_utils.h"
 #include "ui/events/test/event_generator.h"
 #import "ui/gfx/mac/coordinate_conversion.h"
 #include "ui/views/bubble/bubble_dialog_delegate.h"
 #import "ui/views/cocoa/bridged_content_view.h"
 #import "ui/views/cocoa/bridged_native_widget.h"
 #import "ui/views/cocoa/native_widget_mac_nswindow.h"
 #include "ui/views/controls/button/label_button.h"
@@ skipping 45 matching lines @@
 @property(assign, nonatomic) NSRect lastDirtyRect;
 @end
 
 @interface FocusableTestNSView : NSView
 @end
 
 @interface TestNativeParentWindow : NSWindow
 @property(assign, nonatomic) bool* deallocFlag;
 @end
 
+@interface MockValidatedUserInterfaceItem
+    : NSObject<NSValidatedUserInterfaceItem>
+- (instancetype)initWithTagAndAction:(NSInteger)tag action:(SEL)action;
+@property(assign, nonatomic) base::Optional<int> handledById;
+@property(readonly) NSInteger tag;
+@property(readonly) SEL action;
+@end
+
+@interface MockCommandHandler : NSObject<UserInterfaceItemCommandHandler>
+- (instancetype)initWithId:(int)handlerId;
+@end
+
 namespace views {
 namespace test {
 
 // BridgedNativeWidget friend to access private members.
 class BridgedNativeWidgetTestApi {
  public:
   explicit BridgedNativeWidgetTestApi(NSWindow* window) {
     bridge_ = NativeWidgetMac::GetBridgeForNativeWindow(window);
   }
 
@@ skipping 1701 matching lines @@
     hosts_[2]->view(), child_view
   ]]));
 
   native_host_parent_->ReorderChildView(hosts_[2]->host(), 1);
   EXPECT_TRUE(([[GetContentNativeView() subviews] isEqualToArray:@[
     compositor_view_, hosts_[0]->view(), hosts_[2]->view(),
     hosts_[1]->view(), child_view
   ]]));
 }
 
+namespace {
+
+class TestNativeWidgetWithMockCommandHandler
+    : public TestWindowNativeWidgetMac {
+ public:
+  TestNativeWidgetWithMockCommandHandler(Widget* widget, int id)
+      : TestWindowNativeWidgetMac(widget), id_(id) {}
+
+ protected:
+  // TestWindowNativeWidgetMac:
+  NativeWidgetMacNSWindow* CreateNSWindow(
+      const Widget::InitParams& params) override {
+    NativeWidgetMacNSWindow* ns_window =
+        TestWindowNativeWidgetMac::CreateNSWindow(params);
+    [ns_window setCommandHandler:[[[MockCommandHandler alloc] initWithId:id_]
+                                     autorelease]];
+    return ns_window;
+  }
+
+ private:
+  int id_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestNativeWidgetWithMockCommandHandler);
+};
+
+}  // namespace
+
+class NativeWidgetMacCommandHandlerTest : public WidgetTest {
+ public:
+  NativeWidgetMacCommandHandlerTest() = default;
+
+ protected:
+  void SetUp() override {
+    WidgetTest::SetUp();
+
+    parent_ = base::MakeUnique<Widget>();
+    Widget::InitParams params;
+    params.native_widget =
+        new TestNativeWidgetWithMockCommandHandler(parent_.get(), parent_id_);
+    params.ownership = Widget::InitParams::WIDGET_OWNS_NATIVE_WIDGET;
+    parent_->Init(params);
+    parent_->Show();
+  }
+
+  void TearDown() override {
+    parent_.reset();
+
+    WidgetTest::TearDown();
+  }
+
+  base::scoped_nsobject<MockValidatedUserInterfaceItem> CreateUserInterfaceItem(
+      NSInteger tag) {
+    base::scoped_nsobject<MockValidatedUserInterfaceItem> result(
+        [[MockValidatedUserInterfaceItem alloc]
+            initWithTagAndAction:tag
+                          action:@selector(commandDispatch:)]);
+    return result;
+  }
+
+  const int parent_id_ = 0;
+  std::unique_ptr<Widget> parent_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(NativeWidgetMacCommandHandlerTest);
+};
+
+TEST_F(NativeWidgetMacCommandHandlerTest, CommandDispatch) {
+  auto item = CreateUserInterfaceItem(1);
+  id parent_native_window = parent_->GetNativeWindow();
+  ASSERT_TRUE(
+      [parent_native_window respondsToSelector:@selector(commandDispatch:)]);
+  [parent_native_window commandDispatch:item.get()];
+  EXPECT_EQ(parent_id_, [item.get() handledById]);
+}
+
+TEST_F(NativeWidgetMacCommandHandlerTest, CommandDispatchNoForwarding) {
+  std::unique_ptr<Widget> bubble(new Widget());
+  Widget::InitParams params;
+  params.native_widget = new TestWindowNativeWidgetMac(bubble.get());
+  params.ownership = Widget::InitParams::WIDGET_OWNS_NATIVE_WIDGET;
+  params.parent = parent_->GetNativeView();
+  params.pass_accelerator_to_parent = false;
+  bubble->Init(params);
+  bubble->Show();
+
+  auto item = CreateUserInterfaceItem(1);
+  id bubble_native_window = bubble->GetNativeWindow();
+  ASSERT_TRUE(
+      [bubble_native_window respondsToSelector:@selector(commandDispatch:)]);
+  [bubble_native_window commandDispatch:item.get()];
+  EXPECT_EQ(base::nullopt, [item.get() handledById]);
+}
+
+TEST_F(NativeWidgetMacCommandHandlerTest, CommandDispatchForwarding) {
+  std::unique_ptr<Widget> bubble(new Widget());
+  Widget::InitParams params;
+  params.native_widget = new TestWindowNativeWidgetMac(bubble.get());
+  params.ownership = Widget::InitParams::WIDGET_OWNS_NATIVE_WIDGET;
+  params.parent = parent_->GetNativeView();
+  params.pass_accelerator_to_parent = true;
+  bubble->Init(params);
+  bubble->Show();
+
+  auto item = CreateUserInterfaceItem(1);
+  id bubble_native_window = bubble->GetNativeWindow();
+  ASSERT_TRUE(
+      [bubble_native_window respondsToSelector:@selector(commandDispatch:)]);
+  [bubble_native_window commandDispatch:item.get()];
+  EXPECT_EQ(parent_id_, [item.get() handledById]);
+}
+
 }  // namespace test
 }  // namespace views
 
 @implementation TestStopAnimationWaiter
 - (void)setWindowStateForEnd {
   views::test::ScopedSwizzleWaiter::GetMethodAndMarkCalled()(self, _cmd);
 }
 @end
 
 @implementation NativeWidgetMacTestWindow
@@ skipping 42 matching lines @@
 
 - (void)dealloc {
   if (deallocFlag_) {
     DCHECK(!*deallocFlag_);
     *deallocFlag_ = true;
   }
   [super dealloc];
 }
 
 @end
+
+@implementation MockValidatedUserInterfaceItem
+
+@synthesize handledById = handledById_;
+@synthesize tag = tag_;
+@synthesize action = action_;
+
+- (instancetype)initWithTagAndAction:(NSInteger)tag action:(SEL)action {
+  if ((self = [super init])) {
+    tag_ = tag;
+    action_ = action;
+  }
+  return self;
+}
+
+@end
+
+@implementation MockCommandHandler {
+  int id_;
+}
+
+- (instancetype)initWithId:(int)id {
+  if ((self = [super init])) {
+    id_ = id;
+  }
+  return self;
+}
+
+- (BOOL)validateUserInterfaceItem:(id<NSValidatedUserInterfaceItem>)item
+                           window:(NSWindow*)window {
+  if (item.action == @selector(commandDispatch:))
+    return YES;
+  return NO;
+}
+
+- (void)commandDispatch:(id)sender window:(NSWindow*)window {
+  if ([sender respondsToSelector:@selector(setHandledById:)])
+    [sender setHandledById:id_];
+}
+
+- (void)commandDispatchUsingKeyModifiers:(id)sender window:(NSWindow*)window {
+  NOTIMPLEMENTED();
+}
+
+@end