Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/ui/toolbar/mock_component_toolbar_actions_factory.h" | 5 #include "chrome/browser/ui/toolbar/mock_component_toolbar_actions_factory.h" |
| 6 | 6 |
| 7 #include "chrome/browser/ui/browser.h" | 7 #include "chrome/browser/ui/browser.h" |
| 8 #include "chrome/browser/ui/toolbar/test_toolbar_action_view_controller.h" | 8 #include "chrome/browser/ui/toolbar/test_toolbar_action_view_controller.h" |
| 9 #include "chrome/browser/ui/toolbar/toolbar_action_view_controller.h" | 9 #include "chrome/browser/ui/toolbar/toolbar_action_view_controller.h" |
| 10 #include "extensions/common/feature_switch.h" | 10 #include "extensions/common/feature_switch.h" |
| 11 | 11 |
| 12 // static | 12 // static |
| 13 const char MockComponentToolbarActionsFactory::kActionIdForTesting[] = | 13 const char MockComponentToolbarActionsFactory::kActionIdForTesting[] = |
| 14 "mock_action"; | 14 "mock_action"; |
| 15 | 15 |
| 16 MockComponentToolbarActionsFactory::MockComponentToolbarActionsFactory( | 16 MockComponentToolbarActionsFactory::MockComponentToolbarActionsFactory( |
| 17 Browser* browser) | 17 Browser* browser) |
| 18 : migrated_feature_enabled_(false) { | 18 : migrated_feature_enabled_(false) { |
| 19 ComponentToolbarActionsFactory::SetTestingFactory(this); | 19 ComponentToolbarActionsFactory::SetTestingFactory(this); |
| 20 } | 20 } |
| 21 | 21 |
| 22 MockComponentToolbarActionsFactory::~MockComponentToolbarActionsFactory() { | 22 MockComponentToolbarActionsFactory::~MockComponentToolbarActionsFactory() { |
| 23 ComponentToolbarActionsFactory::SetTestingFactory(nullptr); | 23 ComponentToolbarActionsFactory::SetTestingFactory(nullptr); |
| 24 } | 24 } |
| 25 | 25 |
| 26 std::set<std::string> | 26 std::set<std::string> |
| 27 MockComponentToolbarActionsFactory::GetInitialComponentIds(Profile* profile) { | 27 MockComponentToolbarActionsFactory::GetInitialComponentIds(Profile* profile) { |
| 28 std::set<std::string> ids; | 28 std::set<std::string> ids; |
| 29 // kActionIdForTesting is installed by default if we are not testing | 29 if (extensions::FeatureSwitch::extension_action_redesign()->IsEnabled()) { |
|
msw
2017/02/08 02:14:02
nit: remove curlies
takumif
2017/02/08 03:35:41
Done.
| |
| 30 // a migration scenario. | |
| 31 if (extensions::FeatureSwitch::extension_action_redesign()->IsEnabled() && | |
| 32 migrated_extension_id_.empty()) { | |
| 33 ids.insert(kActionIdForTesting); | 30 ids.insert(kActionIdForTesting); |
| 34 } | 31 } |
| 35 return ids; | 32 return ids; |
| 36 } | 33 } |
| 37 | 34 |
| 38 std::unique_ptr<ToolbarActionViewController> | 35 std::unique_ptr<ToolbarActionViewController> |
| 39 MockComponentToolbarActionsFactory::GetComponentToolbarActionForId( | 36 MockComponentToolbarActionsFactory::GetComponentToolbarActionForId( |
| 40 const std::string& id, | 37 const std::string& id, |
| 41 Browser* browser, | 38 Browser* browser, |
| 42 ToolbarActionsBar* bar) { | 39 ToolbarActionsBar* bar) { |
| 43 DCHECK_EQ(kActionIdForTesting, id); | 40 DCHECK_EQ(kActionIdForTesting, id); |
| 44 return std::unique_ptr<ToolbarActionViewController>( | 41 return std::unique_ptr<ToolbarActionViewController>( |
|
msw
2017/02/08 02:14:02
optional nit: base::MakeUnique
takumif
2017/02/08 03:35:41
Done.
| |
| 45 new TestToolbarActionViewController( | 42 new TestToolbarActionViewController( |
| 46 MockComponentToolbarActionsFactory::kActionIdForTesting)); | 43 MockComponentToolbarActionsFactory::kActionIdForTesting)); |
| 47 } | 44 } |
| 48 | |
| 49 void MockComponentToolbarActionsFactory::RegisterComponentMigrations( | |
| 50 extensions::ComponentMigrationHelper* helper) const { | |
| 51 if (!migrated_extension_id_.empty()) { | |
| 52 helper->Register(kActionIdForTesting, migrated_extension_id_); | |
| 53 } | |
| 54 } | |
| 55 | |
| 56 void MockComponentToolbarActionsFactory::HandleComponentMigrations( | |
| 57 extensions::ComponentMigrationHelper* helper, | |
| 58 Profile* profile) const { | |
| 59 if (migrated_extension_id_.empty()) | |
| 60 return; | |
| 61 | |
| 62 if (migrated_feature_enabled_) { | |
| 63 helper->OnFeatureEnabled(kActionIdForTesting); | |
| 64 } else { | |
| 65 helper->OnFeatureDisabled(kActionIdForTesting); | |
| 66 } | |
| 67 } | |
| OLD | NEW |