OLD | NEW |
1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 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 <stddef.h> |
| 6 #include <stdint.h> |
| 7 #include <string.h> |
| 8 |
5 #include <string> | 9 #include <string> |
6 | 10 |
7 #include "base/bind.h" | 11 #include "base/bind.h" |
8 #include "base/guid.h" | 12 #include "base/guid.h" |
9 #include "base/location.h" | 13 #include "base/location.h" |
10 #include "base/single_thread_task_runner.h" | 14 #include "base/single_thread_task_runner.h" |
11 #include "base/sync_socket.h" | 15 #include "base/sync_socket.h" |
12 #include "base/synchronization/waitable_event.h" | 16 #include "base/synchronization/waitable_event.h" |
13 #include "base/threading/thread.h" | 17 #include "base/threading/thread.h" |
14 #include "base/time/time.h" | 18 #include "base/time/time.h" |
| 19 #include "build/build_config.h" |
15 #include "chrome/test/chromedriver/chrome/status.h" | 20 #include "chrome/test/chromedriver/chrome/status.h" |
16 #include "chrome/test/chromedriver/net/port_server.h" | 21 #include "chrome/test/chromedriver/net/port_server.h" |
17 #include "net/base/sys_addrinfo.h" | 22 #include "net/base/sys_addrinfo.h" |
18 #include "testing/gtest/include/gtest/gtest.h" | 23 #include "testing/gtest/include/gtest/gtest.h" |
19 | 24 |
20 #if defined(OS_LINUX) | 25 #if defined(OS_LINUX) |
21 #include <fcntl.h> | 26 #include <fcntl.h> |
22 #include <sys/socket.h> | 27 #include <sys/socket.h> |
23 #include <sys/un.h> | 28 #include <sys/un.h> |
24 #endif | 29 #endif |
(...skipping 109 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
134 base::Thread thread_; | 139 base::Thread thread_; |
135 }; | 140 }; |
136 | 141 |
137 TEST_F(PortServerTest, Reserve) { | 142 TEST_F(PortServerTest, Reserve) { |
138 std::string path = GenerateRandomPath(); | 143 std::string path = GenerateRandomPath(); |
139 PortServer server(path); | 144 PortServer server(path); |
140 | 145 |
141 std::string request; | 146 std::string request; |
142 RunServer(path, "12345\n", &request); | 147 RunServer(path, "12345\n", &request); |
143 | 148 |
144 uint16 port = 0; | 149 uint16_t port = 0; |
145 scoped_ptr<PortReservation> reservation; | 150 scoped_ptr<PortReservation> reservation; |
146 Status status = server.ReservePort(&port, &reservation); | 151 Status status = server.ReservePort(&port, &reservation); |
147 ASSERT_EQ(kOk, status.code()) << status.message(); | 152 ASSERT_EQ(kOk, status.code()) << status.message(); |
148 ASSERT_EQ(12345u, port); | 153 ASSERT_EQ(12345u, port); |
149 } | 154 } |
150 | 155 |
151 TEST_F(PortServerTest, ReserveResetReserve) { | 156 TEST_F(PortServerTest, ReserveResetReserve) { |
152 std::string path = GenerateRandomPath(); | 157 std::string path = GenerateRandomPath(); |
153 PortServer server(path); | 158 PortServer server(path); |
154 | 159 |
155 std::string request; | 160 std::string request; |
156 RunServer(path, "12345\n", &request); | 161 RunServer(path, "12345\n", &request); |
157 | 162 |
158 uint16 port = 0; | 163 uint16_t port = 0; |
159 scoped_ptr<PortReservation> reservation; | 164 scoped_ptr<PortReservation> reservation; |
160 Status status = server.ReservePort(&port, &reservation); | 165 Status status = server.ReservePort(&port, &reservation); |
161 ASSERT_EQ(kOk, status.code()) << status.message(); | 166 ASSERT_EQ(kOk, status.code()) << status.message(); |
162 ASSERT_EQ(12345u, port); | 167 ASSERT_EQ(12345u, port); |
163 | 168 |
164 reservation.reset(); | 169 reservation.reset(); |
165 status = server.ReservePort(&port, &reservation); | 170 status = server.ReservePort(&port, &reservation); |
166 ASSERT_EQ(kOk, status.code()) << status.message(); | 171 ASSERT_EQ(kOk, status.code()) << status.message(); |
167 ASSERT_EQ(12345u, port); | 172 ASSERT_EQ(12345u, port); |
168 } | 173 } |
169 | 174 |
170 TEST_F(PortServerTest, ReserveReserve) { | 175 TEST_F(PortServerTest, ReserveReserve) { |
171 std::string path = GenerateRandomPath(); | 176 std::string path = GenerateRandomPath(); |
172 PortServer server(path); | 177 PortServer server(path); |
173 | 178 |
174 std::string request; | 179 std::string request; |
175 RunServer(path, "12345\n", &request); | 180 RunServer(path, "12345\n", &request); |
176 | 181 |
177 uint16 port = 0; | 182 uint16_t port = 0; |
178 scoped_ptr<PortReservation> reservation; | 183 scoped_ptr<PortReservation> reservation; |
179 Status status = server.ReservePort(&port, &reservation); | 184 Status status = server.ReservePort(&port, &reservation); |
180 ASSERT_EQ(kOk, status.code()) << status.message(); | 185 ASSERT_EQ(kOk, status.code()) << status.message(); |
181 ASSERT_EQ(12345u, port); | 186 ASSERT_EQ(12345u, port); |
182 | 187 |
183 RunServer(path, "12346\n", &request); | 188 RunServer(path, "12346\n", &request); |
184 status = server.ReservePort(&port, &reservation); | 189 status = server.ReservePort(&port, &reservation); |
185 ASSERT_EQ(kOk, status.code()) << status.message(); | 190 ASSERT_EQ(kOk, status.code()) << status.message(); |
186 ASSERT_EQ(12346u, port); | 191 ASSERT_EQ(12346u, port); |
187 } | 192 } |
188 #endif | 193 #endif |
189 | 194 |
190 TEST(PortManagerTest, ReservePort) { | 195 TEST(PortManagerTest, ReservePort) { |
191 PortManager mgr(15000, 16000); | 196 PortManager mgr(15000, 16000); |
192 uint16 port = 0; | 197 uint16_t port = 0; |
193 scoped_ptr<PortReservation> reservation; | 198 scoped_ptr<PortReservation> reservation; |
194 Status status = mgr.ReservePort(&port, &reservation); | 199 Status status = mgr.ReservePort(&port, &reservation); |
195 ASSERT_EQ(kOk, status.code()) << status.message(); | 200 ASSERT_EQ(kOk, status.code()) << status.message(); |
196 | 201 |
197 ASSERT_GE(port, 15000); | 202 ASSERT_GE(port, 15000); |
198 ASSERT_LE(port, 16000); | 203 ASSERT_LE(port, 16000); |
199 ASSERT_TRUE(reservation); | 204 ASSERT_TRUE(reservation); |
200 } | 205 } |
201 | 206 |
202 TEST(PortManagerTest, ReservePortFromPool) { | 207 TEST(PortManagerTest, ReservePortFromPool) { |
203 PortManager mgr(15000, 16000); | 208 PortManager mgr(15000, 16000); |
204 uint16 first_port = 0, port = 1; | 209 uint16_t first_port = 0, port = 1; |
205 for (int i = 0; i < 10; i++) { | 210 for (int i = 0; i < 10; i++) { |
206 scoped_ptr<PortReservation> reservation; | 211 scoped_ptr<PortReservation> reservation; |
207 Status status = mgr.ReservePortFromPool(&port, &reservation); | 212 Status status = mgr.ReservePortFromPool(&port, &reservation); |
208 ASSERT_EQ(kOk, status.code()) << status.message(); | 213 ASSERT_EQ(kOk, status.code()) << status.message(); |
209 ASSERT_TRUE(reservation); | 214 ASSERT_TRUE(reservation); |
210 ASSERT_GE(port, 15000); | 215 ASSERT_GE(port, 15000); |
211 ASSERT_LE(port, 16000); | 216 ASSERT_LE(port, 16000); |
212 if (i == 0) | 217 if (i == 0) |
213 first_port = port; | 218 first_port = port; |
214 ASSERT_EQ(port, first_port); | 219 ASSERT_EQ(port, first_port); |
215 } | 220 } |
216 } | 221 } |
OLD | NEW |