Dismiss context menu when download bar is closed.

chrome/browser/ui/cocoa/download/download_item_controller_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 <Cocoa/Cocoa.h>
 
 #import "base/mac/scoped_nsobject.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/run_loop.h"
 #include "chrome/browser/ui/cocoa/cocoa_profile_test.h"
+#import "chrome/browser/ui/cocoa/download/download_item_button.h"
 #import "chrome/browser/ui/cocoa/download/download_item_controller.h"
 #import "chrome/browser/ui/cocoa/download/download_shelf_controller.h"
 #include "content/public/test/mock_download_item.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/platform_test.h"
 #import "third_party/ocmock/OCMock/OCMock.h"
 #import "third_party/ocmock/gtest_support.h"
 
 using ::testing::Return;
 using ::testing::ReturnRefOfCopy;
@@ skipping 16 matching lines @@
 @interface DownloadItemControllerWithInitCallback : DownloadItemController {
  @private
   base::Closure initCallback_;
 }
 - (id)initWithDownload:(content::DownloadItem*)downloadItem
                  shelf:(DownloadShelfController*)shelf
           initCallback:(const base::Closure&)callback;
 - (void)awakeFromNib;
 @end
 
+@interface DownloadItemButton(DownloadItemButtonTest)
+
+- (BOOL)showingContextMenu;
+
+@end
+
+
 @implementation DownloadItemControllerWithInitCallback
 
 - (id)initWithDownload:(content::DownloadItem*)downloadItem
                  shelf:(DownloadShelfController*)shelf
           initCallback:(const base::Closure&)callback {
   if ((self = [super initWithDownload:downloadItem shelf:shelf navigator:NULL]))
     initCallback_ = callback;
   return self;
 }
 
@@ skipping 115 matching lines @@
   ON_CALL(*download_item_, GetDangerType())
       .WillByDefault(Return(content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS));
   ON_CALL(*download_item_, IsDangerous()).WillByDefault(Return(false));
   download_item_->NotifyObserversDownloadUpdated();
 
   [item verifyProgressViewIsVisible:true];
   [item verifyDangerousDownloadPromptIsVisible:false];
   [(id)shelf_ verify];
 }
 
+TEST_F(DownloadItemControllerTest, DismissesContextMenuWhenRemovedFromWindow) {
+  base::scoped_nsobject<DownloadItemController> item(CreateItemController());
+  DownloadItemButton* downloadItemButton = nil;
+  for (NSView *nextSubview in [[item view] subviews]) {
+    if ([nextSubview isKindOfClass:[DownloadItemButton class]]) {
+      downloadItemButton = static_cast<DownloadItemButton *>(nextSubview);
+      break;
+    }
+  }
+  // showContextMenu: calls [NSMenu popUpContextMenu:...], which blocks until
+  // the menu is dismissed. Use a block to cancel the menu while waiting for
+  // [NSMenu popUpContextMenu:...] to return.
+  [[NSOperationQueue mainQueue] addOperationWithBlock:^{
+    // Wait a bit to make sure [item showContextMenu:nil] is executing before
+    // proceeding.
+    [NSThread sleepForTimeInterval:0.25];
+    EXPECT_TRUE([downloadItemButton showingContextMenu]);
+    [downloadItemButton
+        performSelectorOnMainThread:@selector(removeFromSuperview)
+                         withObject:nil
+                      waitUntilDone:NO];
+  }];
+  [item showContextMenu:nil];
+  EXPECT_FALSE([downloadItemButton showingContextMenu]);
+}
+
+
 }  // namespace