Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "chrome/browser/ui/website_settings/mock_permission_prompt_factory.h" | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "base/memory/ptr_util.h" | |
| 9 #include "base/run_loop.h" | |
| 10 #include "chrome/browser/permissions/permission_request_manager.h" | |
| 11 #include "chrome/browser/ui/website_settings/mock_permission_prompt.h" | |
| 12 | |
| 13 MockPermissionPromptFactory::MockPermissionPromptFactory( | |
| 14 PermissionRequestManager* manager) | |
| 15 : can_update_ui_(false), | |
| 16 show_count_(0), | |
| 17 requests_count_(0), | |
| 18 total_requests_count_(0), | |
| 19 response_type_(PermissionRequestManager::NONE), | |
| 20 manager_(manager) { | |
| 21 manager->view_factory_ = | |
| 22 base::Bind(&MockPermissionPromptFactory::Create, base::Unretained(this)); | |
| 23 } | |
| 24 | |
| 25 MockPermissionPromptFactory::~MockPermissionPromptFactory() { | |
| 26 manager_->view_factory_ = | |
| 27 base::Bind(&MockPermissionPromptFactory::DoNotCreate); | |
| 28 while (!prompts_.empty()) { | |
|
Lei Zhang
2016/08/02 03:59:22
Can this just iterate in reverse, set |factory_| t
tsergeant
2016/08/02 08:05:16
Is there any particular relevance to going in reve
Lei Zhang
2016/08/02 08:21:16
The code is currently going in reverse. Wasn't sur
tsergeant
2016/08/02 23:28:10
Yeah, I don't think there's any reason other than
| |
| 29 prompts_.back()->factory_ = nullptr; | |
| 30 prompts_.pop_back(); | |
| 31 } | |
| 32 } | |
| 33 | |
| 34 std::unique_ptr<PermissionPrompt> MockPermissionPromptFactory::Create( | |
| 35 Browser* browser) { | |
| 36 MockPermissionPrompt* prompt = new MockPermissionPrompt(this, manager_); | |
| 37 prompt->can_update_ui_ = can_update_ui_; | |
| 38 return base::WrapUnique(prompt); | |
| 39 } | |
| 40 | |
| 41 void MockPermissionPromptFactory::SetCanUpdateUi(bool can_update_ui) { | |
| 42 can_update_ui_ = can_update_ui; | |
| 43 for (auto* prompt : prompts_) | |
| 44 prompt->can_update_ui_ = can_update_ui_; | |
| 45 } | |
| 46 | |
| 47 void MockPermissionPromptFactory::ResetCounts() { | |
| 48 show_count_ = 0; | |
| 49 requests_count_ = 0; | |
| 50 total_requests_count_ = 0; | |
| 51 } | |
| 52 | |
| 53 void MockPermissionPromptFactory::DocumentOnLoadCompletedInMainFrame() { | |
| 54 manager_->DocumentOnLoadCompletedInMainFrame(); | |
| 55 } | |
| 56 | |
| 57 bool MockPermissionPromptFactory::is_visible() { | |
| 58 for (auto* prompt : prompts_) { | |
| 59 if (prompt->IsVisible()) | |
| 60 return true; | |
| 61 } | |
| 62 return false; | |
| 63 } | |
| 64 | |
| 65 void MockPermissionPromptFactory::WaitForPermissionBubble() { | |
| 66 if (is_visible()) | |
| 67 return; | |
| 68 DCHECK(show_bubble_quit_closure_.is_null()); | |
| 69 base::RunLoop loop; | |
| 70 show_bubble_quit_closure_ = loop.QuitClosure(); | |
| 71 loop.Run(); | |
| 72 show_bubble_quit_closure_ = base::Closure(); | |
| 73 } | |
| 74 | |
| 75 // static | |
| 76 std::unique_ptr<PermissionPrompt> MockPermissionPromptFactory::DoNotCreate( | |
| 77 Browser* browser) { | |
| 78 NOTREACHED(); | |
| 79 return base::WrapUnique(new MockPermissionPrompt(nullptr, nullptr)); | |
| 80 } | |
| 81 | |
| 82 void MockPermissionPromptFactory::UpdateResponseType() { | |
| 83 manager_->set_auto_response_for_test(response_type_); | |
| 84 } | |
| 85 | |
| 86 void MockPermissionPromptFactory::ShowView(MockPermissionPrompt* prompt) { | |
| 87 for (std::vector<MockPermissionPrompt*>::iterator iter = prompts_.begin(); | |
|
Lei Zhang
2016/08/02 03:59:22
if (ContainsValue(prompts_, prompt))
return;
tsergeant
2016/08/02 08:05:16
Done.
| |
| 88 iter != prompts_.end(); ++iter) { | |
| 89 if (*iter == prompt) | |
| 90 return; | |
| 91 } | |
| 92 prompts_.push_back(prompt); | |
| 93 | |
| 94 if (!show_bubble_quit_closure_.is_null()) | |
| 95 show_bubble_quit_closure_.Run(); | |
| 96 } | |
| 97 | |
| 98 void MockPermissionPromptFactory::HideView(MockPermissionPrompt* prompt) { | |
| 99 for (std::vector<MockPermissionPrompt*>::iterator iter = prompts_.begin(); | |
|
Lei Zhang
2016/08/02 03:59:22
auto it = std::find(prompts_.begin(), prompts_.end
tsergeant
2016/08/02 08:05:16
Done.
| |
| 100 iter != prompts_.end(); ++iter) { | |
| 101 if (*iter == prompt) { | |
| 102 prompts_.erase(iter); | |
| 103 return; | |
| 104 } | |
| 105 } | |
| 106 } | |
| OLD | NEW |