[Extensions UI] Handle multiple warning bubbles racing to show

chrome/browser/ui/views/extensions/extension_message_bubble_view_browsertest.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 "base/macros.h"
 #include "chrome/browser/ui/extensions/extension_message_bubble_browsertest.h"
 #include "chrome/browser/ui/toolbar/toolbar_actions_bar.h"
 #include "chrome/browser/ui/views/frame/browser_view.h"
 #include "chrome/browser/ui/views/toolbar/app_menu_button.h"
 #include "chrome/browser/ui/views/toolbar/browser_actions_container.h"
+#include "chrome/browser/ui/views/toolbar/toolbar_actions_bar_bubble_views.h"
 #include "chrome/browser/ui/views/toolbar/toolbar_view.h"
 #include "ui/views/bubble/bubble_dialog_delegate.h"
+#include "ui/views/controls/link.h"
+#include "ui/views/controls/link_listener.h"
+#include "ui/views/window/dialog_client_view.h"
 
 namespace {
 
 // Returns the ToolbarView for the given |browser|.
 ToolbarView* GetToolbarViewForBrowser(Browser* browser) {
   return BrowserView::GetBrowserViewForBrowser(browser)->toolbar();
 }
 
+ToolbarActionsBarBubbleViews* GetBubbleForBrowser(Browser* browser) {
+  return static_cast<ToolbarActionsBarBubbleViews*>(
+      GetToolbarViewForBrowser(browser)->browser_actions()->active_bubble());
+}
+
 // Checks that the |bubble| is using the |expected_reference_view|, and is in
 // approximately the correct position.
 void CheckBubbleAndReferenceView(views::BubbleDialogDelegateView* bubble,
                                  views::View* expected_reference_view) {
   ASSERT_TRUE(bubble);
   ASSERT_TRUE(expected_reference_view);
   EXPECT_EQ(expected_reference_view, bubble->GetAnchorView());
 
   // Do a rough check that the bubble is in the right place.
   gfx::Rect bubble_bounds = bubble->GetWidget()->GetWindowBoundsInScreen();
@@ skipping 15 matching lines @@
 }  // namespace
 
 class ExtensionMessageBubbleViewBrowserTest
     : public ExtensionMessageBubbleBrowserTest {
  protected:
   ExtensionMessageBubbleViewBrowserTest() {}
   ~ExtensionMessageBubbleViewBrowserTest() override {}
 
  private:
   // ExtensionMessageBubbleBrowserTest:
-  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(ExtensionMessageBubbleViewBrowserTest);
 };
 
 class ExtensionMessageBubbleViewBrowserTestRedesign
     : public ExtensionMessageBubbleViewBrowserTest {
  protected:
   void SetUpCommandLine(base::CommandLine* command_line) override {
     ExtensionMessageBubbleViewBrowserTest::SetUpCommandLine(command_line);
     override_redesign_.reset();
   }
 };
 
-void ExtensionMessageBubbleViewBrowserTest::CheckBubble(Browser* browser,
-                                                        AnchorPosition anchor) {
+void ExtensionMessageBubbleViewBrowserTest::CheckBubbleNative(
+    Browser* browser,
+    AnchorPosition anchor) {
   ToolbarView* toolbar_view = GetToolbarViewForBrowser(browser);
   BrowserActionsContainer* container = toolbar_view->browser_actions();
   views::BubbleDialogDelegateView* bubble = container->active_bubble();
   views::View* anchor_view = nullptr;
   switch (anchor) {
     case ANCHOR_BROWSER_ACTION:
       DCHECK_GT(container->num_toolbar_actions(), 0u);
       anchor_view = container->GetToolbarActionViewAt(0);
       break;
     case ANCHOR_APP_MENU:
       anchor_view = toolbar_view->app_menu_button();
       break;
   }
   CheckBubbleAndReferenceView(bubble, anchor_view);
 }
 
-void ExtensionMessageBubbleViewBrowserTest::CloseBubble(Browser* browser) {
+void ExtensionMessageBubbleViewBrowserTest::CloseBubbleNative(
+    Browser* browser) {
   BrowserActionsContainer* container =
       GetToolbarViewForBrowser(browser)->browser_actions();
   views::BubbleDialogDelegateView* bubble = container->active_bubble();
   ASSERT_TRUE(bubble);
   bubble->GetWidget()->Close();
   EXPECT_EQ(nullptr, container->active_bubble());
 }
 
-void ExtensionMessageBubbleViewBrowserTest::CheckBubbleIsNotPresent(
+void ExtensionMessageBubbleViewBrowserTest::CheckBubbleIsNotPresentNative(
     Browser* browser) {
   EXPECT_EQ(
       nullptr,
       GetToolbarViewForBrowser(browser)->browser_actions()->active_bubble());
 }
 
+void ExtensionMessageBubbleViewBrowserTest::ClickLearnMoreButton(
+    Browser* browser) {
+  ToolbarActionsBarBubbleViews* bubble = GetBubbleForBrowser(browser);
+  static_cast<views::LinkListener*>(bubble)->LinkClicked(
+      const_cast<views::Link*>(bubble->learn_more_button()), 0);
+}
+
+void ExtensionMessageBubbleViewBrowserTest::ClickActionButton(
+    Browser* browser) {
+  ToolbarActionsBarBubbleViews* bubble = GetBubbleForBrowser(browser);
+  bubble->GetDialogClientView()->AcceptWindow();
+}
+
+void ExtensionMessageBubbleViewBrowserTest::ClickDismissButton(
+    Browser* browser) {
+  ToolbarActionsBarBubbleViews* bubble = GetBubbleForBrowser(browser);
+  bubble->GetDialogClientView()->CancelWindow();
+}
+
 IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleViewBrowserTest,
                        ExtensionBubbleAnchoredToExtensionAction) {
   TestBubbleAnchoredToExtensionAction();
 }
 
 IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleViewBrowserTest,
                        ExtensionBubbleAnchoredToAppMenu) {
   TestBubbleAnchoredToAppMenu();
 }
 
@@ skipping 33 matching lines @@
 IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleViewBrowserTestRedesign,
                        TestControlledHomeMessageBubble) {
   TestControlledHomeBubbleShown();
 }
 
 IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleViewBrowserTestRedesign,
                        TestControlledSearchMessageBubble) {
   TestControlledSearchBubbleShown();
 }
 #endif  // defined(OS_WIN)
+
+IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleViewBrowserTest,
+                       TestBubbleWithMultipleWindows) {
+  TestBubbleWithMultipleWindows();
+}
+
+IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleViewBrowserTest,
+                       TestClickingLearnMoreButton) {
+  TestClickingLearnMoreButton();
+}
+
+IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleViewBrowserTest,
+                       TestClickingActionButton) {
+  TestClickingActionButton();
+}
+
+IN_PROC_BROWSER_TEST_F(ExtensionMessageBubbleViewBrowserTest,
+                       TestClickingDismissButton) {
+  TestClickingDismissButton();
+}