Show media router toolbar icon ephemerally for active local routes and issues

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"
 
 class MediaRouterContextualMenuUnitTest : public BrowserWithTestWindowTest {
  public:
   MediaRouterContextualMenuUnitTest() {}
   ~MediaRouterContextualMenuUnitTest() override {}
 
   void SetUp() override {
     BrowserWithTestWindowTest::SetUp();
+
+    toolbar_actions_model_ =
+        extensions::extension_action_test_util::CreateToolbarModelForProfile(
+            profile());
+
     signin_manager_ =
         SigninManagerFactory::GetInstance()->GetForProfile(profile());
     browser_action_test_util_.reset(
         new BrowserActionTestUtil(browser(), false));
     action_.reset(new MediaRouterAction(browser(),
         browser_action_test_util_->GetToolbarActionsBar()));
     model_ = static_cast<ui::SimpleMenuModel*>(action_->GetContextMenu());
   }
 
   void TearDown() override {
     action_.reset();
     browser_action_test_util_.reset();
     BrowserWithTestWindowTest::TearDown();
   }
 
   SigninManagerBase* signin_manager() { return signin_manager_; }
   ui::SimpleMenuModel* model() { return model_; }
+  ToolbarActionsModel* toolbar_actions_model() {
+    return toolbar_actions_model_;
+  }
 
  private:
   std::unique_ptr<BrowserActionTestUtil> browser_action_test_util_;
   std::unique_ptr<MediaRouterAction> action_;
   SigninManagerBase* signin_manager_;
   ui::SimpleMenuModel* model_;
+  ToolbarActionsModel* toolbar_actions_model_;
 
   DISALLOW_COPY_AND_ASSIGN(MediaRouterContextualMenuUnitTest);
 };
 
 // Tests the basic state of the contextual menu.
 TEST_F(MediaRouterContextualMenuUnitTest, Basic) {
   int expected_number_items = 7;
 
 #if defined(OS_WIN) || defined(OS_MACOSX) || defined(OS_CHROMEOS)
   // On all platforms except Linux, there's an additional menu item to access
@@ skipping 54 matching lines @@
 
   menu.ExecuteCommand(IDC_MEDIA_ROUTER_CLOUD_SERVICES_TOGGLE, 0);
   EXPECT_TRUE(menu.IsCommandIdChecked(
       IDC_MEDIA_ROUTER_CLOUD_SERVICES_TOGGLE));
 
   menu.ExecuteCommand(IDC_MEDIA_ROUTER_CLOUD_SERVICES_TOGGLE, 0);
   EXPECT_FALSE(menu.IsCommandIdChecked(
       IDC_MEDIA_ROUTER_CLOUD_SERVICES_TOGGLE));
 }
 #endif  // GOOGLE_CHROME_BUILD
+
+TEST_F(MediaRouterContextualMenuUnitTest, ToggleAlwaysShowIconItem) {
+  MediaRouterContextualMenu menu(browser());
+  extensions::ComponentMigrationHelper* const component_migration_helper =
+      toolbar_actions_model()->component_migration_helper();
+
+  // Whether the option is checked should reflect the pref.
+  component_migration_helper->SetComponentActionPref(
+      ComponentToolbarActionsFactory::kMediaRouterActionId, true);
+  EXPECT_TRUE(
+      menu.IsCommandIdChecked(IDC_MEDIA_ROUTER_ALWAYS_SHOW_TOOLBAR_ACTION));
+
+  component_migration_helper->SetComponentActionPref(
+      ComponentToolbarActionsFactory::kMediaRouterActionId, false);
+  EXPECT_FALSE(
+      menu.IsCommandIdChecked(IDC_MEDIA_ROUTER_ALWAYS_SHOW_TOOLBAR_ACTION));
+
+  // 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));
+}