1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef COMPONENTS_PERMISSIONS_TEST_MOCK_PERMISSION_PROMPT_FACTORY_H_
#define COMPONENTS_PERMISSIONS_TEST_MOCK_PERMISSION_PROMPT_FACTORY_H_
#include <memory>
#include <vector>
#include "base/memory/raw_ptr.h"
#include "base/scoped_observation.h"
#include "components/permissions/permission_prompt.h"
#include "components/permissions/permission_request.h"
#include "components/permissions/permission_request_manager.h"
#include "url/gurl.h"
namespace content {
class WebContents;
}
namespace permissions {
class MockPermissionPrompt;
enum class RequestType;
// Provides a skeleton class for both unit and browser testing when trying to
// test the bubble manager logic. Should not be used for anything that requires
// actual UI.
// See example usage in
// chrome/browser/permissions/permission_request_manager_unittest.cc
class MockPermissionPromptFactory : PermissionRequestManager::Observer {
public:
explicit MockPermissionPromptFactory(PermissionRequestManager* manager);
MockPermissionPromptFactory(const MockPermissionPromptFactory&) = delete;
MockPermissionPromptFactory& operator=(const MockPermissionPromptFactory&) =
delete;
~MockPermissionPromptFactory() override;
// Create method called by the PBM to show a bubble.
std::unique_ptr<PermissionPrompt> Create(
content::WebContents* web_contents,
PermissionPrompt::Delegate* delegate);
void ResetCounts();
void DocumentOnLoadCompletedInPrimaryMainFrame();
void set_response_type(PermissionRequestManager::AutoResponseType type) {
response_type_ = type;
}
PermissionRequestManager::AutoResponseType response_type() {
return response_type_;
}
// If the current view is visible.
bool is_visible();
// Number of times |Show| was called on any bubble.
int show_count() { return show_count_; }
// Number of requests seen by the last |Show|.
int request_count() { return requests_count_; }
// Number of requests seen.
int TotalRequestCount();
// Whether the specified permission was shown in a prompt.
bool RequestTypeSeen(RequestType type);
// Whether a prompt with the given origin was shown.
bool RequestOriginSeen(const GURL& origin);
void WaitForPermissionBubble();
private:
friend class MockPermissionPrompt;
// This shouldn't be called. Is here to fail tests that try to create a bubble
// after the factory has been destroyed.
static std::unique_ptr<PermissionPrompt> DoNotCreate(
content::WebContents* web_contents,
PermissionPrompt::Delegate* delegate);
void HideView(MockPermissionPrompt* view);
// PermissionRequestManager::Observer
void OnPermissionRequestManagerDestructed() override;
int show_count_;
int requests_count_;
std::vector<RequestType> request_types_seen_;
std::vector<GURL> request_origins_seen_;
std::vector<raw_ptr<MockPermissionPrompt, VectorExperimental>> prompts_;
PermissionRequestManager::AutoResponseType response_type_;
base::RepeatingClosure show_bubble_quit_closure_;
base::ScopedObservation<PermissionRequestManager,
PermissionRequestManager::Observer>
observation_{this};
// The bubble manager that will be associated with this factory.
raw_ptr<PermissionRequestManager> manager_;
};
} // namespace permissions
#endif // COMPONENTS_PERMISSIONS_TEST_MOCK_PERMISSION_PROMPT_FACTORY_H_
|
