| // 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/permission_bubble/mock_permission_prompt_factory.h" |
| |
| #include "base/bind.h" |
| #include "base/memory/ptr_util.h" |
| #include "base/run_loop.h" |
| #include "base/stl_util.h" |
| #include "chrome/browser/permissions/permission_request_manager.h" |
| #include "chrome/browser/ui/permission_bubble/mock_permission_prompt.h" |
| #include "content/public/browser/web_contents.h" |
| |
| MockPermissionPromptFactory::MockPermissionPromptFactory( |
| PermissionRequestManager* manager) |
| : show_count_(0), |
| 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); |
| for (auto* prompt : prompts_) |
| prompt->factory_ = nullptr; |
| prompts_.clear(); |
| } |
| |
| std::unique_ptr<PermissionPrompt> MockPermissionPromptFactory::Create( |
| content::WebContents* web_contents, |
| PermissionPrompt::Delegate* delegate) { |
| MockPermissionPrompt* prompt = new MockPermissionPrompt(this, delegate); |
| |
| prompts_.push_back(prompt); |
| show_count_++; |
| requests_count_ = delegate->Requests().size(); |
| for (const PermissionRequest* request : delegate->Requests()) { |
| request_types_seen_.push_back(request->GetPermissionRequestType()); |
| request_origins_seen_.push_back(request->GetOrigin()); |
| } |
| |
| if (!show_bubble_quit_closure_.is_null()) |
| show_bubble_quit_closure_.Run(); |
| |
| manager_->set_auto_response_for_test(response_type_); |
| return base::WrapUnique(prompt); |
| } |
| |
| void MockPermissionPromptFactory::ResetCounts() { |
| show_count_ = 0; |
| requests_count_ = 0; |
| request_types_seen_.clear(); |
| request_origins_seen_.clear(); |
| } |
| |
| void MockPermissionPromptFactory::DocumentOnLoadCompletedInMainFrame() { |
| manager_->DocumentOnLoadCompletedInMainFrame(); |
| } |
| |
| bool MockPermissionPromptFactory::is_visible() { |
| return !prompts_.empty(); |
| } |
| |
| int MockPermissionPromptFactory::TotalRequestCount() { |
| return request_types_seen_.size(); |
| } |
| |
| bool MockPermissionPromptFactory::RequestTypeSeen(PermissionRequestType type) { |
| return base::ContainsValue(request_types_seen_, type); |
| } |
| |
| bool MockPermissionPromptFactory::RequestOriginSeen(const GURL& origin) { |
| return base::ContainsValue(request_origins_seen_, origin); |
| } |
| |
| 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( |
| content::WebContents* web_contents, |
| PermissionPrompt::Delegate* delegate) { |
| NOTREACHED(); |
| return base::WrapUnique(new MockPermissionPrompt(nullptr, nullptr)); |
| } |
| |
| void MockPermissionPromptFactory::HideView(MockPermissionPrompt* prompt) { |
| auto it = std::find(prompts_.begin(), prompts_.end(), prompt); |
| if (it != prompts_.end()) |
| prompts_.erase(it); |
| } |