Index: chrome/browser/ui/permission_bubble/mock_permission_prompt_factory.h |
diff --git a/chrome/browser/ui/permission_bubble/mock_permission_prompt_factory.h b/chrome/browser/ui/permission_bubble/mock_permission_prompt_factory.h |
index ad02121efe13ce523a2256a0c498a9e5f9aa1639..32d689b1eb03a0cc5fdf907ce4fe361b337d5369 100644 |
--- a/chrome/browser/ui/permission_bubble/mock_permission_prompt_factory.h |
+++ b/chrome/browser/ui/permission_bubble/mock_permission_prompt_factory.h |
@@ -8,6 +8,7 @@ |
#include <memory> |
#include <vector> |
+#include "chrome/browser/permissions/permission_request.h" |
#include "chrome/browser/permissions/permission_request_manager.h" |
class MockPermissionPrompt; |
@@ -51,7 +52,9 @@ class MockPermissionPromptFactory { |
// Number of requests seen by the last |Show|. |
int request_count() { return requests_count_; } |
// Number of requests seen. |
- int total_request_count() { return total_requests_count_; } |
+ int TotalRequestCount(); |
+ // Whether the specified permission was shown in a prompt. |
+ bool RequestTypeSeen(PermissionRequestType type); |
void WaitForPermissionBubble(); |
@@ -70,7 +73,8 @@ class MockPermissionPromptFactory { |
bool can_update_ui_; |
int show_count_; |
int requests_count_; |
- int total_requests_count_; |
+ std::vector<PermissionRequestType> request_types_seen_; |
+ |
std::vector<MockPermissionPrompt*> prompts_; |
PermissionRequestManager::AutoResponseType response_type_; |