[Extensions UI] Fix a crash when clicking on a bubble's learn more

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 18 matching lines @@
     : public ExtensionMessageBubbleBrowserTest {
  protected:
   ExtensionMessageBubbleViewBrowserTest() {}
   ~ExtensionMessageBubbleViewBrowserTest() override {}
 
  private:
   // ExtensionMessageBubbleBrowserTest:
   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();
@@ skipping 29 matching lines @@
   EXPECT_EQ(nullptr, container->active_bubble());
 }
 
 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 38 matching lines @@
 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();
+}