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 "net/tools/quic/quic_time_wait_list_manager.h" | 5 #include "net/tools/quic/quic_time_wait_list_manager.h" |
6 | 6 |
7 #include <errno.h> | 7 #include <errno.h> |
8 #include <memory> | 8 #include <memory> |
9 | 9 |
10 #include "net/quic/crypto/crypto_protocol.h" | 10 #include "net/quic/crypto/crypto_protocol.h" |
11 #include "net/quic/crypto/null_encrypter.h" | 11 #include "net/quic/crypto/null_encrypter.h" |
12 #include "net/quic/crypto/quic_decrypter.h" | 12 #include "net/quic/crypto/quic_decrypter.h" |
13 #include "net/quic/crypto/quic_encrypter.h" | 13 #include "net/quic/crypto/quic_encrypter.h" |
14 #include "net/quic/quic_chromium_connection_helper.h" | 14 #include "net/quic/quic_chromium_connection_helper.h" |
15 #include "net/quic/quic_data_reader.h" | 15 #include "net/quic/quic_data_reader.h" |
16 #include "net/quic/quic_flags.h" | 16 #include "net/quic/quic_flags.h" |
17 #include "net/quic/quic_framer.h" | 17 #include "net/quic/quic_framer.h" |
18 #include "net/quic/quic_packet_writer.h" | 18 #include "net/quic/quic_packet_writer.h" |
19 #include "net/quic/quic_protocol.h" | 19 #include "net/quic/quic_protocol.h" |
20 #include "net/quic/quic_utils.h" | 20 #include "net/quic/quic_utils.h" |
21 #include "net/quic/test_tools/quic_test_utils.h" | 21 #include "net/quic/test_tools/quic_test_utils.h" |
| 22 #include "net/tools/quic/quic_epoll_alarm_factory.h" |
22 #include "net/tools/quic/quic_epoll_connection_helper.h" | 23 #include "net/tools/quic/quic_epoll_connection_helper.h" |
23 #include "net/tools/quic/test_tools/mock_epoll_server.h" | 24 #include "net/tools/quic/test_tools/mock_epoll_server.h" |
24 #include "net/tools/quic/test_tools/mock_quic_server_session_visitor.h" | 25 #include "net/tools/quic/test_tools/mock_quic_server_session_visitor.h" |
25 #include "testing/gmock/include/gmock/gmock.h" | 26 #include "testing/gmock/include/gmock/gmock.h" |
26 #include "testing/gtest/include/gtest/gtest.h" | 27 #include "testing/gtest/include/gtest/gtest.h" |
27 | 28 |
28 using net::test::kTestPort; | 29 using net::test::kTestPort; |
29 using net::test::BuildUnsizedDataPacket; | 30 using net::test::BuildUnsizedDataPacket; |
30 using net::test::NoOpFramerVisitor; | 31 using net::test::NoOpFramerVisitor; |
31 using net::test::QuicVersionMax; | 32 using net::test::QuicVersionMax; |
(...skipping 57 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
89 class MockFakeTimeEpollServer : public FakeTimeEpollServer { | 90 class MockFakeTimeEpollServer : public FakeTimeEpollServer { |
90 public: | 91 public: |
91 MOCK_METHOD2(RegisterAlarm, | 92 MOCK_METHOD2(RegisterAlarm, |
92 void(int64_t timeout_in_us, EpollAlarmCallbackInterface* alarm)); | 93 void(int64_t timeout_in_us, EpollAlarmCallbackInterface* alarm)); |
93 }; | 94 }; |
94 | 95 |
95 class QuicTimeWaitListManagerTest : public ::testing::Test { | 96 class QuicTimeWaitListManagerTest : public ::testing::Test { |
96 protected: | 97 protected: |
97 QuicTimeWaitListManagerTest() | 98 QuicTimeWaitListManagerTest() |
98 : helper_(&epoll_server_, QuicAllocator::BUFFER_POOL), | 99 : helper_(&epoll_server_, QuicAllocator::BUFFER_POOL), |
99 time_wait_list_manager_(&writer_, &visitor_, &helper_), | 100 alarm_factory_(&epoll_server_), |
| 101 time_wait_list_manager_(&writer_, &visitor_, &helper_, &alarm_factory_), |
100 connection_id_(45), | 102 connection_id_(45), |
101 client_address_(net::test::TestPeerIPAddress(), kTestPort), | 103 client_address_(net::test::TestPeerIPAddress(), kTestPort), |
102 writer_is_blocked_(false) {} | 104 writer_is_blocked_(false) {} |
103 | 105 |
104 ~QuicTimeWaitListManagerTest() override {} | 106 ~QuicTimeWaitListManagerTest() override {} |
105 | 107 |
106 void SetUp() override { | 108 void SetUp() override { |
107 EXPECT_CALL(writer_, IsWriteBlocked()) | 109 EXPECT_CALL(writer_, IsWriteBlocked()) |
108 .WillRepeatedly(ReturnPointee(&writer_is_blocked_)); | 110 .WillRepeatedly(ReturnPointee(&writer_is_blocked_)); |
109 EXPECT_CALL(writer_, IsWriteBlockedDataBuffered()) | 111 EXPECT_CALL(writer_, IsWriteBlockedDataBuffered()) |
(...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
145 QuicEncryptedPacket* ConstructEncryptedPacket( | 147 QuicEncryptedPacket* ConstructEncryptedPacket( |
146 QuicConnectionId connection_id, | 148 QuicConnectionId connection_id, |
147 QuicPacketNumber packet_number) { | 149 QuicPacketNumber packet_number) { |
148 return net::test::ConstructEncryptedPacket(connection_id, false, false, | 150 return net::test::ConstructEncryptedPacket(connection_id, false, false, |
149 false, kDefaultPathId, | 151 false, kDefaultPathId, |
150 packet_number, "data"); | 152 packet_number, "data"); |
151 } | 153 } |
152 | 154 |
153 NiceMock<MockFakeTimeEpollServer> epoll_server_; | 155 NiceMock<MockFakeTimeEpollServer> epoll_server_; |
154 QuicEpollConnectionHelper helper_; | 156 QuicEpollConnectionHelper helper_; |
| 157 QuicEpollAlarmFactory alarm_factory_; |
155 StrictMock<MockPacketWriter> writer_; | 158 StrictMock<MockPacketWriter> writer_; |
156 StrictMock<MockQuicServerSessionVisitor> visitor_; | 159 StrictMock<MockQuicServerSessionVisitor> visitor_; |
157 QuicTimeWaitListManager time_wait_list_manager_; | 160 QuicTimeWaitListManager time_wait_list_manager_; |
158 QuicConnectionId connection_id_; | 161 QuicConnectionId connection_id_; |
159 IPEndPoint server_address_; | 162 IPEndPoint server_address_; |
160 IPEndPoint client_address_; | 163 IPEndPoint client_address_; |
161 bool writer_is_blocked_; | 164 bool writer_is_blocked_; |
162 }; | 165 }; |
163 | 166 |
164 class ValidatePublicResetPacketPredicate | 167 class ValidatePublicResetPacketPredicate |
(...skipping 387 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
552 EXPECT_EQ(static_cast<size_t>(FLAGS_quic_time_wait_list_max_connections), | 555 EXPECT_EQ(static_cast<size_t>(FLAGS_quic_time_wait_list_max_connections), |
553 time_wait_list_manager_.num_connections()); | 556 time_wait_list_manager_.num_connections()); |
554 EXPECT_FALSE(IsConnectionIdInTimeWait(id_to_evict)); | 557 EXPECT_FALSE(IsConnectionIdInTimeWait(id_to_evict)); |
555 EXPECT_TRUE(IsConnectionIdInTimeWait(current_connection_id)); | 558 EXPECT_TRUE(IsConnectionIdInTimeWait(current_connection_id)); |
556 } | 559 } |
557 } | 560 } |
558 | 561 |
559 } // namespace | 562 } // namespace |
560 } // namespace test | 563 } // namespace test |
561 } // namespace net | 564 } // namespace net |
OLD | NEW |