Clicking tab close with option key close the other tabs.

chrome/browser/ui/cocoa/tabs/tab_controller_unittest.mm

 // Copyright (c) 2011 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 <Cocoa/Cocoa.h>
 
 #import "base/memory/scoped_nsobject.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/ui/cocoa/cocoa_test_helper.h"
 #import "chrome/browser/ui/cocoa/tabs/tab_controller.h"
 #import "chrome/browser/ui/cocoa/tabs/tab_controller_target.h"
 #import "chrome/browser/ui/cocoa/tabs/tab_strip_drag_controller.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #import "testing/gtest_mac.h"
 #include "testing/platform_test.h"
+#import "third_party/ocmock/OCMock/OCMock.h"
 
 // Implements the target interface for the tab, which gets sent messages when
 // the tab is clicked on by the user and when its close box is clicked.
 @interface TabControllerTestTarget : NSObject<TabControllerTarget> {
  @private
   bool selected_;
   bool closed_;
+  bool closedOtherTabs_;
   scoped_nsobject<TabStripDragController> dragController_;
 }
 - (bool)selected;
 - (bool)closed;
+- (bool)closedOtherTabs;
 @end
 
 @implementation TabControllerTestTarget
 - (id)init {
   if ((self = [super init])) {
     dragController_.reset(
         [[TabStripDragController alloc] initWithTabStripController:nil]);
   }
   return self;
 }
 - (bool)selected {
   return selected_;
 }
 - (bool)closed {
   return closed_;
 }
+- (bool)closedOtherTabs {
+  return closedOtherTabs_;
+}
 - (void)selectTab:(id)sender {
   selected_ = true;
 }
 - (void)closeTab:(id)sender {
   closed_ = true;
 }
+- (void)closeOtherTabs:(id)sender {
+  closedOtherTabs_ = true;
+}
 - (void)mouseTimer:(NSTimer*)timer {
   // Fire the mouseUp to break the TabView drag loop.
   NSEvent* current = [NSApp currentEvent];
   NSWindow* window = [timer userInfo];
   NSEvent* up = [NSEvent mouseEventWithType:NSLeftMouseUp
                                    location:[current locationInWindow]
-                              modifierFlags:0
+                              modifierFlags:[current modifierFlags]
                                   timestamp:[current timestamp]
                                windowNumber:[window windowNumber]
                                     context:nil
                                 eventNumber:0
                                  clickCount:1
                                    pressure:1.0];
   [window postEvent:up atStart:YES];
 }
 - (void)commandDispatch:(TabStripModel::ContextMenuCommand)command
           forController:(TabController*)controller {
@@ skipping 55 matching lines @@
   EXPECT_FALSE([target closed]);
   [controller setTarget:target];
   EXPECT_EQ(target.get(), [controller target]);
 
   [controller closeTab:nil];
   EXPECT_TRUE([target closed]);
 
   [[controller view] removeFromSuperview];
 }
 
+// Tests sending it a closeTab message with a mousedown event and ensuring
+// that the target/action get called. closeOtherTabs message should be called
+// when the tab is clicked with option key.
+TEST_F(TabControllerTest, CloseOtherTabs) {
+  NSWindow* window = test_window();
+  scoped_nsobject<TabController> controller([[TabController alloc] init]);
+  [[window contentView] addSubview:[controller view]];
+
+  scoped_nsobject<TabControllerTestTarget> target(
+      [[TabControllerTestTarget alloc] init]);
+  EXPECT_FALSE([target closedOtherTabs]);
+  [controller setTarget:target];
+  [controller setAction:@selector(closeTab:)];
+  EXPECT_EQ(target.get(), [controller target]);
+  EXPECT_EQ(@selector(closeTab:), [controller action]);
+
+  // Create a mouse event with |NSAlternateKeyMask| and mock NSApp so that
+  // [NSApp currentEvent] returns the event.
+  NSEvent* current = [NSApp currentEvent];
+  NSEvent* down = [NSEvent mouseEventWithType:NSLeftMouseDown
+                                     location:[current locationInWindow]
+                                modifierFlags:NSAlternateKeyMask
+                                    timestamp:[current timestamp]
+                                 windowNumber:[window windowNumber]
+                                      context:nil
+                                  eventNumber:0
+                                   clickCount:1
+                                     pressure:1.0];
+  // OCMock will swap NSApp internally and will restore it when the mock gets
+  // deallocated. This happens when the autorelease pool drains at the end of
+  // the test.
+  id fakeApp = [OCMockObject partialMockForObject:NSApp];
+  [[[fakeApp stub] andReturn:down] currentEvent];
+
+  [controller closeTab:nil];
+  EXPECT_TRUE([target closedOtherTabs]);
+  EXPECT_FALSE([target closed]);
+  [[controller view] removeFromSuperview];
+}
+
 // Tests setting the |selected| property via code.
 TEST_F(TabControllerTest, APISelection) {
   NSWindow* window = test_window();
   scoped_nsobject<TabController> controller([[TabController alloc] init]);
   [[window contentView] addSubview:[controller view]];
 
   EXPECT_FALSE([controller selected]);
   [controller setSelected:YES];
   EXPECT_TRUE([controller selected]);
 
@@ skipping 208 matching lines @@
             NSWidth([[controller titleView] frame]));
   EXPECT_EQ(LeftMargin(originalTabFrame, originalTitleFrame),
             LeftMargin([[controller view] frame],
                        [[controller titleView] frame]));
   EXPECT_EQ(RightMargin(originalTabFrame, originalTitleFrame),
             RightMargin([[controller view] frame],
                         [[controller titleView] frame]));
 }
 
 }  // namespace