| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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/memory/scoped_ptr.h" | 6 #include "base/memory/scoped_ptr.h" |
| 7 #include "base/task.h" | 7 #include "base/task.h" |
| 8 #include "remoting/base/logger.h" | |
| 9 #include "remoting/host/capturer_fake.h" | 8 #include "remoting/host/capturer_fake.h" |
| 10 #include "remoting/host/chromoting_host.h" | 9 #include "remoting/host/chromoting_host.h" |
| 11 #include "remoting/host/chromoting_host_context.h" | 10 #include "remoting/host/chromoting_host_context.h" |
| 12 #include "remoting/host/host_mock_objects.h" | 11 #include "remoting/host/host_mock_objects.h" |
| 13 #include "remoting/host/in_memory_host_config.h" | 12 #include "remoting/host/in_memory_host_config.h" |
| 14 #include "remoting/proto/video.pb.h" | 13 #include "remoting/proto/video.pb.h" |
| 15 #include "remoting/protocol/protocol_mock_objects.h" | 14 #include "remoting/protocol/protocol_mock_objects.h" |
| 16 #include "remoting/protocol/session_config.h" | 15 #include "remoting/protocol/session_config.h" |
| 17 #include "testing/gmock_mutant.h" | 16 #include "testing/gmock_mutant.h" |
| 18 #include "testing/gmock/include/gmock/gmock.h" | 17 #include "testing/gmock/include/gmock/gmock.h" |
| (...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 80 .WillByDefault(Return(&message_loop_)); | 79 .WillByDefault(Return(&message_loop_)); |
| 81 EXPECT_CALL(context_, main_message_loop()) | 80 EXPECT_CALL(context_, main_message_loop()) |
| 82 .Times(AnyNumber()); | 81 .Times(AnyNumber()); |
| 83 EXPECT_CALL(context_, encode_message_loop()) | 82 EXPECT_CALL(context_, encode_message_loop()) |
| 84 .Times(AnyNumber()); | 83 .Times(AnyNumber()); |
| 85 EXPECT_CALL(context_, network_message_loop()) | 84 EXPECT_CALL(context_, network_message_loop()) |
| 86 .Times(AnyNumber()); | 85 .Times(AnyNumber()); |
| 87 EXPECT_CALL(context_, ui_message_loop()) | 86 EXPECT_CALL(context_, ui_message_loop()) |
| 88 .Times(AnyNumber()); | 87 .Times(AnyNumber()); |
| 89 | 88 |
| 90 logger_.reset(new Logger()); | |
| 91 | |
| 92 context_.SetUITaskPostFunction(base::Bind( | 89 context_.SetUITaskPostFunction(base::Bind( |
| 93 static_cast<void(MessageLoop::*)( | 90 static_cast<void(MessageLoop::*)( |
| 94 const tracked_objects::Location&, | 91 const tracked_objects::Location&, |
| 95 const base::Closure&)>(&MessageLoop::PostTask), | 92 const base::Closure&)>(&MessageLoop::PostTask), |
| 96 base::Unretained(&message_loop_))); | 93 base::Unretained(&message_loop_))); |
| 97 | 94 |
| 98 Capturer* capturer = new CapturerFake(); | 95 Capturer* capturer = new CapturerFake(); |
| 99 event_executor_ = new MockEventExecutor(); | 96 event_executor_ = new MockEventExecutor(); |
| 100 curtain_ = new MockCurtain(); | 97 curtain_ = new MockCurtain(); |
| 101 disconnect_window_ = new MockDisconnectWindow(); | 98 disconnect_window_ = new MockDisconnectWindow(); |
| 102 continue_window_ = new MockContinueWindow(); | 99 continue_window_ = new MockContinueWindow(); |
| 103 local_input_monitor_ = new MockLocalInputMonitor(); | 100 local_input_monitor_ = new MockLocalInputMonitor(); |
| 104 desktop_environment_.reset( | 101 desktop_environment_.reset( |
| 105 new DesktopEnvironment(&context_, capturer, event_executor_, curtain_, | 102 new DesktopEnvironment(&context_, capturer, event_executor_, curtain_, |
| 106 disconnect_window_, continue_window_, | 103 disconnect_window_, continue_window_, |
| 107 local_input_monitor_)); | 104 local_input_monitor_)); |
| 108 MockAccessVerifier* access_verifier = new MockAccessVerifier(); | 105 MockAccessVerifier* access_verifier = new MockAccessVerifier(); |
| 109 | 106 |
| 110 host_ = ChromotingHost::Create(&context_, config_, | 107 host_ = ChromotingHost::Create(&context_, config_, |
| 111 desktop_environment_.get(), | 108 desktop_environment_.get(), |
| 112 access_verifier, logger_.get(), false); | 109 access_verifier, false); |
| 113 credentials_.set_type(protocol::PASSWORD); | 110 credentials_.set_type(protocol::PASSWORD); |
| 114 credentials_.set_username("user"); | 111 credentials_.set_username("user"); |
| 115 credentials_.set_credential("password"); | 112 credentials_.set_credential("password"); |
| 116 connection_ = new MockConnectionToClient( | 113 connection_ = new MockConnectionToClient( |
| 117 &message_loop_, &handler_, &host_stub_, event_executor_); | 114 &message_loop_, &handler_, &host_stub_, event_executor_); |
| 118 connection2_ = new MockConnectionToClient( | 115 connection2_ = new MockConnectionToClient( |
| 119 &message_loop_, &handler_, &host_stub2_, &event_executor2_); | 116 &message_loop_, &handler_, &host_stub2_, &event_executor2_); |
| 120 session_.reset(new MockSession()); | 117 session_.reset(new MockSession()); |
| 121 session2_.reset(new MockSession()); | 118 session2_.reset(new MockSession()); |
| 122 session_config_.reset(SessionConfig::CreateDefault()); | 119 session_config_.reset(SessionConfig::CreateDefault()); |
| (...skipping 84 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 207 scoped_refptr<ClientSession> session) { | 204 scoped_refptr<ClientSession> session) { |
| 208 host->clients_.push_back(session); | 205 host->clients_.push_back(session); |
| 209 } | 206 } |
| 210 | 207 |
| 211 void ShutdownHost() { | 208 void ShutdownHost() { |
| 212 host_->Shutdown( | 209 host_->Shutdown( |
| 213 NewRunnableFunction(&PostQuitTask, &message_loop_)); | 210 NewRunnableFunction(&PostQuitTask, &message_loop_)); |
| 214 } | 211 } |
| 215 | 212 |
| 216 protected: | 213 protected: |
| 217 scoped_ptr<Logger> logger_; | |
| 218 MessageLoop message_loop_; | 214 MessageLoop message_loop_; |
| 219 MockConnectionToClientEventHandler handler_; | 215 MockConnectionToClientEventHandler handler_; |
| 220 scoped_ptr<DesktopEnvironment> desktop_environment_; | 216 scoped_ptr<DesktopEnvironment> desktop_environment_; |
| 221 scoped_refptr<ChromotingHost> host_; | 217 scoped_refptr<ChromotingHost> host_; |
| 222 scoped_refptr<InMemoryHostConfig> config_; | 218 scoped_refptr<InMemoryHostConfig> config_; |
| 223 MockChromotingHostContext context_; | 219 MockChromotingHostContext context_; |
| 224 protocol::LocalLoginCredentials credentials_; | 220 protocol::LocalLoginCredentials credentials_; |
| 225 scoped_refptr<MockConnectionToClient> connection_; | 221 scoped_refptr<MockConnectionToClient> connection_; |
| 226 scoped_ptr<MockSession> session_; | 222 scoped_ptr<MockSession> session_; |
| 227 scoped_ptr<SessionConfig> session_config_; | 223 scoped_ptr<SessionConfig> session_config_; |
| (...skipping 270 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 498 EXPECT_CALL(*connection_.get(), Disconnect()) | 494 EXPECT_CALL(*connection_.get(), Disconnect()) |
| 499 .InSequence(s1, s2) | 495 .InSequence(s1, s2) |
| 500 .RetiresOnSaturation(); | 496 .RetiresOnSaturation(); |
| 501 } | 497 } |
| 502 SimulateClientConnection(0, true); | 498 SimulateClientConnection(0, true); |
| 503 message_loop_.Run(); | 499 message_loop_.Run(); |
| 504 host_->set_it2me(false); | 500 host_->set_it2me(false); |
| 505 EXPECT_THAT(curtain_activated, false); | 501 EXPECT_THAT(curtain_activated, false); |
| 506 } | 502 } |
| 507 } // namespace remoting | 503 } // namespace remoting |
| OLD | NEW |