| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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/website_settings/mock_permission_bubble_factory.h" | 5 #include "chrome/browser/ui/website_settings/mock_permission_bubble_factory.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/memory/ptr_util.h" |
| 8 #include "chrome/browser/ui/website_settings/mock_permission_bubble_view.h" | 9 #include "chrome/browser/ui/website_settings/mock_permission_bubble_view.h" |
| 9 #include "chrome/browser/ui/website_settings/permission_bubble_manager.h" | 10 #include "chrome/browser/ui/website_settings/permission_bubble_manager.h" |
| 10 | 11 |
| 11 MockPermissionBubbleFactory::MockPermissionBubbleFactory( | 12 MockPermissionBubbleFactory::MockPermissionBubbleFactory( |
| 12 bool is_browser_test, | 13 bool is_browser_test, |
| 13 PermissionBubbleManager* manager) | 14 PermissionBubbleManager* manager) |
| 14 : is_browser_test_(is_browser_test), | 15 : is_browser_test_(is_browser_test), |
| 15 can_update_ui_(false), | 16 can_update_ui_(false), |
| 16 show_count_(0), | 17 show_count_(0), |
| 17 requests_count_(0), | 18 requests_count_(0), |
| 18 total_requests_count_(0), | 19 total_requests_count_(0), |
| 19 response_type_(PermissionBubbleManager::NONE), | 20 response_type_(PermissionBubbleManager::NONE), |
| 20 manager_(manager) { | 21 manager_(manager) { |
| 21 manager->view_factory_ = | 22 manager->view_factory_ = |
| 22 base::Bind(&MockPermissionBubbleFactory::Create, base::Unretained(this)); | 23 base::Bind(&MockPermissionBubbleFactory::Create, base::Unretained(this)); |
| 23 } | 24 } |
| 24 | 25 |
| 25 MockPermissionBubbleFactory::~MockPermissionBubbleFactory() { | 26 MockPermissionBubbleFactory::~MockPermissionBubbleFactory() { |
| 26 manager_->view_factory_ = | 27 manager_->view_factory_ = |
| 27 base::Bind(&MockPermissionBubbleFactory::DoNotCreate); | 28 base::Bind(&MockPermissionBubbleFactory::DoNotCreate); |
| 28 while (!views_.empty()) { | 29 while (!views_.empty()) { |
| 29 views_.back()->factory_ = nullptr; | 30 views_.back()->factory_ = nullptr; |
| 30 views_.pop_back(); | 31 views_.pop_back(); |
| 31 } | 32 } |
| 32 } | 33 } |
| 33 | 34 |
| 34 scoped_ptr<PermissionBubbleView> MockPermissionBubbleFactory::Create( | 35 std::unique_ptr<PermissionBubbleView> MockPermissionBubbleFactory::Create( |
| 35 Browser* browser) { | 36 Browser* browser) { |
| 36 MockPermissionBubbleView* view = | 37 MockPermissionBubbleView* view = |
| 37 new MockPermissionBubbleView(this, manager_, is_browser_test_); | 38 new MockPermissionBubbleView(this, manager_, is_browser_test_); |
| 38 view->can_update_ui_ = can_update_ui_; | 39 view->can_update_ui_ = can_update_ui_; |
| 39 return make_scoped_ptr(view); | 40 return base::WrapUnique(view); |
| 40 } | 41 } |
| 41 | 42 |
| 42 void MockPermissionBubbleFactory::SetCanUpdateUi(bool can_update_ui) { | 43 void MockPermissionBubbleFactory::SetCanUpdateUi(bool can_update_ui) { |
| 43 can_update_ui_ = can_update_ui; | 44 can_update_ui_ = can_update_ui; |
| 44 for (auto* view : views_) | 45 for (auto* view : views_) |
| 45 view->can_update_ui_ = can_update_ui_; | 46 view->can_update_ui_ = can_update_ui_; |
| 46 } | 47 } |
| 47 | 48 |
| 48 void MockPermissionBubbleFactory::ResetCounts() { | 49 void MockPermissionBubbleFactory::ResetCounts() { |
| 49 show_count_ = 0; | 50 show_count_ = 0; |
| 50 requests_count_ = 0; | 51 requests_count_ = 0; |
| 51 total_requests_count_ = 0; | 52 total_requests_count_ = 0; |
| 52 } | 53 } |
| 53 | 54 |
| 54 void MockPermissionBubbleFactory::DocumentOnLoadCompletedInMainFrame() { | 55 void MockPermissionBubbleFactory::DocumentOnLoadCompletedInMainFrame() { |
| 55 manager_->DocumentOnLoadCompletedInMainFrame(); | 56 manager_->DocumentOnLoadCompletedInMainFrame(); |
| 56 } | 57 } |
| 57 | 58 |
| 58 bool MockPermissionBubbleFactory::is_visible() { | 59 bool MockPermissionBubbleFactory::is_visible() { |
| 59 for (auto* view : views_) { | 60 for (auto* view : views_) { |
| 60 if (view->IsVisible()) | 61 if (view->IsVisible()) |
| 61 return true; | 62 return true; |
| 62 } | 63 } |
| 63 return false; | 64 return false; |
| 64 } | 65 } |
| 65 | 66 |
| 66 // static | 67 // static |
| 67 scoped_ptr<PermissionBubbleView> MockPermissionBubbleFactory::DoNotCreate( | 68 std::unique_ptr<PermissionBubbleView> MockPermissionBubbleFactory::DoNotCreate( |
| 68 Browser* browser) { | 69 Browser* browser) { |
| 69 NOTREACHED(); | 70 NOTREACHED(); |
| 70 return make_scoped_ptr(new MockPermissionBubbleView(nullptr, nullptr, false)); | 71 return base::WrapUnique( |
| 72 new MockPermissionBubbleView(nullptr, nullptr, false)); |
| 71 } | 73 } |
| 72 | 74 |
| 73 void MockPermissionBubbleFactory::UpdateResponseType() { | 75 void MockPermissionBubbleFactory::UpdateResponseType() { |
| 74 manager_->set_auto_response_for_test(response_type_); | 76 manager_->set_auto_response_for_test(response_type_); |
| 75 } | 77 } |
| 76 | 78 |
| 77 void MockPermissionBubbleFactory::ShowView(MockPermissionBubbleView* view) { | 79 void MockPermissionBubbleFactory::ShowView(MockPermissionBubbleView* view) { |
| 78 for (std::vector<MockPermissionBubbleView*>::iterator iter = views_.begin(); | 80 for (std::vector<MockPermissionBubbleView*>::iterator iter = views_.begin(); |
| 79 iter != views_.end(); ++iter) { | 81 iter != views_.end(); ++iter) { |
| 80 if (*iter == view) | 82 if (*iter == view) |
| 81 return; | 83 return; |
| 82 } | 84 } |
| 83 views_.push_back(view); | 85 views_.push_back(view); |
| 84 } | 86 } |
| 85 | 87 |
| 86 void MockPermissionBubbleFactory::HideView(MockPermissionBubbleView* view) { | 88 void MockPermissionBubbleFactory::HideView(MockPermissionBubbleView* view) { |
| 87 for (std::vector<MockPermissionBubbleView*>::iterator iter = views_.begin(); | 89 for (std::vector<MockPermissionBubbleView*>::iterator iter = views_.begin(); |
| 88 iter != views_.end(); ++iter) { | 90 iter != views_.end(); ++iter) { |
| 89 if (*iter == view) { | 91 if (*iter == view) { |
| 90 views_.erase(iter); | 92 views_.erase(iter); |
| 91 return; | 93 return; |
| 92 } | 94 } |
| 93 } | 95 } |
| 94 } | 96 } |
| OLD | NEW |