blob: 6309685c150d967d9c1f8214531062ecc643d16f (
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
|
// Copyright (c) 2009 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 CHROME_BROWSER_RENDERER_HOST_MOCK_RENDER_PROCESS_HOST_H_
#define CHROME_BROWSER_RENDERER_HOST_MOCK_RENDER_PROCESS_HOST_H_
#pragma once
#include "base/basictypes.h"
#include "base/scoped_vector.h"
#include "chrome/browser/renderer_host/render_process_host.h"
#include "chrome/common/ipc_test_sink.h"
class MockRenderProcessHostFactory;
class TransportDIB;
class URLRequestContextGetter;
// 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(Profile* profile);
virtual ~MockRenderProcessHost();
// Provides access to all IPC messages that would have been sent to the
// renderer via this RenderProcessHost.
IPC::TestSink& sink() { return sink_; }
// Provides tests access to the max page ID currently used for this process.
int max_page_id() const { return max_page_id_; }
// 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).
virtual bool Init(bool is_accessibility_enabled, bool is_extensions_process);
virtual int GetNextRoutingID();
virtual void CancelResourceRequests(int render_widget_id);
virtual void CrossSiteClosePageACK(const ViewMsg_ClosePage_Params& params);
virtual bool WaitForUpdateMsg(int render_widget_id,
const base::TimeDelta& max_delay,
IPC::Message* msg);
virtual void ReceivedBadMessage();
virtual void WidgetRestored();
virtual void WidgetHidden();
virtual void ViewCreated();
virtual void AddWord(const string16& word);
virtual void SendVisitedLinkTable(base::SharedMemory* table_memory);
virtual void AddVisitedLinks(
const VisitedLinkCommon::Fingerprints& visited_links);
virtual void ResetVisitedLinks();
virtual bool FastShutdownIfPossible();
virtual bool SendWithTimeout(IPC::Message* msg, int timeout_ms);
virtual base::ProcessHandle GetHandle();
virtual TransportDIB* GetTransportDIB(TransportDIB::Id dib_id);
// IPC::Channel::Sender via RenderProcessHost.
virtual bool Send(IPC::Message* msg);
// IPC::Channel::Listener via RenderProcessHost.
virtual bool OnMessageReceived(const IPC::Message& msg);
virtual void OnChannelConnected(int32 peer_pid);
// 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;
}
private:
// Stores IPC messages that would have been sent to the renderer.
IPC::TestSink sink_;
TransportDIB* transport_dib_;
int bad_msg_count_;
const MockRenderProcessHostFactory* factory_;
DISALLOW_COPY_AND_ASSIGN(MockRenderProcessHost);
};
class MockRenderProcessHostFactory : public RenderProcessHostFactory {
public:
MockRenderProcessHostFactory();
virtual ~MockRenderProcessHostFactory();
virtual RenderProcessHost* CreateRenderProcessHost(Profile* profile) const;
// 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);
};
#endif // CHROME_BROWSER_RENDERER_HOST_MOCK_RENDER_PROCESS_HOST_H_
|
