Chromium Code Reviews| Index: chrome/browser/ui/website_settings/mock_permission_prompt_factory.cc |
| diff --git a/chrome/browser/ui/website_settings/mock_permission_prompt_factory.cc b/chrome/browser/ui/website_settings/mock_permission_prompt_factory.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..cab5df61116b74bedd98ae1192598730e4b93b1d |
| --- /dev/null |
| +++ b/chrome/browser/ui/website_settings/mock_permission_prompt_factory.cc |
| @@ -0,0 +1,106 @@ |
| +// Copyright 2016 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#include "chrome/browser/ui/website_settings/mock_permission_prompt_factory.h" |
| + |
| +#include "base/bind.h" |
| +#include "base/memory/ptr_util.h" |
| +#include "base/run_loop.h" |
| +#include "chrome/browser/permissions/permission_request_manager.h" |
| +#include "chrome/browser/ui/website_settings/mock_permission_prompt.h" |
| + |
| +MockPermissionPromptFactory::MockPermissionPromptFactory( |
| + PermissionRequestManager* manager) |
| + : can_update_ui_(false), |
| + show_count_(0), |
| + requests_count_(0), |
| + total_requests_count_(0), |
| + response_type_(PermissionRequestManager::NONE), |
| + manager_(manager) { |
| + manager->view_factory_ = |
| + base::Bind(&MockPermissionPromptFactory::Create, base::Unretained(this)); |
| +} |
| + |
| +MockPermissionPromptFactory::~MockPermissionPromptFactory() { |
| + manager_->view_factory_ = |
| + base::Bind(&MockPermissionPromptFactory::DoNotCreate); |
| + 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
|
| + prompts_.back()->factory_ = nullptr; |
| + prompts_.pop_back(); |
| + } |
| +} |
| + |
| +std::unique_ptr<PermissionPrompt> MockPermissionPromptFactory::Create( |
| + Browser* browser) { |
| + MockPermissionPrompt* prompt = new MockPermissionPrompt(this, manager_); |
| + prompt->can_update_ui_ = can_update_ui_; |
| + return base::WrapUnique(prompt); |
| +} |
| + |
| +void MockPermissionPromptFactory::SetCanUpdateUi(bool can_update_ui) { |
| + can_update_ui_ = can_update_ui; |
| + for (auto* prompt : prompts_) |
| + prompt->can_update_ui_ = can_update_ui_; |
| +} |
| + |
| +void MockPermissionPromptFactory::ResetCounts() { |
| + show_count_ = 0; |
| + requests_count_ = 0; |
| + total_requests_count_ = 0; |
| +} |
| + |
| +void MockPermissionPromptFactory::DocumentOnLoadCompletedInMainFrame() { |
| + manager_->DocumentOnLoadCompletedInMainFrame(); |
| +} |
| + |
| +bool MockPermissionPromptFactory::is_visible() { |
| + for (auto* prompt : prompts_) { |
| + if (prompt->IsVisible()) |
| + return true; |
| + } |
| + return false; |
| +} |
| + |
| +void MockPermissionPromptFactory::WaitForPermissionBubble() { |
| + if (is_visible()) |
| + return; |
| + DCHECK(show_bubble_quit_closure_.is_null()); |
| + base::RunLoop loop; |
| + show_bubble_quit_closure_ = loop.QuitClosure(); |
| + loop.Run(); |
| + show_bubble_quit_closure_ = base::Closure(); |
| +} |
| + |
| +// static |
| +std::unique_ptr<PermissionPrompt> MockPermissionPromptFactory::DoNotCreate( |
| + Browser* browser) { |
| + NOTREACHED(); |
| + return base::WrapUnique(new MockPermissionPrompt(nullptr, nullptr)); |
| +} |
| + |
| +void MockPermissionPromptFactory::UpdateResponseType() { |
| + manager_->set_auto_response_for_test(response_type_); |
| +} |
| + |
| +void MockPermissionPromptFactory::ShowView(MockPermissionPrompt* prompt) { |
| + 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.
|
| + iter != prompts_.end(); ++iter) { |
| + if (*iter == prompt) |
| + return; |
| + } |
| + prompts_.push_back(prompt); |
| + |
| + if (!show_bubble_quit_closure_.is_null()) |
| + show_bubble_quit_closure_.Run(); |
| +} |
| + |
| +void MockPermissionPromptFactory::HideView(MockPermissionPrompt* prompt) { |
| + 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.
|
| + iter != prompts_.end(); ++iter) { |
| + if (*iter == prompt) { |
| + prompts_.erase(iter); |
| + return; |
| + } |
| + } |
| +} |