OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/bind.h" | 5 #include "base/bind.h" |
6 #include "base/bind_helpers.h" | 6 #include "base/bind_helpers.h" |
| 7 #include "base/location.h" |
7 #include "base/memory/ref_counted.h" | 8 #include "base/memory/ref_counted.h" |
8 #include "base/process.h" | 9 #include "base/process.h" |
9 #include "ipc/ipc_message.h" | 10 #include "ipc/ipc_message.h" |
10 #include "ipc/ipc_message_macros.h" | 11 #include "ipc/ipc_message_macros.h" |
11 #include "ipc/ipc_platform_file.h" | 12 #include "ipc/ipc_platform_file.h" |
12 #include "remoting/base/auto_thread_task_runner.h" | 13 #include "remoting/base/auto_thread_task_runner.h" |
13 #include "remoting/host/chromoting_messages.h" | 14 #include "remoting/host/chromoting_messages.h" |
14 #include "remoting/host/daemon_process.h" | 15 #include "remoting/host/daemon_process.h" |
15 #include "remoting/host/desktop_session.h" | 16 #include "remoting/host/desktop_session.h" |
16 #include "testing/gmock_mutant.h" | 17 #include "testing/gmock_mutant.h" |
17 #include "testing/gmock/include/gmock/gmock.h" | 18 #include "testing/gmock/include/gmock/gmock.h" |
18 #include "testing/gtest/include/gtest/gtest.h" | 19 #include "testing/gtest/include/gtest/gtest.h" |
19 | 20 |
20 using testing::_; | 21 using testing::_; |
21 using testing::AnyNumber; | 22 using testing::AnyNumber; |
22 using testing::InSequence; | 23 using testing::InSequence; |
23 | 24 |
24 namespace remoting { | 25 namespace remoting { |
25 | 26 |
26 namespace { | 27 namespace { |
27 | 28 |
28 enum Messages { | 29 enum Messages { |
29 kMessageCrash = ChromotingDaemonNetworkMsg_Crash::ID, | 30 kMessageCrash = ChromotingDaemonMsg_Crash::ID, |
30 kMessageConfiguration = ChromotingDaemonNetworkMsg_Configuration::ID, | 31 kMessageConfiguration = ChromotingDaemonNetworkMsg_Configuration::ID, |
31 kMessageConnectTerminal = ChromotingNetworkHostMsg_ConnectTerminal::ID, | 32 kMessageConnectTerminal = ChromotingNetworkHostMsg_ConnectTerminal::ID, |
32 kMessageDisconnectTerminal = ChromotingNetworkHostMsg_DisconnectTerminal::ID, | 33 kMessageDisconnectTerminal = ChromotingNetworkHostMsg_DisconnectTerminal::ID, |
33 kMessageTerminalDisconnected = | 34 kMessageTerminalDisconnected = |
34 ChromotingDaemonNetworkMsg_TerminalDisconnected::ID | 35 ChromotingDaemonNetworkMsg_TerminalDisconnected::ID |
35 }; | 36 }; |
36 | 37 |
37 // Provides a public constructor allowing the test to create instances of | 38 // Provides a public constructor allowing the test to create instances of |
38 // DesktopSession directly. | 39 // DesktopSession directly. |
39 class FakeDesktopSession : public DesktopSession { | 40 class FakeDesktopSession : public DesktopSession { |
(...skipping 18 matching lines...) Expand all Loading... |
58 virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; | 59 virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; |
59 virtual void SendToNetwork(IPC::Message* message) OVERRIDE; | 60 virtual void SendToNetwork(IPC::Message* message) OVERRIDE; |
60 | 61 |
61 MOCK_METHOD1(Received, void(const IPC::Message&)); | 62 MOCK_METHOD1(Received, void(const IPC::Message&)); |
62 MOCK_METHOD1(Sent, void(const IPC::Message&)); | 63 MOCK_METHOD1(Sent, void(const IPC::Message&)); |
63 | 64 |
64 MOCK_METHOD3(OnDesktopSessionAgentAttached, | 65 MOCK_METHOD3(OnDesktopSessionAgentAttached, |
65 bool(int, base::ProcessHandle, IPC::PlatformFileForTransit)); | 66 bool(int, base::ProcessHandle, IPC::PlatformFileForTransit)); |
66 | 67 |
67 MOCK_METHOD1(DoCreateDesktopSessionPtr, DesktopSession*(int)); | 68 MOCK_METHOD1(DoCreateDesktopSessionPtr, DesktopSession*(int)); |
| 69 MOCK_METHOD1(DoCrashNetworkProcess, void(const tracked_objects::Location&)); |
68 MOCK_METHOD0(LaunchNetworkProcess, void()); | 70 MOCK_METHOD0(LaunchNetworkProcess, void()); |
69 | 71 |
70 private: | 72 private: |
71 DISALLOW_COPY_AND_ASSIGN(MockDaemonProcess); | 73 DISALLOW_COPY_AND_ASSIGN(MockDaemonProcess); |
72 }; | 74 }; |
73 | 75 |
74 FakeDesktopSession::FakeDesktopSession(DaemonProcess* daemon_process, int id) | 76 FakeDesktopSession::FakeDesktopSession(DaemonProcess* daemon_process, int id) |
75 : DesktopSession(daemon_process, id) { | 77 : DesktopSession(daemon_process, id) { |
76 } | 78 } |
77 | 79 |
(...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
112 class DaemonProcessTest : public testing::Test { | 114 class DaemonProcessTest : public testing::Test { |
113 public: | 115 public: |
114 DaemonProcessTest(); | 116 DaemonProcessTest(); |
115 virtual ~DaemonProcessTest(); | 117 virtual ~DaemonProcessTest(); |
116 | 118 |
117 virtual void SetUp() OVERRIDE; | 119 virtual void SetUp() OVERRIDE; |
118 virtual void TearDown() OVERRIDE; | 120 virtual void TearDown() OVERRIDE; |
119 | 121 |
120 // DaemonProcess mocks | 122 // DaemonProcess mocks |
121 DesktopSession* DoCreateDesktopSession(int terminal_id); | 123 DesktopSession* DoCreateDesktopSession(int terminal_id); |
| 124 void DoCrashNetworkProcess(const tracked_objects::Location& location); |
122 void LaunchNetworkProcess(); | 125 void LaunchNetworkProcess(); |
123 | 126 |
124 // Deletes |daemon_process_|. | 127 // Deletes |daemon_process_|. |
125 void DeleteDaemonProcess(); | 128 void DeleteDaemonProcess(); |
126 | 129 |
127 // Quits |message_loop_|. | 130 // Quits |message_loop_|. |
128 void QuitMessageLoop(); | 131 void QuitMessageLoop(); |
129 | 132 |
130 void StartDaemonProcess(); | 133 void StartDaemonProcess(); |
131 | 134 |
(...skipping 24 matching lines...) Expand all Loading... |
156 base::Unretained(this))); | 159 base::Unretained(this))); |
157 daemon_process_.reset( | 160 daemon_process_.reset( |
158 new MockDaemonProcess(task_runner, task_runner, | 161 new MockDaemonProcess(task_runner, task_runner, |
159 base::Bind(&DaemonProcessTest::DeleteDaemonProcess, | 162 base::Bind(&DaemonProcessTest::DeleteDaemonProcess, |
160 base::Unretained(this)))); | 163 base::Unretained(this)))); |
161 | 164 |
162 // Set up daemon process mocks. | 165 // Set up daemon process mocks. |
163 EXPECT_CALL(*daemon_process_, DoCreateDesktopSessionPtr(_)) | 166 EXPECT_CALL(*daemon_process_, DoCreateDesktopSessionPtr(_)) |
164 .Times(AnyNumber()) | 167 .Times(AnyNumber()) |
165 .WillRepeatedly(Invoke(this, &DaemonProcessTest::DoCreateDesktopSession)); | 168 .WillRepeatedly(Invoke(this, &DaemonProcessTest::DoCreateDesktopSession)); |
| 169 EXPECT_CALL(*daemon_process_, DoCrashNetworkProcess(_)) |
| 170 .Times(AnyNumber()) |
| 171 .WillRepeatedly(Invoke(this, &DaemonProcessTest::DoCrashNetworkProcess)); |
166 EXPECT_CALL(*daemon_process_, LaunchNetworkProcess()) | 172 EXPECT_CALL(*daemon_process_, LaunchNetworkProcess()) |
167 .Times(AnyNumber()) | 173 .Times(AnyNumber()) |
168 .WillRepeatedly(Invoke(this, &DaemonProcessTest::LaunchNetworkProcess)); | 174 .WillRepeatedly(Invoke(this, &DaemonProcessTest::LaunchNetworkProcess)); |
169 } | 175 } |
170 | 176 |
171 void DaemonProcessTest::TearDown() { | 177 void DaemonProcessTest::TearDown() { |
172 daemon_process_->Stop(); | 178 daemon_process_->Stop(); |
173 message_loop_.Run(); | 179 message_loop_.Run(); |
174 } | 180 } |
175 | 181 |
176 DesktopSession* DaemonProcessTest::DoCreateDesktopSession(int terminal_id) { | 182 DesktopSession* DaemonProcessTest::DoCreateDesktopSession(int terminal_id) { |
177 return new FakeDesktopSession(daemon_process_.get(), terminal_id); | 183 return new FakeDesktopSession(daemon_process_.get(), terminal_id); |
178 } | 184 } |
179 | 185 |
| 186 void DaemonProcessTest::DoCrashNetworkProcess( |
| 187 const tracked_objects::Location& location) { |
| 188 daemon_process_->SendToNetwork( |
| 189 new ChromotingDaemonMsg_Crash(location.function_name(), |
| 190 location.file_name(), |
| 191 location.line_number())); |
| 192 } |
| 193 |
180 void DaemonProcessTest::LaunchNetworkProcess() { | 194 void DaemonProcessTest::LaunchNetworkProcess() { |
181 terminal_id_ = 0; | 195 terminal_id_ = 0; |
182 daemon_process_->OnChannelConnected(0); | 196 daemon_process_->OnChannelConnected(0); |
183 } | 197 } |
184 | 198 |
185 void DaemonProcessTest::DeleteDaemonProcess() { | 199 void DaemonProcessTest::DeleteDaemonProcess() { |
186 daemon_process_.reset(); | 200 daemon_process_.reset(); |
187 } | 201 } |
188 | 202 |
189 void DaemonProcessTest::QuitMessageLoop() { | 203 void DaemonProcessTest::QuitMessageLoop() { |
(...skipping 120 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
310 EXPECT_EQ(1u, desktop_sessions().size()); | 324 EXPECT_EQ(1u, desktop_sessions().size()); |
311 EXPECT_EQ(id, desktop_sessions().front()->id()); | 325 EXPECT_EQ(id, desktop_sessions().front()->id()); |
312 | 326 |
313 EXPECT_TRUE(daemon_process_->OnMessageReceived( | 327 EXPECT_TRUE(daemon_process_->OnMessageReceived( |
314 ChromotingNetworkHostMsg_ConnectTerminal(id))); | 328 ChromotingNetworkHostMsg_ConnectTerminal(id))); |
315 EXPECT_TRUE(desktop_sessions().empty()); | 329 EXPECT_TRUE(desktop_sessions().empty()); |
316 EXPECT_EQ(0, terminal_id_); | 330 EXPECT_EQ(0, terminal_id_); |
317 } | 331 } |
318 | 332 |
319 } // namespace remoting | 333 } // namespace remoting |
OLD | NEW |