| 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 "base/memory/ptr_util.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 Profile* profile) |
| 18 ComponentToolbarActionsFactory::SetTestingFactory(this); | 18 : ComponentToolbarActionsFactory(profile) {} |
| 19 } | |
| 20 | 19 |
| 21 MockComponentToolbarActionsFactory::~MockComponentToolbarActionsFactory() { | 20 MockComponentToolbarActionsFactory::~MockComponentToolbarActionsFactory() {} |
| 22 ComponentToolbarActionsFactory::SetTestingFactory(nullptr); | |
| 23 } | |
| 24 | 21 |
| 25 std::set<std::string> | 22 std::set<std::string> |
| 26 MockComponentToolbarActionsFactory::GetInitialComponentIds(Profile* profile) { | 23 MockComponentToolbarActionsFactory::GetInitialComponentIds() { |
| 27 std::set<std::string> ids; | 24 std::set<std::string> ids; |
| 28 if (extensions::FeatureSwitch::extension_action_redesign()->IsEnabled()) | 25 if (extensions::FeatureSwitch::extension_action_redesign()->IsEnabled()) |
| 29 ids.insert(kActionIdForTesting); | 26 ids.insert(kActionIdForTesting); |
| 30 return ids; | 27 return ids; |
| 31 } | 28 } |
| 32 | 29 |
| 33 std::unique_ptr<ToolbarActionViewController> | 30 std::unique_ptr<ToolbarActionViewController> |
| 34 MockComponentToolbarActionsFactory::GetComponentToolbarActionForId( | 31 MockComponentToolbarActionsFactory::GetComponentToolbarActionForId( |
| 35 const std::string& id, | 32 const std::string& id, |
| 36 Browser* browser, | 33 Browser* browser, |
| 37 ToolbarActionsBar* bar) { | 34 ToolbarActionsBar* bar) { |
| 38 DCHECK_EQ(kActionIdForTesting, id); | 35 DCHECK_EQ(kActionIdForTesting, id); |
| 39 return base::MakeUnique<TestToolbarActionViewController>( | 36 return base::MakeUnique<TestToolbarActionViewController>( |
| 40 MockComponentToolbarActionsFactory::kActionIdForTesting); | 37 MockComponentToolbarActionsFactory::kActionIdForTesting); |
| 41 } | 38 } |
| OLD | NEW |