Improve settings override bubble to indicate policy installed extensions

chrome/browser/ui/views/toolbar/toolbar_actions_bar_bubble_views_unittest.cc

 // Copyright 2016 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 "chrome/browser/ui/views/toolbar/toolbar_actions_bar_bubble_views.h"
 
 #include <memory>
 
 #include "base/run_loop.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/ui/toolbar/test_toolbar_actions_bar_bubble_delegate.h"
+#include "chrome/browser/ui/toolbar/toolbar_actions_bar_bubble_delegate.h"
+#include "chrome/grit/generated_resources.h"
+#include "components/grit/components_scaled_resources.h"
+#include "ui/base/l10n/l10n_util.h"
 #include "ui/events/event_utils.h"
 #include "ui/events/test/event_generator.h"
 #include "ui/views/controls/button/label_button.h"
 #include "ui/views/controls/link.h"
 #include "ui/views/test/test_widget_observer.h"
 #include "ui/views/test/views_test_base.h"
 #include "ui/views/widget/widget.h"
 #include "ui/views/window/dialog_client_view.h"
 
 class ToolbarActionsBarBubbleViewsTest : public views::ViewsTestBase {
+ public:
+  views::View* TestCreateExtraView() {
+    return (bubble_) ? bubble_->CreateExtraView() : nullptr;
+  }
+
  protected:
   ToolbarActionsBarBubbleViewsTest() {}
   ~ToolbarActionsBarBubbleViewsTest() override {}
 
   void TearDown() override {
     anchor_widget_.reset();
     views::ViewsTestBase::TearDown();
   }
 
   std::unique_ptr<views::Widget> CreateAnchorWidget() {
@@ skipping 86 matching lines @@
   EXPECT_FALSE(bubble()->item_list());
 
   CloseBubble();
 }
 
 TEST_F(ToolbarActionsBarBubbleViewsTest,
        TestBubbleLayoutActionDismissAndLearnMoreButton) {
   TestToolbarActionsBarBubbleDelegate delegate(HeadingString(), BodyString(),
                                                ActionString());
   delegate.set_dismiss_button_text(DismissString());
-  delegate.set_learn_more_button_text(LearnMoreString());
+
+  ToolbarActionsBarBubbleDelegate::ExtraViewInfo extra_view_info_linked_text;
+
+  extra_view_info_linked_text.text = LearnMoreString();
+  extra_view_info_linked_text.is_text_linked = true;
+  delegate.set_extra_view_info(extra_view_info_linked_text);
+
   ShowBubble(&delegate);
 
   EXPECT_TRUE(bubble()->GetDialogClientView()->ok_button());
   EXPECT_EQ(ActionString(),
             bubble()->GetDialogClientView()->ok_button()->GetText());
   EXPECT_TRUE(bubble()->GetDialogClientView()->cancel_button());
   EXPECT_EQ(DismissString(),
             bubble()->GetDialogClientView()->cancel_button()->GetText());
   EXPECT_TRUE(bubble()->learn_more_button());
   EXPECT_EQ(LearnMoreString(), bubble()->learn_more_button()->text());
@@ skipping 114 matching lines @@
   ShowBubble(&delegate);
   views::test::TestWidgetObserver bubble_observer(bubble_widget());
 
   EXPECT_FALSE(delegate.close_action());
   // Activate another widget. The bubble shouldn't close.
   anchor_widget()->Activate();
   base::RunLoop().RunUntilIdle();
   EXPECT_FALSE(delegate.close_action());
   CloseBubble();
 }
+
+TEST_F(ToolbarActionsBarBubbleViewsTest, TestCreateExtraViewEmpty) {
+  TestToolbarActionsBarBubbleDelegate delegate(HeadingString(), BodyString(),
+                                               ActionString());
+  ToolbarActionsBarBubbleDelegate::ExtraViewInfo extra_view_info;

[Devlin, 2016/09/26 20:13:10]

Don't we need to set this on the delegate?

[catmullings, 2016/10/06 18:24:19]

Done.

+
+  ShowBubble(&delegate);
+
+  views::View* extra_view = TestCreateExtraView();
+  EXPECT_FALSE(extra_view);
+  CloseBubble();
+}
+
+TEST_F(ToolbarActionsBarBubbleViewsTest, TestCreateExtraViewIconOnly) {
+  TestToolbarActionsBarBubbleDelegate delegate(HeadingString(), BodyString(),
+                                               ActionString());
+  ToolbarActionsBarBubbleDelegate::ExtraViewInfo extra_view_info;
+
+  extra_view_info.resource_id = IDR_OMNIBOX_HTTPS_POLICY_WARNING;
+  delegate.set_extra_view_info(extra_view_info);
+  ShowBubble(&delegate);
+  std::unique_ptr<views::View> extra_view(TestCreateExtraView());
+  EXPECT_TRUE(extra_view);
+  EXPECT_EQ("ImageView", std::string(extra_view->GetClassName()));
+  CloseBubble();
+}
+
+TEST_F(ToolbarActionsBarBubbleViewsTest, TestCreateExtraViewLinkedTextOnly) {
+  TestToolbarActionsBarBubbleDelegate delegate(HeadingString(), BodyString(),
+                                               ActionString());
+  ToolbarActionsBarBubbleDelegate::ExtraViewInfo extra_view_info_linked_text;
+
+  extra_view_info_linked_text.text =
+      l10n_util::GetStringUTF16(IDS_EXTENSIONS_INSTALLED_BY_ADMIN);
+  extra_view_info_linked_text.is_text_linked = true;
+  delegate.set_extra_view_info(extra_view_info_linked_text);
+
+  ShowBubble(&delegate);
+
+  views::View* extra_view = TestCreateExtraView();
+  EXPECT_TRUE(extra_view);
+  EXPECT_EQ("Link", std::string(extra_view->GetClassName()));
+  CloseBubble();
+}
+
+TEST_F(ToolbarActionsBarBubbleViewsTest, TestCreateExtraViewLabelTextOnly) {
+  TestToolbarActionsBarBubbleDelegate delegate(HeadingString(), BodyString(),
+                                               ActionString());
+  ToolbarActionsBarBubbleDelegate::ExtraViewInfo extra_view_info;
+
+  extra_view_info.text =
+      l10n_util::GetStringUTF16(IDS_EXTENSIONS_INSTALLED_BY_ADMIN);
+  extra_view_info.is_text_linked = false;
+  delegate.set_extra_view_info(extra_view_info);
+
+  ShowBubble(&delegate);
+
+  views::View* extra_view = TestCreateExtraView();
+  EXPECT_TRUE(extra_view);
+  EXPECT_EQ("Label", std::string(extra_view->GetClassName()));
+  CloseBubble();
+}
+
+TEST_F(ToolbarActionsBarBubbleViewsTest, TestCreateExtraViewImageAndText) {
+  TestToolbarActionsBarBubbleDelegate delegate(HeadingString(), BodyString(),
+                                               ActionString());
+  ToolbarActionsBarBubbleDelegate::ExtraViewInfo extra_view_info;
+  extra_view_info.resource_id = IDR_OMNIBOX_HTTPS_POLICY_WARNING;
+  extra_view_info.text =
+      l10n_util::GetStringUTF16(IDS_EXTENSIONS_INSTALLED_BY_ADMIN);
+  extra_view_info.is_text_linked = false;
+  delegate.set_extra_view_info(extra_view_info);
+
+  ShowBubble(&delegate);
+
+  views::View* extra_view = TestCreateExtraView();
+  EXPECT_TRUE(extra_view);
+  EXPECT_EQ("View", std::string(extra_view->GetClassName()));
+  EXPECT_EQ(2, extra_view->child_count());
+  CloseBubble();
+}