1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
6 #include "base/bind_helpers.h"
7 #include "base/location.h"
8 #include "base/memory/ref_counted.h"
9 #include "base/process/process.h"
10 #include "ipc/ipc_message.h"
11 #include "ipc/ipc_message_macros.h"
12 #include "ipc/ipc_platform_file.h"
13 #include "remoting/base/auto_thread_task_runner.h"
14 #include "remoting/host/chromoting_messages.h"
15 #include "remoting/host/daemon_process.h"
16 #include "remoting/host/desktop_session.h"
17 #include "testing/gmock_mutant.h"
18 #include "testing/gmock/include/gmock/gmock.h"
19 #include "testing/gtest/include/gtest/gtest.h"
22 using testing::AnyNumber;
23 using testing::InSequence;
30 kMessageCrash = ChromotingDaemonMsg_Crash::ID,
31 kMessageConfiguration = ChromotingDaemonNetworkMsg_Configuration::ID,
32 kMessageConnectTerminal = ChromotingNetworkHostMsg_ConnectTerminal::ID,
33 kMessageDisconnectTerminal = ChromotingNetworkHostMsg_DisconnectTerminal::ID,
34 kMessageTerminalDisconnected =
35 ChromotingDaemonNetworkMsg_TerminalDisconnected::ID
38 // Provides a public constructor allowing the test to create instances of
39 // DesktopSession directly.
40 class FakeDesktopSession : public DesktopSession {
42 FakeDesktopSession(DaemonProcess* daemon_process, int id);
43 virtual ~FakeDesktopSession();
45 virtual void SetScreenResolution(
46 const ScreenResolution& resolution) OVERRIDE {}
49 DISALLOW_COPY_AND_ASSIGN(FakeDesktopSession);
52 class MockDaemonProcess : public DaemonProcess {
55 scoped_refptr<AutoThreadTaskRunner> caller_task_runner,
56 scoped_refptr<AutoThreadTaskRunner> io_task_runner,
57 const base::Closure& stopped_callback);
58 virtual ~MockDaemonProcess();
60 virtual scoped_ptr<DesktopSession> DoCreateDesktopSession(
62 const ScreenResolution& resolution,
63 bool virtual_terminal) OVERRIDE;
65 virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE;
66 virtual void SendToNetwork(IPC::Message* message) OVERRIDE;
68 MOCK_METHOD1(Received, void(const IPC::Message&));
69 MOCK_METHOD1(Sent, void(const IPC::Message&));
71 MOCK_METHOD3(OnDesktopSessionAgentAttached,
72 bool(int, base::ProcessHandle, IPC::PlatformFileForTransit));
74 MOCK_METHOD1(DoCreateDesktopSessionPtr, DesktopSession*(int));
75 MOCK_METHOD1(DoCrashNetworkProcess, void(const tracked_objects::Location&));
76 MOCK_METHOD0(LaunchNetworkProcess, void());
79 DISALLOW_COPY_AND_ASSIGN(MockDaemonProcess);
82 FakeDesktopSession::FakeDesktopSession(DaemonProcess* daemon_process, int id)
83 : DesktopSession(daemon_process, id) {
86 FakeDesktopSession::~FakeDesktopSession() {
89 MockDaemonProcess::MockDaemonProcess(
90 scoped_refptr<AutoThreadTaskRunner> caller_task_runner,
91 scoped_refptr<AutoThreadTaskRunner> io_task_runner,
92 const base::Closure& stopped_callback)
93 : DaemonProcess(caller_task_runner, io_task_runner, stopped_callback) {
96 MockDaemonProcess::~MockDaemonProcess() {
99 scoped_ptr<DesktopSession> MockDaemonProcess::DoCreateDesktopSession(
101 const ScreenResolution& resolution,
102 bool virtual_terminal) {
103 return scoped_ptr<DesktopSession>(DoCreateDesktopSessionPtr(terminal_id));
106 bool MockDaemonProcess::OnMessageReceived(const IPC::Message& message) {
107 // Notify the mock method.
110 // Call the actual handler.
111 return DaemonProcess::OnMessageReceived(message);
114 void MockDaemonProcess::SendToNetwork(IPC::Message* message) {
115 // Notify the mock method.
122 class DaemonProcessTest : public testing::Test {
125 virtual ~DaemonProcessTest();
127 virtual void SetUp() OVERRIDE;
128 virtual void TearDown() OVERRIDE;
130 // DaemonProcess mocks
131 DesktopSession* DoCreateDesktopSession(int terminal_id);
132 void DoCrashNetworkProcess(const tracked_objects::Location& location);
133 void LaunchNetworkProcess();
135 // Deletes |daemon_process_|.
136 void DeleteDaemonProcess();
138 // Quits |message_loop_|.
139 void QuitMessageLoop();
141 void StartDaemonProcess();
143 const DaemonProcess::DesktopSessionList& desktop_sessions() const {
144 return daemon_process_->desktop_sessions();
148 base::MessageLoopForIO message_loop_;
150 scoped_ptr<MockDaemonProcess> daemon_process_;
154 DaemonProcessTest::DaemonProcessTest() : terminal_id_(0) {
157 DaemonProcessTest::~DaemonProcessTest() {
160 void DaemonProcessTest::SetUp() {
161 scoped_refptr<AutoThreadTaskRunner> task_runner = new AutoThreadTaskRunner(
162 message_loop_.message_loop_proxy(),
163 base::Bind(&DaemonProcessTest::QuitMessageLoop,
164 base::Unretained(this)));
165 daemon_process_.reset(
166 new MockDaemonProcess(task_runner, task_runner,
167 base::Bind(&DaemonProcessTest::DeleteDaemonProcess,
168 base::Unretained(this))));
170 // Set up daemon process mocks.
171 EXPECT_CALL(*daemon_process_, DoCreateDesktopSessionPtr(_))
173 .WillRepeatedly(Invoke(this, &DaemonProcessTest::DoCreateDesktopSession));
174 EXPECT_CALL(*daemon_process_, DoCrashNetworkProcess(_))
176 .WillRepeatedly(Invoke(this, &DaemonProcessTest::DoCrashNetworkProcess));
177 EXPECT_CALL(*daemon_process_, LaunchNetworkProcess())
179 .WillRepeatedly(Invoke(this, &DaemonProcessTest::LaunchNetworkProcess));
182 void DaemonProcessTest::TearDown() {
183 daemon_process_->Stop();
187 DesktopSession* DaemonProcessTest::DoCreateDesktopSession(int terminal_id) {
188 return new FakeDesktopSession(daemon_process_.get(), terminal_id);
191 void DaemonProcessTest::DoCrashNetworkProcess(
192 const tracked_objects::Location& location) {
193 daemon_process_->SendToNetwork(
194 new ChromotingDaemonMsg_Crash(location.function_name(),
195 location.file_name(),
196 location.line_number()));
199 void DaemonProcessTest::LaunchNetworkProcess() {
201 daemon_process_->OnChannelConnected(0);
204 void DaemonProcessTest::DeleteDaemonProcess() {
205 daemon_process_.reset();
208 void DaemonProcessTest::QuitMessageLoop() {
209 message_loop_.PostTask(FROM_HERE, base::MessageLoop::QuitClosure());
212 void DaemonProcessTest::StartDaemonProcess() {
213 // DaemonProcess::Initialize() sets up the config watcher that this test does
214 // not support. Launch the process directly.
215 daemon_process_->LaunchNetworkProcess();
218 MATCHER_P(Message, type, "") {
219 return arg.type() == static_cast<uint32>(type);
222 TEST_F(DaemonProcessTest, OpenClose) {
224 EXPECT_CALL(*daemon_process_, Sent(Message(kMessageConfiguration)));
225 EXPECT_CALL(*daemon_process_, Received(Message(kMessageConnectTerminal)));
226 EXPECT_CALL(*daemon_process_, Received(Message(kMessageDisconnectTerminal)));
227 EXPECT_CALL(*daemon_process_, Sent(Message(kMessageTerminalDisconnected)));
229 StartDaemonProcess();
231 int id = terminal_id_++;
232 ScreenResolution resolution;
234 EXPECT_TRUE(daemon_process_->OnMessageReceived(
235 ChromotingNetworkHostMsg_ConnectTerminal(id, resolution, false)));
236 EXPECT_EQ(1u, desktop_sessions().size());
237 EXPECT_EQ(id, desktop_sessions().front()->id());
239 EXPECT_TRUE(daemon_process_->OnMessageReceived(
240 ChromotingNetworkHostMsg_DisconnectTerminal(id)));
241 EXPECT_TRUE(desktop_sessions().empty());
244 TEST_F(DaemonProcessTest, CallCloseDesktopSession) {
246 EXPECT_CALL(*daemon_process_, Sent(Message(kMessageConfiguration)));
247 EXPECT_CALL(*daemon_process_, Received(Message(kMessageConnectTerminal)));
248 EXPECT_CALL(*daemon_process_, Sent(Message(kMessageTerminalDisconnected)));
250 StartDaemonProcess();
252 int id = terminal_id_++;
253 ScreenResolution resolution;
255 EXPECT_TRUE(daemon_process_->OnMessageReceived(
256 ChromotingNetworkHostMsg_ConnectTerminal(id, resolution, false)));
257 EXPECT_EQ(1u, desktop_sessions().size());
258 EXPECT_EQ(id, desktop_sessions().front()->id());
260 daemon_process_->CloseDesktopSession(id);
261 EXPECT_TRUE(desktop_sessions().empty());
264 // Sends two CloseDesktopSession messages and expects the second one to be
266 TEST_F(DaemonProcessTest, DoubleDisconnectTerminal) {
268 EXPECT_CALL(*daemon_process_, Sent(Message(kMessageConfiguration)));
269 EXPECT_CALL(*daemon_process_, Received(Message(kMessageConnectTerminal)));
270 EXPECT_CALL(*daemon_process_, Received(Message(kMessageDisconnectTerminal)));
271 EXPECT_CALL(*daemon_process_, Sent(Message(kMessageTerminalDisconnected)));
272 EXPECT_CALL(*daemon_process_, Received(Message(kMessageDisconnectTerminal)));
274 StartDaemonProcess();
276 int id = terminal_id_++;
277 ScreenResolution resolution;
279 EXPECT_TRUE(daemon_process_->OnMessageReceived(
280 ChromotingNetworkHostMsg_ConnectTerminal(id, resolution, false)));
281 EXPECT_EQ(1u, desktop_sessions().size());
282 EXPECT_EQ(id, desktop_sessions().front()->id());
284 EXPECT_TRUE(daemon_process_->OnMessageReceived(
285 ChromotingNetworkHostMsg_DisconnectTerminal(id)));
286 EXPECT_TRUE(desktop_sessions().empty());
288 EXPECT_TRUE(daemon_process_->OnMessageReceived(
289 ChromotingNetworkHostMsg_DisconnectTerminal(id)));
290 EXPECT_TRUE(desktop_sessions().empty());
293 // Tries to close an invalid terminal ID and expects the network process to be
295 TEST_F(DaemonProcessTest, InvalidDisconnectTerminal) {
297 EXPECT_CALL(*daemon_process_, Sent(Message(kMessageConfiguration)));
298 EXPECT_CALL(*daemon_process_, Received(Message(kMessageDisconnectTerminal)));
299 EXPECT_CALL(*daemon_process_, Sent(Message(kMessageCrash)))
300 .WillOnce(InvokeWithoutArgs(this,
301 &DaemonProcessTest::LaunchNetworkProcess));
302 EXPECT_CALL(*daemon_process_, Sent(Message(kMessageConfiguration)));
304 StartDaemonProcess();
306 int id = terminal_id_++;
308 EXPECT_TRUE(daemon_process_->OnMessageReceived(
309 ChromotingNetworkHostMsg_DisconnectTerminal(id)));
310 EXPECT_TRUE(desktop_sessions().empty());
311 EXPECT_EQ(0, terminal_id_);
314 // Tries to open an invalid terminal ID and expects the network process to be
316 TEST_F(DaemonProcessTest, InvalidConnectTerminal) {
318 EXPECT_CALL(*daemon_process_, Sent(Message(kMessageConfiguration)));
319 EXPECT_CALL(*daemon_process_, Received(Message(kMessageConnectTerminal)));
320 EXPECT_CALL(*daemon_process_, Received(Message(kMessageConnectTerminal)));
321 EXPECT_CALL(*daemon_process_, Sent(Message(kMessageCrash)))
322 .WillOnce(InvokeWithoutArgs(this,
323 &DaemonProcessTest::LaunchNetworkProcess));
324 EXPECT_CALL(*daemon_process_, Sent(Message(kMessageConfiguration)));
326 StartDaemonProcess();
328 int id = terminal_id_++;
329 ScreenResolution resolution;
331 EXPECT_TRUE(daemon_process_->OnMessageReceived(
332 ChromotingNetworkHostMsg_ConnectTerminal(id, resolution, false)));
333 EXPECT_EQ(1u, desktop_sessions().size());
334 EXPECT_EQ(id, desktop_sessions().front()->id());
336 EXPECT_TRUE(daemon_process_->OnMessageReceived(
337 ChromotingNetworkHostMsg_ConnectTerminal(id, resolution, false)));
338 EXPECT_TRUE(desktop_sessions().empty());
339 EXPECT_EQ(0, terminal_id_);
342 } // namespace remoting