OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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/message_loop/message_loop.h" | 5 #include "base/message_loop/message_loop.h" |
6 #include "base/strings/stringprintf.h" | 6 #include "base/strings/stringprintf.h" |
7 #include "base/timer/mock_timer.h" | 7 #include "base/timer/mock_timer.h" |
8 #include "net/socket/stream_listen_socket.h" | 8 #include "net/socket/stream_listen_socket.h" |
9 #include "remoting/host/gnubby_auth_handler_posix.h" | 9 #include "remoting/host/gnubby_auth_handler_posix.h" |
10 #include "remoting/host/gnubby_socket.h" | 10 #include "remoting/host/gnubby_socket.h" |
(...skipping 85 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
96 EXPECT_CALL(client_stub_, | 96 EXPECT_CALL(client_stub_, |
97 DeliverHostMessage( | 97 DeliverHostMessage( |
98 EqualsDataMessage(42, "[116,101,115,116,95,109,115,103]"))); | 98 EqualsDataMessage(42, "[116,101,115,116,95,109,115,103]"))); |
99 | 99 |
100 auth_handler_->DeliverHostDataMessage(42, "test_msg"); | 100 auth_handler_->DeliverHostDataMessage(42, "test_msg"); |
101 } | 101 } |
102 | 102 |
103 TEST_F(GnubbyAuthHandlerPosixTest, DidClose) { | 103 TEST_F(GnubbyAuthHandlerPosixTest, DidClose) { |
104 net::StreamListenSocket* socket = new MockStreamListenSocket(delegate_); | 104 net::StreamListenSocket* socket = new MockStreamListenSocket(delegate_); |
105 | 105 |
106 delegate_->DidAccept(NULL, make_scoped_ptr(socket)); | 106 delegate_->DidAccept(nullptr, make_scoped_ptr(socket)); |
107 ASSERT_TRUE(auth_handler_posix_->HasActiveSocketForTesting(socket)); | 107 ASSERT_TRUE(auth_handler_posix_->HasActiveSocketForTesting(socket)); |
108 | 108 |
109 delegate_->DidClose(socket); | 109 delegate_->DidClose(socket); |
110 ASSERT_FALSE(auth_handler_posix_->HasActiveSocketForTesting(socket)); | 110 ASSERT_FALSE(auth_handler_posix_->HasActiveSocketForTesting(socket)); |
111 } | 111 } |
112 | 112 |
113 TEST_F(GnubbyAuthHandlerPosixTest, DidRead) { | 113 TEST_F(GnubbyAuthHandlerPosixTest, DidRead) { |
114 EXPECT_CALL(client_stub_, DeliverHostMessage(_)); | 114 EXPECT_CALL(client_stub_, DeliverHostMessage(_)); |
115 | 115 |
116 net::StreamListenSocket* socket = new MockStreamListenSocket(delegate_); | 116 net::StreamListenSocket* socket = new MockStreamListenSocket(delegate_); |
117 | 117 |
118 delegate_->DidAccept(NULL, make_scoped_ptr(socket)); | 118 delegate_->DidAccept(nullptr, make_scoped_ptr(socket)); |
119 delegate_->DidRead(socket, | 119 delegate_->DidRead(socket, |
120 reinterpret_cast<const char*>(request_data), | 120 reinterpret_cast<const char*>(request_data), |
121 sizeof(request_data)); | 121 sizeof(request_data)); |
122 } | 122 } |
123 | 123 |
124 TEST_F(GnubbyAuthHandlerPosixTest, DidReadByteByByte) { | 124 TEST_F(GnubbyAuthHandlerPosixTest, DidReadByteByByte) { |
125 EXPECT_CALL(client_stub_, DeliverHostMessage(_)); | 125 EXPECT_CALL(client_stub_, DeliverHostMessage(_)); |
126 | 126 |
127 net::StreamListenSocket* socket = new MockStreamListenSocket(delegate_); | 127 net::StreamListenSocket* socket = new MockStreamListenSocket(delegate_); |
128 | 128 |
129 delegate_->DidAccept(NULL, make_scoped_ptr(socket)); | 129 delegate_->DidAccept(nullptr, make_scoped_ptr(socket)); |
130 for (unsigned int i = 0; i < sizeof(request_data); ++i) { | 130 for (unsigned int i = 0; i < sizeof(request_data); ++i) { |
131 delegate_->DidRead( | 131 delegate_->DidRead( |
132 socket, reinterpret_cast<const char*>(request_data + i), 1); | 132 socket, reinterpret_cast<const char*>(request_data + i), 1); |
133 } | 133 } |
134 } | 134 } |
135 | 135 |
136 TEST_F(GnubbyAuthHandlerPosixTest, DidReadTimeout) { | 136 TEST_F(GnubbyAuthHandlerPosixTest, DidReadTimeout) { |
137 net::StreamListenSocket* socket = new MockStreamListenSocket(delegate_); | 137 net::StreamListenSocket* socket = new MockStreamListenSocket(delegate_); |
138 | 138 |
139 delegate_->DidAccept(NULL, make_scoped_ptr(socket)); | 139 delegate_->DidAccept(nullptr, make_scoped_ptr(socket)); |
140 ASSERT_TRUE(auth_handler_posix_->HasActiveSocketForTesting(socket)); | 140 ASSERT_TRUE(auth_handler_posix_->HasActiveSocketForTesting(socket)); |
141 | 141 |
142 base::MockTimer* mock_timer = new base::MockTimer(false, false); | 142 base::MockTimer* mock_timer = new base::MockTimer(false, false); |
143 auth_handler_posix_->GetGnubbySocketForTesting(socket) | 143 auth_handler_posix_->GetGnubbySocketForTesting(socket) |
144 ->SetTimerForTesting(make_scoped_ptr(mock_timer)); | 144 ->SetTimerForTesting(make_scoped_ptr(mock_timer)); |
145 delegate_->DidRead(socket, reinterpret_cast<const char*>(request_data), 1); | 145 delegate_->DidRead(socket, reinterpret_cast<const char*>(request_data), 1); |
146 mock_timer->Fire(); | 146 mock_timer->Fire(); |
147 | 147 |
148 ASSERT_FALSE(auth_handler_posix_->HasActiveSocketForTesting(socket)); | 148 ASSERT_FALSE(auth_handler_posix_->HasActiveSocketForTesting(socket)); |
149 } | 149 } |
150 | 150 |
151 TEST_F(GnubbyAuthHandlerPosixTest, ClientErrorMessageDelivered) { | 151 TEST_F(GnubbyAuthHandlerPosixTest, ClientErrorMessageDelivered) { |
152 net::StreamListenSocket* socket = new MockStreamListenSocket(delegate_); | 152 net::StreamListenSocket* socket = new MockStreamListenSocket(delegate_); |
153 | 153 |
154 delegate_->DidAccept(NULL, make_scoped_ptr(socket)); | 154 delegate_->DidAccept(nullptr, make_scoped_ptr(socket)); |
155 | 155 |
156 std::string error_json = base::StringPrintf( | 156 std::string error_json = base::StringPrintf( |
157 "{\"type\":\"error\",\"connectionId\":%d}", | 157 "{\"type\":\"error\",\"connectionId\":%d}", |
158 auth_handler_posix_->GetConnectionIdForTesting(socket)); | 158 auth_handler_posix_->GetConnectionIdForTesting(socket)); |
159 | 159 |
160 ASSERT_TRUE(auth_handler_posix_->HasActiveSocketForTesting(socket)); | 160 ASSERT_TRUE(auth_handler_posix_->HasActiveSocketForTesting(socket)); |
161 auth_handler_->DeliverClientMessage(error_json); | 161 auth_handler_->DeliverClientMessage(error_json); |
162 ASSERT_FALSE(auth_handler_posix_->HasActiveSocketForTesting(socket)); | 162 ASSERT_FALSE(auth_handler_posix_->HasActiveSocketForTesting(socket)); |
163 } | 163 } |
164 | 164 |
165 } // namespace remoting | 165 } // namespace remoting |
OLD | NEW |