Index: remoting/host/client_session_unittest.cc |
diff --git a/remoting/host/client_session_unittest.cc b/remoting/host/client_session_unittest.cc |
index 5ce97af55dafa1f0bd4c84bcc14a8ce7aad2d53d..bd6860ad7cf01d1dc4c5773cf1ab91526ed75f8d 100644 |
--- a/remoting/host/client_session_unittest.cc |
+++ b/remoting/host/client_session_unittest.cc |
@@ -2,10 +2,8 @@ |
// 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 "remoting/host/client_session.h" |
#include "remoting/host/host_mock_objects.h" |
-#include "remoting/host/user_authenticator_fake.h" |
#include "remoting/protocol/protocol_mock_objects.h" |
#include "testing/gtest/include/gtest/gtest.h" |
@@ -13,10 +11,6 @@ namespace remoting { |
namespace { |
-UserAuthenticator* MakeUserAuthenticator() { |
- return new UserAuthenticatorFake(); |
-} |
- |
// A task that does nothing. |
class DummyTask : public Task { |
public: |
@@ -33,41 +27,37 @@ using protocol::MockInputStub; |
using testing::_; |
using testing::DeleteArg; |
using testing::InSequence; |
+using testing::Return; |
class ClientSessionTest : public testing::Test { |
public: |
- ClientSessionTest() { |
- } |
+ ClientSessionTest() {} |
virtual void SetUp() { |
connection_ = new MockConnectionToClient(&message_loop_, |
&connection_event_handler_, |
&host_stub_, |
&input_stub_); |
- client_session_ = new ClientSession(&session_event_handler_, |
- base::Bind(MakeUserAuthenticator), |
- connection_, |
- &input_stub_); |
- credentials_.set_type(protocol::PASSWORD); |
- credentials_.set_username("user"); |
- credentials_.set_credential("password"); |
+ user_authenticator_ = new MockUserAuthenticator(); |
+ client_session_ = new ClientSession( |
+ &session_event_handler_, |
+ user_authenticator_, |
+ connection_, |
+ &input_stub_); |
ON_CALL(input_stub_, InjectKeyEvent(_, _)).WillByDefault(DeleteArg<1>()); |
ON_CALL(input_stub_, InjectMouseEvent(_, _)).WillByDefault(DeleteArg<1>()); |
} |
- virtual void TearDown() { |
- } |
- |
protected: |
MessageLoop message_loop_; |
MockConnectionToClientEventHandler connection_event_handler_; |
MockHostStub host_stub_; |
MockInputStub input_stub_; |
MockClientSessionEventHandler session_event_handler_; |
+ MockUserAuthenticator* user_authenticator_; |
scoped_refptr<MockConnectionToClient> connection_; |
scoped_refptr<ClientSession> client_session_; |
- protocol::LocalLoginCredentials credentials_; |
}; |
TEST_F(ClientSessionTest, InputStubFilter) { |
@@ -95,11 +85,14 @@ TEST_F(ClientSessionTest, InputStubFilter) { |
mouse_event3.set_x(300); |
mouse_event3.set_y(301); |
- credentials_.set_type(protocol::PASSWORD); |
- credentials_.set_username("user"); |
- credentials_.set_credential("password"); |
+ protocol::LocalLoginCredentials credentials; |
+ credentials.set_type(protocol::PASSWORD); |
+ credentials.set_username("user"); |
+ credentials.set_credential("password"); |
InSequence s; |
+ EXPECT_CALL(*user_authenticator_, Authenticate(_, _)) |
+ .WillOnce(Return(true)); |
EXPECT_CALL(session_event_handler_, LocalLoginSucceeded(_)); |
EXPECT_CALL(input_stub_, InjectKeyEvent(&key_event2, _)); |
EXPECT_CALL(input_stub_, InjectMouseEvent(&mouse_event2, _)); |
@@ -109,7 +102,7 @@ TEST_F(ClientSessionTest, InputStubFilter) { |
// because the client isn't authenticated yet. |
client_session_->InjectKeyEvent(&key_event1, new DummyTask()); |
client_session_->InjectMouseEvent(&mouse_event1, new DummyTask()); |
- client_session_->BeginSessionRequest(&credentials_, new DummyTask()); |
+ client_session_->BeginSessionRequest(&credentials, new DummyTask()); |
// These events should get through to the input stub. |
client_session_->InjectKeyEvent(&key_event2, new DummyTask()); |
client_session_->InjectMouseEvent(&mouse_event2, new DummyTask()); |