Chromium Code Reviews| Index: remoting/host/chromoting_host_unittest.cc |
| diff --git a/remoting/host/chromoting_host_unittest.cc b/remoting/host/chromoting_host_unittest.cc |
| index 5338c080771088911aad2eb709968a0c0e58d462..b614fe07633da657663caaae4100908b82072a27 100644 |
| --- a/remoting/host/chromoting_host_unittest.cc |
| +++ b/remoting/host/chromoting_host_unittest.cc |
| @@ -2,12 +2,14 @@ |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| +#include "base/bind.h" |
| #include "base/task.h" |
| #include "remoting/host/capturer_fake.h" |
| #include "remoting/host/chromoting_host.h" |
| #include "remoting/host/chromoting_host_context.h" |
| #include "remoting/host/host_mock_objects.h" |
| #include "remoting/host/in_memory_host_config.h" |
| +#include "remoting/host/user_authenticator_fake.h" |
| #include "remoting/proto/video.pb.h" |
| #include "remoting/protocol/protocol_mock_objects.h" |
| #include "remoting/protocol/session_config.h" |
| @@ -38,6 +40,10 @@ namespace remoting { |
| namespace { |
| +UserAuthenticator* MakeUserAuthenticator() { |
| + return new UserAuthenticatorFake(); |
| +} |
| + |
| void PostQuitTask(MessageLoop* message_loop) { |
| message_loop->PostTask(FROM_HERE, new MessageLoop::QuitTask()); |
| } |
| @@ -53,6 +59,12 @@ ACTION_P(QuitMainMessageLoop, message_loop) { |
| PostQuitTask(message_loop); |
| } |
| +// A task that does nothing. |
| +class DummyTask : public Task { |
|
Alpha Left Google
2011/03/28 16:22:26
Alternatively you can also do:
void DummyTask() {
simonmorris
2011/03/29 17:07:51
Done.
|
| + public: |
| + void Run() {} |
| +}; |
| + |
| } // namespace |
| class ChromotingHostTest : public testing::Test { |
| @@ -83,6 +95,9 @@ class ChromotingHostTest : public testing::Test { |
| DesktopEnvironment* desktop = |
| new DesktopEnvironment(capturer, input_stub_); |
| host_ = ChromotingHost::Create(&context_, config_, desktop); |
| + credentials_.set_type(protocol::PASSWORD); |
| + credentials_.set_username("user"); |
| + credentials_.set_credential("password"); |
| connection_ = new MockConnectionToClient( |
| &message_loop_, &handler_, host_stub_, input_stub_); |
| connection2_ = new MockConnectionToClient( |
| @@ -140,8 +155,11 @@ class ChromotingHostTest : public testing::Test { |
| void SimulateClientConnection(int connection_index) { |
| scoped_refptr<MockConnectionToClient> connection = |
| (connection_index == 0) ? connection_ : connection2_; |
| - scoped_refptr<ClientSession> client = new ClientSession(host_.get(), |
| - connection); |
| + scoped_refptr<ClientSession> client = new ClientSession( |
| + host_.get(), |
| + base::Bind(MakeUserAuthenticator), |
| + connection, |
| + input_stub_); |
| connection->set_host_stub(client.get()); |
| context_.network_message_loop()->PostTask( |
| @@ -156,9 +174,10 @@ class ChromotingHostTest : public testing::Test { |
| connection)); |
| context_.network_message_loop()->PostTask( |
| FROM_HERE, |
| - NewRunnableMethod(host_.get(), |
| - &ChromotingHost::LocalLoginSucceeded, |
| - connection)); |
| + NewRunnableMethod(client.get(), |
| + &ClientSession::BeginSessionRequest, |
| + &credentials_, |
| + new DummyTask())); |
| } |
| // Helper method to remove a client connection from ChromotingHost. |
| @@ -176,6 +195,7 @@ class ChromotingHostTest : public testing::Test { |
| scoped_refptr<ChromotingHost> host_; |
| scoped_refptr<InMemoryHostConfig> config_; |
| MockChromotingHostContext context_; |
| + protocol::LocalLoginCredentials credentials_; |
| scoped_refptr<MockConnectionToClient> connection_; |
| scoped_refptr<MockSession> session_; |
| scoped_ptr<SessionConfig> session_config_; |