Add a policy to always show the Cast toolbar icon

chrome/browser/policy/policy_browsertest.cc

Index: chrome/browser/policy/policy_browsertest.cc
diff --git a/chrome/browser/policy/policy_browsertest.cc b/chrome/browser/policy/policy_browsertest.cc
index a284e2ed2a39f02e410e86e6b2be98992a305c85..ec88ea2269953a172bc23a2de081398f3a1a2328 100644
--- a/chrome/browser/policy/policy_browsertest.cc
+++ b/chrome/browser/policy/policy_browsertest.cc
@@ -229,6 +229,11 @@
 #include "ui/base/window_open_disposition.h"
 #endif
 
+#if !defined(OS_ANDROID)

[imcheng, 2016/12/15 19:57:38]

#if defined(ENABLE_MEDIA_ROUTER)

[takumif, 2016/12/15 22:04:14]

Done.

+#include "chrome/browser/ui/toolbar/component_toolbar_actions_factory.h"
+#include "chrome/browser/ui/toolbar/media_router_action_controller.h"
+#endif
+
 using content::BrowserThread;
 using net::URLRequestMockHTTPJob;
 using testing::Mock;
@@ -3485,6 +3490,47 @@ IN_PROC_BROWSER_TEST_F(MediaRouterEnabledPolicyTest, MediaRouterEnabled) {
 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)