Add a policy to always show the Cast toolbar icon

chrome/browser/ui/toolbar/media_router_contextual_menu_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 <memory>
 
 #include "base/macros.h"
 #include "chrome/app/chrome_command_ids.h"
 #include "chrome/browser/extensions/browser_action_test_util.h"
 #include "chrome/browser/extensions/extension_action_test_util.h"
 #include "chrome/browser/signin/fake_signin_manager_builder.h"
 #include "chrome/browser/signin/signin_manager_factory.h"
 #include "chrome/browser/ui/toolbar/component_toolbar_actions_factory.h"
 #include "chrome/browser/ui/toolbar/media_router_action.h"
 #include "chrome/browser/ui/toolbar/media_router_contextual_menu.h"
 #include "chrome/test/base/browser_with_test_window_test.h"
 
+namespace {
+
+bool HasCommandId(ui::MenuModel* menu_model, int command_id) {
+  for (int i = 0; i < menu_model->GetItemCount(); i++) {
+    if (menu_model->GetCommandIdAt(i) == command_id)
+      return true;
+  }
+  return false;
+}
+
+}  // namespace
+
 class MediaRouterContextualMenuUnitTest : public BrowserWithTestWindowTest {
  public:
   MediaRouterContextualMenuUnitTest() {}
   ~MediaRouterContextualMenuUnitTest() override {}
 
   void SetUp() override {
     BrowserWithTestWindowTest::SetUp();
 
     toolbar_actions_model_ =
         extensions::extension_action_test_util::CreateToolbarModelForProfile(
@@ skipping 115 matching lines @@
 
   // Executing the option should toggle the pref.
   menu.ExecuteCommand(IDC_MEDIA_ROUTER_ALWAYS_SHOW_TOOLBAR_ACTION, 0);
   EXPECT_TRUE(component_migration_helper->GetComponentActionPref(
       ComponentToolbarActionsFactory::kMediaRouterActionId));
 
   menu.ExecuteCommand(IDC_MEDIA_ROUTER_ALWAYS_SHOW_TOOLBAR_ACTION, 0);
   EXPECT_FALSE(component_migration_helper->GetComponentActionPref(
       ComponentToolbarActionsFactory::kMediaRouterActionId));
 }
+
+TEST_F(MediaRouterContextualMenuUnitTest, ActionShownByPolicy) {
+  // Create a contextual menu for an icon shown by administrator policy.
+  MediaRouterContextualMenu menu(browser(), true);
+
+  // The item "Added by your administrator" should be shown disabled.
+  EXPECT_TRUE(menu.IsCommandIdVisible(IDC_MEDIA_ROUTER_SHOWN_BY_POLICY));
+  EXPECT_FALSE(menu.IsCommandIdEnabled(IDC_MEDIA_ROUTER_SHOWN_BY_POLICY));
+
+  // The checkbox item "Always show icon" should not be shown.
+  EXPECT_FALSE(HasCommandId(menu.menu_model(),
+                            IDC_MEDIA_ROUTER_ALWAYS_SHOW_TOOLBAR_ACTION));
+}