blob: 802b7126a61d37835e15f66d578372589dc6a76c (
plain)
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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
// Copyright (c) 2012 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.
#ifndef CONTENT_PUBLIC_TEST_MOCK_RENDER_PROCESS_HOST_H_
#define CONTENT_PUBLIC_TEST_MOCK_RENDER_PROCESS_HOST_H_
#include "base/basictypes.h"
#include "base/memory/scoped_vector.h"
#include "base/observer_list.h"
#include "content/public/browser/render_process_host.h"
#include "content/public/browser/render_process_host_factory.h"
#include "ipc/ipc_test_sink.h"
class StoragePartition;
namespace content {
class MockRenderProcessHostFactory;
// A mock render process host that has no corresponding renderer process. All
// IPC messages are sent into the message sink for inspection by tests.
class MockRenderProcessHost : public RenderProcessHost {
public:
explicit MockRenderProcessHost(BrowserContext* browser_context);
~MockRenderProcessHost() override;
// Provides access to all IPC messages that would have been sent to the
// renderer via this RenderProcessHost.
IPC::TestSink& sink() { return sink_; }
// Provides test access to how many times a bad message has been received.
int bad_msg_count() const { return bad_msg_count_; }
// RenderProcessHost implementation (public portion).
void EnableSendQueue() override;
bool Init() override;
int GetNextRoutingID() override;
void AddRoute(int32 routing_id, IPC::Listener* listener) override;
void RemoveRoute(int32 routing_id) override;
void AddObserver(RenderProcessHostObserver* observer) override;
void RemoveObserver(RenderProcessHostObserver* observer) override;
void ReceivedBadMessage() override;
void WidgetRestored() override;
void WidgetHidden() override;
int VisibleWidgetCount() const override;
bool IsIsolatedGuest() const override;
StoragePartition* GetStoragePartition() const override;
virtual void AddWord(const base::string16& word);
bool FastShutdownIfPossible() override;
bool FastShutdownStarted() const override;
void DumpHandles() override;
base::ProcessHandle GetHandle() const override;
int GetID() const override;
bool HasConnection() const override;
void SetIgnoreInputEvents(bool ignore_input_events) override;
bool IgnoreInputEvents() const override;
void Cleanup() override;
void AddPendingView() override;
void RemovePendingView() override;
void SetSuddenTerminationAllowed(bool allowed) override;
bool SuddenTerminationAllowed() const override;
BrowserContext* GetBrowserContext() const override;
bool InSameStoragePartition(StoragePartition* partition) const override;
IPC::ChannelProxy* GetChannel() override;
void AddFilter(BrowserMessageFilter* filter) override;
bool FastShutdownForPageCount(size_t count) override;
base::TimeDelta GetChildProcessIdleTime() const override;
void ResumeRequestsForView(int route_id) override;
void FilterURL(bool empty_allowed, GURL* url) override;
#if defined(ENABLE_WEBRTC)
void EnableAecDump(const base::FilePath& file) override;
void DisableAecDump() override;
void SetWebRtcLogMessageCallback(
base::Callback<void(const std::string&)> callback) override;
WebRtcStopRtpDumpCallback StartRtpDump(
bool incoming,
bool outgoing,
const WebRtcRtpPacketCallback& packet_callback) override;
#endif
void ResumeDeferredNavigation(const GlobalRequestID& request_id) override;
void NotifyTimezoneChange() override;
ServiceRegistry* GetServiceRegistry() override;
// IPC::Sender via RenderProcessHost.
bool Send(IPC::Message* msg) override;
// IPC::Listener via RenderProcessHost.
bool OnMessageReceived(const IPC::Message& msg) override;
void OnChannelConnected(int32 peer_pid) override;
// Attaches the factory object so we can remove this object in its destructor
// and prevent MockRenderProcessHostFacotry from deleting it.
void SetFactory(const MockRenderProcessHostFactory* factory) {
factory_ = factory;
}
int GetActiveViewCount();
void set_is_isolated_guest(bool is_isolated_guest) {
is_isolated_guest_ = is_isolated_guest;
}
void SetProcessHandle(scoped_ptr<base::ProcessHandle> new_handle) {
process_handle = new_handle.Pass();
}
private:
// Stores IPC messages that would have been sent to the renderer.
IPC::TestSink sink_;
int bad_msg_count_;
const MockRenderProcessHostFactory* factory_;
int id_;
BrowserContext* browser_context_;
ObserverList<RenderProcessHostObserver> observers_;
IDMap<RenderWidgetHost> render_widget_hosts_;
int prev_routing_id_;
IDMap<IPC::Listener> listeners_;
bool fast_shutdown_started_;
bool deletion_callback_called_;
bool is_isolated_guest_;
scoped_ptr<base::ProcessHandle> process_handle;
DISALLOW_COPY_AND_ASSIGN(MockRenderProcessHost);
};
class MockRenderProcessHostFactory : public RenderProcessHostFactory {
public:
MockRenderProcessHostFactory();
~MockRenderProcessHostFactory() override;
RenderProcessHost* CreateRenderProcessHost(
BrowserContext* browser_context,
SiteInstance* site_instance) const override;
// Removes the given MockRenderProcessHost from the MockRenderProcessHost list
// without deleting it. When a test deletes a MockRenderProcessHost, we need
// to remove it from |processes_| to prevent it from being deleted twice.
void Remove(MockRenderProcessHost* host) const;
private:
// A list of MockRenderProcessHosts created by this object. This list is used
// for deleting all MockRenderProcessHosts that have not deleted by a test in
// the destructor and prevent them from being leaked.
mutable ScopedVector<MockRenderProcessHost> processes_;
DISALLOW_COPY_AND_ASSIGN(MockRenderProcessHostFactory);
};
} // namespace content
#endif // CONTENT_PUBLIC_TEST_MOCK_RENDER_PROCESS_HOST_H_
|