Chromium Code Reviews| Index: chrome/browser/ui/toolbar/media_router_contextual_menu.h |
| diff --git a/chrome/browser/ui/toolbar/media_router_contextual_menu.h b/chrome/browser/ui/toolbar/media_router_contextual_menu.h |
| index 32bd1cac45ab204017e035887707173d9bc0896b..4433950a7bff2b79a4bdbe5fde3e4bb4c0a96636 100644 |
| --- a/chrome/browser/ui/toolbar/media_router_contextual_menu.h |
| +++ b/chrome/browser/ui/toolbar/media_router_contextual_menu.h |
| @@ -19,6 +19,9 @@ class ComponentMigrationHelper; |
| class MediaRouterContextualMenu : public ui::SimpleMenuModel::Delegate { |
| public: |
| explicit MediaRouterContextualMenu(Browser* browser); |
| + |
| + // Constructor for injecting values in tests. |
| + MediaRouterContextualMenu(Browser* browser, bool shown_by_policy); |
|
msw
2017/01/05 01:14:57
Can the test mark the preference as managed (polic
takumif
2017/01/06 22:06:57
That'd require moving the test to MediaRouterActio
msw
2017/01/06 23:01:23
Hmm, okay; this ctor flag seems fine to use a unit
|
| ~MediaRouterContextualMenu() override; |
| ui::MenuModel* menu_model() { return &menu_model_; } |
| @@ -28,6 +31,8 @@ class MediaRouterContextualMenu : public ui::SimpleMenuModel::Delegate { |
| ToggleCloudServicesItem); |
| FRIEND_TEST_ALL_PREFIXES(MediaRouterContextualMenuUnitTest, |
| ToggleAlwaysShowIconItem); |
| + FRIEND_TEST_ALL_PREFIXES(MediaRouterContextualMenuUnitTest, |
| + ActionShownByPolicy); |
| // Gets or sets the "Always show icon" option. |
| bool GetAlwaysShowActionPref() const; |
| @@ -45,6 +50,9 @@ class MediaRouterContextualMenu : public ui::SimpleMenuModel::Delegate { |
| ui::SimpleMenuModel menu_model_; |
| extensions::ComponentMigrationHelper* const component_migration_helper_; |
| + // Whether the media router action is shown by an administrator policy. |
| + bool shown_by_policy_; |
| + |
| DISALLOW_COPY_AND_ASSIGN(MediaRouterContextualMenu); |
| }; |