[Extensions UI] Handle multiple warning bubbles racing to show

chrome/browser/ui/cocoa/extensions/extension_message_bubble_browsertest_mac.mm

 // 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 "base/macros.h"
 #include "chrome/browser/ui/browser_window.h"
 #import "chrome/browser/ui/cocoa/browser_window_controller.h"
 #import "chrome/browser/ui/cocoa/extensions/browser_action_button.h"
 #import "chrome/browser/ui/cocoa/extensions/browser_actions_controller.h"
 #import "chrome/browser/ui/cocoa/extensions/toolbar_actions_bar_bubble_mac.h"
+#import "chrome/browser/ui/cocoa/run_loop_testing.h"
 #import "chrome/browser/ui/cocoa/toolbar/toolbar_controller.h"
 #include "chrome/browser/ui/extensions/extension_message_bubble_browsertest.h"
 #include "ui/base/cocoa/cocoa_base_utils.h"
+#include "ui/events/test/cocoa_test_event_utils.h"
 
 namespace {
 
 // Returns the ToolbarController for the given browser.
 ToolbarController* ToolbarControllerForBrowser(Browser* browser) {
   return [[BrowserWindowController browserWindowControllerForWindow:
              browser->window()->GetNativeWindow()] toolbarController];
 }
 
+ToolbarActionsBarBubbleMac* GetBubbleForBrowser(Browser* browser) {
+  ToolbarController* toolbarController = ToolbarControllerForBrowser(browser);
+  BrowserActionsController* actionsController =
+      [toolbarController browserActionsController];
+  return [actionsController activeBubble];
+}
+
+void ClickInView(NSView* view) {
+  ASSERT_TRUE(view);
+  std::pair<NSEvent*, NSEvent*> events =
+      cocoa_test_event_utils::MouseClickInView(view, 1);
+  [NSApp postEvent:events.second atStart:YES];
+  [NSApp sendEvent:events.first];
+  chrome::testing::NSRunLoopRunAllPending();
+}
+
 // Checks that the |bubble| is using the |expectedReferenceView|, and is in
 // approximately the correct position.
 void CheckBubbleAndReferenceView(ToolbarActionsBarBubbleMac* bubble,
                                  NSView* expectedReferenceView) {
   ASSERT_TRUE(bubble);
   ASSERT_TRUE(expectedReferenceView);
 
   // Do a rough check that the bubble is in the right place.
   // A window's frame (like the bubble's) is already in screen coordinates.
   NSRect bubbleFrame = [[bubble window] frame];
@@ skipping 33 matching lines @@
 
 }  // namespace
 
 class ExtensionMessageBubbleBrowserTestMac
     : public ExtensionMessageBubbleBrowserTest {
  public:
   ExtensionMessageBubbleBrowserTestMac() {}
   ~ExtensionMessageBubbleBrowserTestMac() override {}
 
  private:
+  // ExtensionMessageBubbleBrowserTest:
   void SetUpCommandLine(base::CommandLine* command_line) override;
-  void CheckBubble(Browser* browser, AnchorPosition anchor) override;
-  void CloseBubble(Browser* browser) override;
-  void CheckBubbleIsNotPresent(Browser* browser) override;
+  void CheckBubbleNative(Browser* browser, AnchorPosition anchor) override;
+  void CloseBubbleNative(Browser* browser) override;
+  void CheckBubbleIsNotPresentNative(Browser* browser) override;
+  void ClickLearnMoreButton(Browser* browser) override;
+  void ClickActionButton(Browser* browser) override;
+  void ClickDismissButton(Browser* browser) override;
 
   DISALLOW_COPY_AND_ASSIGN(ExtensionMessageBubbleBrowserTestMac);
 };
 
 void ExtensionMessageBubbleBrowserTestMac::SetUpCommandLine(
     base::CommandLine* command_line) {
   ExtensionMessageBubbleBrowserTest::SetUpCommandLine(command_line);
   [ToolbarActionsBarBubbleMac setAnimationEnabledForTesting:NO];
 }
 
-void ExtensionMessageBubbleBrowserTestMac::CheckBubble(
+void ExtensionMessageBubbleBrowserTestMac::CheckBubbleNative(
     Browser* browser,
     AnchorPosition anchor) {
   ToolbarController* toolbarController = ToolbarControllerForBrowser(browser);
   BrowserActionsController* actionsController =
       [toolbarController browserActionsController];
   NSView* anchorView = nil;
   ToolbarActionsBarBubbleMac* bubble = [actionsController activeBubble];
   switch (anchor) {
     case ANCHOR_BROWSER_ACTION:
       anchorView = [actionsController buttonWithIndex:0];
       break;
     case ANCHOR_APP_MENU:
       anchorView = [toolbarController appMenuButton];
       break;
   }
   CheckBubbleAndReferenceView(bubble, anchorView);
 }
 
-void ExtensionMessageBubbleBrowserTestMac::CloseBubble(Browser* browser) {
+void ExtensionMessageBubbleBrowserTestMac::CloseBubbleNative(Browser* browser) {
   BrowserActionsController* controller =
       [ToolbarControllerForBrowser(browser) browserActionsController];
   ToolbarActionsBarBubbleMac* bubble = [controller activeBubble];
   ASSERT_TRUE(bubble);
   [bubble close];
   EXPECT_EQ(nil, [controller activeBubble]);
 }
 
-void ExtensionMessageBubbleBrowserTestMac::CheckBubbleIsNotPresent(
+void ExtensionMessageBubbleBrowserTestMac::CheckBubbleIsNotPresentNative(
     Browser* browser) {
   EXPECT_EQ(
       nil,
       [[ToolbarControllerForBrowser(browser) browserActionsController]
           activeBubble]);
 }
 
+void ExtensionMessageBubbleBrowserTestMac::ClickLearnMoreButton(
+    Browser* browser) {
+  ToolbarActionsBarBubbleMac* bubble = GetBubbleForBrowser(browser);
+  ClickInView([bubble learnMoreButton]);
+}
+
+void ExtensionMessageBubbleBrowserTestMac::ClickActionButton(Browser* browser) {
+  ToolbarActionsBarBubbleMac* bubble = GetBubbleForBrowser(browser);
+  ClickInView([bubble actionButton]);
+}
+
+void ExtensionMessageBubbleBrowserTestMac::ClickDismissButton(
+    Browser* browser) {
+  ToolbarActionsBarBubbleMac* bubble = GetBubbleForBrowser(browser);
+  ClickInView([bubble dismissButton]);
+}
+
 IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleBrowserTestMac,
                        ExtensionBubbleAnchoredToExtensionAction) {
   TestBubbleAnchoredToExtensionAction();
 }
 
 IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleBrowserTestMac,
                        ExtensionBubbleAnchoredToAppMenu) {
   TestBubbleAnchoredToAppMenu();
 }
 
@@ skipping 14 matching lines @@
 
 IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleBrowserTestMac,
                        TestUninstallDangerousExtension) {
   TestUninstallDangerousExtension();
 }
 
 IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleBrowserTestMac,
                        TestDevModeBubbleIsntShownTwice) {
   TestDevModeBubbleIsntShownTwice();
 }
+
+IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleBrowserTestMac,
+                       TestBubbleWithMultipleWindows) {
+  TestBubbleWithMultipleWindows();
+}
+
+IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleBrowserTestMac,
+                       TestClickingLearnMoreButton) {
+  TestClickingLearnMoreButton();
+}
+
+IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleBrowserTestMac,
+                       TestClickingActionButton) {
+  TestClickingActionButton();
+}
+
+IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleBrowserTestMac,
+                       TestClickingDismissButton) {
+  TestClickingDismissButton();
+}