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
|
// 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_RENDERER_MOCK_RENDER_THREAD_H_
#define CHROME_RENDERER_MOCK_RENDER_THREAD_H_
#include <string>
#include <vector>
#include "chrome/common/ipc_test_sink.h"
#include "chrome/renderer/mock_printer.h"
#include "chrome/renderer/render_thread.h"
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);
// Our mock thread doesn't do filtering.
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() { }
//////////////////////////////////////////////////////////////////////////
// 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,
bool activatable,
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(size_t buffer_size,
base::SharedMemoryHandle* handle);
#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_;
};
#endif // CHROME_RENDERER_MOCK_RENDER_THREAD_H_
|