Show Media Router toolbar icon ephemerally for MR dialogs

chrome/browser/ui/webui/media_router/media_router_web_ui_test.cc

+// Copyright 2015 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/webui/media_router/media_router_web_ui_test.h"
+
+#include "chrome/browser/media/router/media_router_factory.h"
+#include "chrome/browser/media/router/media_router_ui_service_factory.h"
+#include "chrome/browser/media/router/test_helper.h"
+#include "chrome/browser/ui/toolbar/toolbar_actions_model.h"
+#include "chrome/browser/ui/toolbar/toolbar_actions_model_factory.h"
+#include "chrome/test/base/dialog_test_browser_window.h"
+
+std::unique_ptr<KeyedService> BuildMockMediaRouterUIService(
+    content::BrowserContext* context) {
+  return base::MakeUnique<MockMediaRouterUIService>(
+      static_cast<Profile*>(context));
+}
+
+std::unique_ptr<KeyedService> BuildToolbarActionsModel(
+    content::BrowserContext* context) {
+  return base::MakeUnique<ToolbarActionsModel>(static_cast<Profile*>(context),
+                                               nullptr);
+}
+
+MockMediaRouterActionController::MockMediaRouterActionController(
+    Profile* profile)
+    : MediaRouterActionController(
+          profile,
+          media_router::MediaRouterFactory::GetApiForBrowserContext(profile),
+          nullptr,
+          nullptr) {}
+MockMediaRouterActionController::~MockMediaRouterActionController() {}
+
+MockMediaRouterUIService::MockMediaRouterUIService(Profile* profile)
+    : media_router::MediaRouterUIService(profile),
+      action_controller_(
+          base::MakeUnique<MockMediaRouterActionController>(profile)) {}
+MockMediaRouterUIService::~MockMediaRouterUIService() {}
+
+MediaRouterActionController* MockMediaRouterUIService::action_controller()
+    const {
+  return action_controller_.get();
+}
+
+MediaRouterWebUITest::MediaRouterWebUITest() : MediaRouterWebUITest(false) {}
+MediaRouterWebUITest::MediaRouterWebUITest(bool require_mock_ui_service)
+    : feature_override_(extensions::FeatureSwitch::media_router(), true),
+      require_mock_ui_service_(require_mock_ui_service) {}
+
+MediaRouterWebUITest::~MediaRouterWebUITest() {}
+
+TestingProfile* MediaRouterWebUITest::CreateProfile() {
+  TestingProfile::Builder builder;
+  if (require_mock_ui_service_) {
+    builder.AddTestingFactory(
+        media_router::MediaRouterUIServiceFactory::GetInstance(),
+        BuildMockMediaRouterUIService);
+    builder.AddTestingFactory(ToolbarActionsModelFactory::GetInstance(),
+                              BuildToolbarActionsModel);
+  }
+  return builder.Build().release();
+}
+
+BrowserWindow* MediaRouterWebUITest::CreateBrowserWindow() {
+  return new DialogTestBrowserWindow;
+}