| 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 "base/stl_util.h" |
| 11 #include "chrome/browser/permissions/permission_request_manager.h" |
| 12 #include "chrome/browser/ui/website_settings/mock_permission_prompt.h" |
| 13 |
| 14 MockPermissionPromptFactory::MockPermissionPromptFactory( |
| 15 PermissionRequestManager* manager) |
| 16 : can_update_ui_(false), |
| 17 show_count_(0), |
| 18 requests_count_(0), |
| 19 total_requests_count_(0), |
| 20 response_type_(PermissionRequestManager::NONE), |
| 21 manager_(manager) { |
| 22 manager->view_factory_ = |
| 23 base::Bind(&MockPermissionPromptFactory::Create, base::Unretained(this)); |
| 24 } |
| 25 |
| 26 MockPermissionPromptFactory::~MockPermissionPromptFactory() { |
| 27 manager_->view_factory_ = |
| 28 base::Bind(&MockPermissionPromptFactory::DoNotCreate); |
| 29 for (auto* prompt : prompts_) |
| 30 prompt->factory_ = nullptr; |
| 31 prompts_.clear(); |
| 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 if (ContainsValue(prompts_, prompt)) |
| 88 return; |
| 89 prompts_.push_back(prompt); |
| 90 |
| 91 if (!show_bubble_quit_closure_.is_null()) |
| 92 show_bubble_quit_closure_.Run(); |
| 93 } |
| 94 |
| 95 void MockPermissionPromptFactory::HideView(MockPermissionPrompt* prompt) { |
| 96 auto it = std::find(prompts_.begin(), prompts_.end(), prompt); |
| 97 if (it != prompts_.end()) |
| 98 prompts_.erase(it); |
| 99 } |
| OLD | NEW |