OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 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 <memory> | 5 #include <memory> |
6 #include <queue> | 6 #include <queue> |
7 #include <vector> | 7 #include <vector> |
8 | 8 |
9 #include "base/location.h" | 9 #include "base/location.h" |
10 #include "base/macros.h" | 10 #include "base/macros.h" |
11 #include "base/memory/ptr_util.h" | 11 #include "base/memory/ptr_util.h" |
12 #include "base/memory/ref_counted.h" | 12 #include "base/memory/ref_counted.h" |
13 #include "base/message_loop/message_loop.h" | 13 #include "base/message_loop/message_loop.h" |
| 14 #include "base/run_loop.h" |
14 #include "base/single_thread_task_runner.h" | 15 #include "base/single_thread_task_runner.h" |
15 #include "base/threading/thread_task_runner_handle.h" | 16 #include "base/threading/thread_task_runner_handle.h" |
16 #include "base/time/clock.h" | 17 #include "base/time/clock.h" |
17 #include "base/time/default_clock.h" | 18 #include "base/time/default_clock.h" |
18 #include "base/timer/mock_timer.h" | 19 #include "base/timer/mock_timer.h" |
19 #include "net/base/ip_address.h" | 20 #include "net/base/ip_address.h" |
20 #include "net/base/rand_callback.h" | 21 #include "net/base/rand_callback.h" |
21 #include "net/base/test_completion_callback.h" | 22 #include "net/base/test_completion_callback.h" |
22 #include "net/dns/mdns_client_impl.h" | 23 #include "net/dns/mdns_client_impl.h" |
23 #include "net/dns/mock_mdns_socket_factory.h" | 24 #include "net/dns/mock_mdns_socket_factory.h" |
(...skipping 434 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
458 listener1_.reset(); | 459 listener1_.reset(); |
459 listener2_.reset(); | 460 listener2_.reset(); |
460 } | 461 } |
461 | 462 |
462 void MDnsTest::RunFor(base::TimeDelta time_period) { | 463 void MDnsTest::RunFor(base::TimeDelta time_period) { |
463 base::CancelableCallback<void()> callback(base::Bind(&MDnsTest::Stop, | 464 base::CancelableCallback<void()> callback(base::Bind(&MDnsTest::Stop, |
464 base::Unretained(this))); | 465 base::Unretained(this))); |
465 base::ThreadTaskRunnerHandle::Get()->PostDelayedTask( | 466 base::ThreadTaskRunnerHandle::Get()->PostDelayedTask( |
466 FROM_HERE, callback.callback(), time_period); | 467 FROM_HERE, callback.callback(), time_period); |
467 | 468 |
468 base::MessageLoop::current()->Run(); | 469 base::RunLoop().Run(); |
469 callback.Cancel(); | 470 callback.Cancel(); |
470 } | 471 } |
471 | 472 |
472 void MDnsTest::Stop() { | 473 void MDnsTest::Stop() { |
473 base::MessageLoop::current()->QuitWhenIdle(); | 474 base::MessageLoop::current()->QuitWhenIdle(); |
474 } | 475 } |
475 | 476 |
476 TEST_F(MDnsTest, PassiveListeners) { | 477 TEST_F(MDnsTest, PassiveListeners) { |
477 StrictMock<MockListenerDelegate> delegate_privet; | 478 StrictMock<MockListenerDelegate> delegate_privet; |
478 StrictMock<MockListenerDelegate> delegate_printer; | 479 StrictMock<MockListenerDelegate> delegate_printer; |
(...skipping 699 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1178 EXPECT_CALL(*socket_ipv6_, RecvFrom(_, _, _, _)) | 1179 EXPECT_CALL(*socket_ipv6_, RecvFrom(_, _, _, _)) |
1179 .Times(2) | 1180 .Times(2) |
1180 .WillOnce( | 1181 .WillOnce( |
1181 Invoke(socket_ipv6_, &MockMDnsDatagramServerSocket::HandleRecvLater)) | 1182 Invoke(socket_ipv6_, &MockMDnsDatagramServerSocket::HandleRecvLater)) |
1182 .WillOnce(Return(ERR_IO_PENDING)); | 1183 .WillOnce(Return(ERR_IO_PENDING)); |
1183 | 1184 |
1184 ASSERT_TRUE(InitConnection()); | 1185 ASSERT_TRUE(InitConnection()); |
1185 | 1186 |
1186 EXPECT_CALL(delegate_, HandlePacketInternal(sample_packet_)); | 1187 EXPECT_CALL(delegate_, HandlePacketInternal(sample_packet_)); |
1187 | 1188 |
1188 base::MessageLoop::current()->RunUntilIdle(); | 1189 base::RunLoop().RunUntilIdle(); |
1189 } | 1190 } |
1190 | 1191 |
1191 TEST_F(MDnsConnectionTest, Error) { | 1192 TEST_F(MDnsConnectionTest, Error) { |
1192 CompletionCallback callback; | 1193 CompletionCallback callback; |
1193 | 1194 |
1194 EXPECT_CALL(*socket_ipv4_, RecvFrom(_, _, _, _)) | 1195 EXPECT_CALL(*socket_ipv4_, RecvFrom(_, _, _, _)) |
1195 .WillOnce(Return(ERR_IO_PENDING)); | 1196 .WillOnce(Return(ERR_IO_PENDING)); |
1196 EXPECT_CALL(*socket_ipv6_, RecvFrom(_, _, _, _)) | 1197 EXPECT_CALL(*socket_ipv6_, RecvFrom(_, _, _, _)) |
1197 .WillOnce(DoAll(SaveArg<3>(&callback), Return(ERR_IO_PENDING))); | 1198 .WillOnce(DoAll(SaveArg<3>(&callback), Return(ERR_IO_PENDING))); |
1198 | 1199 |
1199 ASSERT_TRUE(InitConnection()); | 1200 ASSERT_TRUE(InitConnection()); |
1200 | 1201 |
1201 EXPECT_CALL(delegate_, OnConnectionError(ERR_SOCKET_NOT_CONNECTED)); | 1202 EXPECT_CALL(delegate_, OnConnectionError(ERR_SOCKET_NOT_CONNECTED)); |
1202 callback.Run(ERR_SOCKET_NOT_CONNECTED); | 1203 callback.Run(ERR_SOCKET_NOT_CONNECTED); |
1203 base::MessageLoop::current()->RunUntilIdle(); | 1204 base::RunLoop().RunUntilIdle(); |
1204 } | 1205 } |
1205 | 1206 |
1206 class MDnsConnectionSendTest : public MDnsConnectionTest { | 1207 class MDnsConnectionSendTest : public MDnsConnectionTest { |
1207 protected: | 1208 protected: |
1208 void SetUp() override { | 1209 void SetUp() override { |
1209 MDnsConnectionTest::SetUp(); | 1210 MDnsConnectionTest::SetUp(); |
1210 EXPECT_CALL(*socket_ipv4_, RecvFrom(_, _, _, _)) | 1211 EXPECT_CALL(*socket_ipv4_, RecvFrom(_, _, _, _)) |
1211 .WillOnce(Return(ERR_IO_PENDING)); | 1212 .WillOnce(Return(ERR_IO_PENDING)); |
1212 EXPECT_CALL(*socket_ipv6_, RecvFrom(_, _, _, _)) | 1213 EXPECT_CALL(*socket_ipv6_, RecvFrom(_, _, _, _)) |
1213 .WillOnce(Return(ERR_IO_PENDING)); | 1214 .WillOnce(Return(ERR_IO_PENDING)); |
(...skipping 14 matching lines...) Expand all Loading... |
1228 CompletionCallback callback; | 1229 CompletionCallback callback; |
1229 | 1230 |
1230 EXPECT_CALL(*socket_ipv4_, | 1231 EXPECT_CALL(*socket_ipv4_, |
1231 SendToInternal(sample_packet_, "224.0.0.251:5353", _)); | 1232 SendToInternal(sample_packet_, "224.0.0.251:5353", _)); |
1232 EXPECT_CALL(*socket_ipv6_, | 1233 EXPECT_CALL(*socket_ipv6_, |
1233 SendToInternal(sample_packet_, "[ff02::fb]:5353", _)) | 1234 SendToInternal(sample_packet_, "[ff02::fb]:5353", _)) |
1234 .WillOnce(DoAll(SaveArg<2>(&callback), Return(ERR_SOCKET_NOT_CONNECTED))); | 1235 .WillOnce(DoAll(SaveArg<2>(&callback), Return(ERR_SOCKET_NOT_CONNECTED))); |
1235 | 1236 |
1236 connection_.Send(sample_buffer_, sample_packet_.size()); | 1237 connection_.Send(sample_buffer_, sample_packet_.size()); |
1237 EXPECT_CALL(delegate_, OnConnectionError(ERR_SOCKET_NOT_CONNECTED)); | 1238 EXPECT_CALL(delegate_, OnConnectionError(ERR_SOCKET_NOT_CONNECTED)); |
1238 base::MessageLoop::current()->RunUntilIdle(); | 1239 base::RunLoop().RunUntilIdle(); |
1239 } | 1240 } |
1240 | 1241 |
1241 TEST_F(MDnsConnectionSendTest, SendQueued) { | 1242 TEST_F(MDnsConnectionSendTest, SendQueued) { |
1242 // Send data immediately. | 1243 // Send data immediately. |
1243 EXPECT_CALL(*socket_ipv4_, | 1244 EXPECT_CALL(*socket_ipv4_, |
1244 SendToInternal(sample_packet_, "224.0.0.251:5353", _)) | 1245 SendToInternal(sample_packet_, "224.0.0.251:5353", _)) |
1245 .Times(2) | 1246 .Times(2) |
1246 .WillRepeatedly(Return(OK)); | 1247 .WillRepeatedly(Return(OK)); |
1247 | 1248 |
1248 CompletionCallback callback; | 1249 CompletionCallback callback; |
(...skipping 10 matching lines...) Expand all Loading... |
1259 SendToInternal(sample_packet_, "224.0.0.251:5353", _)) | 1260 SendToInternal(sample_packet_, "224.0.0.251:5353", _)) |
1260 .Times(0); | 1261 .Times(0); |
1261 // Expect call for the second IPv6 packed. | 1262 // Expect call for the second IPv6 packed. |
1262 EXPECT_CALL(*socket_ipv6_, | 1263 EXPECT_CALL(*socket_ipv6_, |
1263 SendToInternal(sample_packet_, "[ff02::fb]:5353", _)) | 1264 SendToInternal(sample_packet_, "[ff02::fb]:5353", _)) |
1264 .WillOnce(Return(OK)); | 1265 .WillOnce(Return(OK)); |
1265 callback.Run(OK); | 1266 callback.Run(OK); |
1266 } | 1267 } |
1267 | 1268 |
1268 } // namespace net | 1269 } // namespace net |
OLD | NEW |