Add a policy to always show the Cast toolbar icon

chrome/browser/policy/policy_browsertest.cc

 // Copyright (c) 2012 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 <stddef.h>
 #include <stdint.h>
 
 #include <algorithm>
 #include <memory>
 #include <string>
@@ skipping 209 matching lines @@
 #if !defined(OS_MACOSX)
 #include "base/compiler_specific.h"
 #include "chrome/browser/ui/extensions/app_launch_params.h"
 #include "chrome/browser/ui/extensions/application_launch.h"
 #include "extensions/browser/app_window/app_window.h"
 #include "extensions/browser/app_window/app_window_registry.h"
 #include "extensions/browser/app_window/native_app_window.h"
 #include "ui/base/window_open_disposition.h"
 #endif
 
+#if defined(ENABLE_MEDIA_ROUTER)

[pastarmovj, 2017/01/24 23:03:46]

nit: you can combine the two if-defs in one.

[takumif, 2017/01/26 23:12:35]

Done.

+#if !defined(OS_ANDROID)
+#include "chrome/browser/ui/toolbar/component_toolbar_actions_factory.h"
+#include "chrome/browser/ui/toolbar/media_router_action_controller.h"
+#endif  // !defined(OS_ANDROID)
+#endif  // defined(ENABLE_MEDIA_ROUTER)
+
 using content::BrowserThread;
 using net::URLRequestMockHTTPJob;
 using testing::Mock;
 using testing::Return;
 using testing::_;
 
 namespace policy {
 
 namespace {
 
@@ skipping 3334 matching lines @@
 using MediaRouterEnabledPolicyTest = MediaRouterPolicyTest<true>;
 using MediaRouterDisabledPolicyTest = MediaRouterPolicyTest<false>;
 
 IN_PROC_BROWSER_TEST_F(MediaRouterEnabledPolicyTest, MediaRouterEnabled) {
   EXPECT_TRUE(media_router::MediaRouterEnabled(browser()->profile()));
 }
 
 IN_PROC_BROWSER_TEST_F(MediaRouterDisabledPolicyTest, MediaRouterDisabled) {
   EXPECT_FALSE(media_router::MediaRouterEnabled(browser()->profile()));
 }
+
+#if !defined(OS_ANDROID)
+template <bool enable>
+class MediaRouterActionPolicyTest : public PolicyTest {
+ public:
+  void SetUpInProcessBrowserTestFixture() override {
+    PolicyTest::SetUpInProcessBrowserTestFixture();
+    PolicyMap policies;
+    policies.Set(key::kShowCastIconInToolbar, POLICY_LEVEL_MANDATORY,
+                 POLICY_SCOPE_USER, POLICY_SOURCE_CLOUD,
+                 base::MakeUnique<base::FundamentalValue>(enable), nullptr);
+    provider_.UpdateChromePolicy(policies);
+  }
+
+ protected:
+  bool HasMediaRouterActionAtInit() const {
+    const std::set<std::string>& component_ids =
+        ComponentToolbarActionsFactory::GetInstance()->GetInitialComponentIds(
+            browser()->profile());
+    return base::ContainsKey(
+        component_ids, ComponentToolbarActionsFactory::kMediaRouterActionId);
+  }
+};
+
+using MediaRouterActionEnabledPolicyTest = MediaRouterActionPolicyTest<true>;
+using MediaRouterActionDisabledPolicyTest = MediaRouterActionPolicyTest<false>;
+
+IN_PROC_BROWSER_TEST_F(MediaRouterActionEnabledPolicyTest,
+                       MediaRouterActionEnabled) {
+  EXPECT_TRUE(
+      MediaRouterActionController::IsActionShownByPolicy(browser()->profile()));
+  EXPECT_TRUE(HasMediaRouterActionAtInit());
+}
+
+IN_PROC_BROWSER_TEST_F(MediaRouterActionDisabledPolicyTest,
+                       MediaRouterActionDisabled) {
+  EXPECT_FALSE(
+      MediaRouterActionController::IsActionShownByPolicy(browser()->profile()));
+  EXPECT_FALSE(HasMediaRouterActionAtInit());
+}
+#endif  // !defined(OS_ANDROID)
 #endif  // defined(ENABLE_MEDIA_ROUTER)
 
 #if BUILDFLAG(ENABLE_WEBRTC)
 // Sets the proper policy before the browser is started.
 template <bool enable>
 class WebRtcUdpPortRangePolicyTest : public PolicyTest {
  public:
   WebRtcUdpPortRangePolicyTest() = default;
   void SetUpInProcessBrowserTestFixture() override {
     PolicyTest::SetUpInProcessBrowserTestFixture();
@@ skipping 726 matching lines @@
 
   SetEmptyPolicy();
   // Policy not set.
   CheckSystemTimezoneAutomaticDetectionPolicyUnset();
   EXPECT_TRUE(CheckResolveTimezoneByGeolocation(true, false));
   EXPECT_TRUE(manager->TimeZoneResolverShouldBeRunningForTests());
 }
 #endif  // defined(OS_CHROMEOS)
 
 }  // namespace policy