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
|
// Copyright (c) 2010 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_RENDERER_MOCK_RENDER_THREAD_H_
#define CHROME_RENDERER_MOCK_RENDER_THREAD_H_
#pragma once
#include <string>
#include "chrome/common/ipc_test_sink.h"
#include "chrome/renderer/mock_printer.h"
#include "chrome/renderer/render_thread.h"
#include "third_party/WebKit/WebKit/chromium/public/WebPopupType.h"
namespace IPC {
class MessageReplyDeserializer;
}
struct ViewMsg_Print_Params;
struct ViewMsg_PrintPages_Params;
struct ViewHostMsg_ScriptedPrint_Params;
// This class is very simple mock of RenderThread. It simulates an IPC channel
// which supports only two messages:
// ViewHostMsg_CreateWidget : sync message sent by the Widget.
// ViewMsg_Close : async, send to the Widget.
class MockRenderThread : public RenderThreadBase {
public:
MockRenderThread();
virtual ~MockRenderThread();
// Provides access to the messages that have been received by this thread.
IPC::TestSink& sink() { return sink_; }
// Called by the Widget. The routing_id must match the routing id assigned
// to the Widget in reply to ViewHostMsg_CreateWidget message.
virtual void AddRoute(int32 routing_id, IPC::Channel::Listener* listener);
// Called by the Widget. The routing id must match the routing id of AddRoute.
virtual void RemoveRoute(int32 routing_id);
// Called by the Widget. Used to send messages to the browser.
// We short-circuit the mechanim and handle the messages right here on this
// class.
virtual bool Send(IPC::Message* msg);
// Filtering support.
virtual void AddFilter(IPC::ChannelProxy::MessageFilter* filter);
virtual void RemoveFilter(IPC::ChannelProxy::MessageFilter* filter);
// Our mock thread doesn't deal with hidden and restored tabs.
virtual void WidgetHidden() { }
virtual void WidgetRestored() { }
virtual bool IsExtensionProcess() const { return is_extension_process_; }
virtual bool IsIncognitoProcess() const { return false; }
void SetExtensionProcess(bool value) { is_extension_process_ = value; }
//////////////////////////////////////////////////////////////////////////
// The following functions are called by the test itself.
void set_routing_id(int32 id) {
routing_id_ = id;
}
int32 opener_id() const {
return opener_id_;
}
bool has_widget() const {
return widget_ ? true : false;
}
// Simulates the Widget receiving a close message. This should result
// on releasing the internal reference counts and destroying the internal
// state.
void SendCloseMessage();
// Returns the pseudo-printer instance.
MockPrinter* printer() const { return printer_.get(); }
private:
// This function operates as a regular IPC listener.
void OnMessageReceived(const IPC::Message& msg);
// The Widget expects to be returned valid route_id.
void OnMsgCreateWidget(int opener_id,
WebKit::WebPopupType popup_type,
int* route_id);
// The callee expects to be returned a valid channel_id.
void OnMsgOpenChannelToExtension(
int routing_id, const std::string& extension_id,
const std::string& source_extension_id,
const std::string& target_extension_id, int* port_id);
#if defined(OS_WIN)
void OnDuplicateSection(base::SharedMemoryHandle renderer_handle,
base::SharedMemoryHandle* browser_handle);
#endif
#if defined(OS_MACOSX)
void OnAllocatePDFTransport(uint32 buffer_size,
base::SharedMemoryHandle* handle);
#endif
#if defined(OS_LINUX)
void OnAllocateTempFileForPrinting(base::FileDescriptor* renderer_fd,
int* browser_fd);
void OnTempFileForPrintingWritten(int browser_fd);
#endif
// The RenderView expects default print settings.
void OnGetDefaultPrintSettings(ViewMsg_Print_Params* setting);
// The RenderView expects final print settings from the user.
void OnScriptedPrint(const ViewHostMsg_ScriptedPrint_Params& params,
ViewMsg_PrintPages_Params* settings);
void OnDidGetPrintedPagesCount(int cookie, int number_pages);
void OnDidPrintPage(const ViewHostMsg_DidPrintPage_Params& params);
IPC::TestSink sink_;
// Routing id what will be assigned to the Widget.
int32 routing_id_;
// Opener id reported by the Widget.
int32 opener_id_;
// We only keep track of one Widget, we learn its pointer when it
// adds a new route.
IPC::Channel::Listener* widget_;
// The last known good deserializer for sync messages.
scoped_ptr<IPC::MessageReplyDeserializer> reply_deserializer_;
// A mock printer device used for printing tests.
scoped_ptr<MockPrinter> printer_;
bool is_extension_process_;
};
#endif // CHROME_RENDERER_MOCK_RENDER_THREAD_H_
|
