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_server.h" | 5 #include "net/tools/quic/quic_server.h" |
6 | 6 |
7 #include "net/quic/core/crypto/quic_random.h" | 7 #include "net/quic/core/crypto/quic_random.h" |
8 #include "net/quic/core/quic_utils.h" | 8 #include "net/quic/core/quic_utils.h" |
9 #include "net/quic/platform/api/quic_flags.h" | 9 #include "net/quic/platform/api/quic_flags.h" |
10 #include "net/quic/platform/api/quic_logging.h" | 10 #include "net/quic/platform/api/quic_logging.h" |
11 #include "net/quic/platform/api/quic_socket_address.h" | 11 #include "net/quic/platform/api/quic_socket_address.h" |
| 12 #include "net/quic/platform/api/quic_test.h" |
12 #include "net/quic/test_tools/crypto_test_utils.h" | 13 #include "net/quic/test_tools/crypto_test_utils.h" |
13 #include "net/quic/test_tools/mock_quic_dispatcher.h" | 14 #include "net/quic/test_tools/mock_quic_dispatcher.h" |
14 #include "net/tools/quic/quic_epoll_alarm_factory.h" | 15 #include "net/tools/quic/quic_epoll_alarm_factory.h" |
15 #include "net/tools/quic/quic_epoll_connection_helper.h" | 16 #include "net/tools/quic/quic_epoll_connection_helper.h" |
16 #include "net/tools/quic/quic_simple_crypto_server_stream_helper.h" | 17 #include "net/tools/quic/quic_simple_crypto_server_stream_helper.h" |
17 #include "net/tools/quic/test_tools/quic_server_peer.h" | 18 #include "net/tools/quic/test_tools/quic_server_peer.h" |
18 #include "testing/gtest/include/gtest/gtest.h" | |
19 | 19 |
20 using ::testing::_; | 20 using ::testing::_; |
21 | 21 |
22 namespace net { | 22 namespace net { |
23 namespace test { | 23 namespace test { |
24 | 24 |
25 namespace { | 25 namespace { |
26 | 26 |
27 class MockQuicSimpleDispatcher : public QuicSimpleDispatcher { | 27 class MockQuicSimpleDispatcher : public QuicSimpleDispatcher { |
28 public: | 28 public: |
(...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
71 std::unique_ptr<QuicEpollAlarmFactory>( | 71 std::unique_ptr<QuicEpollAlarmFactory>( |
72 new QuicEpollAlarmFactory(epoll_server())), | 72 new QuicEpollAlarmFactory(epoll_server())), |
73 &response_cache_); | 73 &response_cache_); |
74 return mock_dispatcher_; | 74 return mock_dispatcher_; |
75 } | 75 } |
76 | 76 |
77 MockQuicSimpleDispatcher* mock_dispatcher_; | 77 MockQuicSimpleDispatcher* mock_dispatcher_; |
78 QuicHttpResponseCache response_cache_; | 78 QuicHttpResponseCache response_cache_; |
79 }; | 79 }; |
80 | 80 |
81 class QuicServerEpollInTest : public ::testing::Test { | 81 class QuicServerEpollInTest : public QuicTest { |
82 public: | 82 public: |
83 QuicServerEpollInTest() | 83 QuicServerEpollInTest() |
84 : port_(net::test::kTestPort), | 84 : port_(net::test::kTestPort), |
85 server_address_(QuicIpAddress::Loopback4(), port_) {} | 85 server_address_(QuicIpAddress::Loopback4(), port_) {} |
86 | 86 |
87 void StartListening() { | 87 void StartListening() { |
88 server_.CreateUDPSocketAndListen(server_address_); | 88 server_.CreateUDPSocketAndListen(server_address_); |
89 ASSERT_TRUE(QuicServerPeer::SetSmallSocket(&server_)); | 89 ASSERT_TRUE(QuicServerPeer::SetSmallSocket(&server_)); |
90 | 90 |
91 if (!server_.overflow_supported()) { | 91 if (!server_.overflow_supported()) { |
92 QUIC_LOG(WARNING) << "Overflow not supported. Not testing."; | 92 QUIC_LOG(WARNING) << "Overflow not supported. Not testing."; |
93 return; | 93 return; |
94 } | 94 } |
95 } | 95 } |
96 | 96 |
97 protected: | 97 protected: |
98 QuicFlagSaver saver_; | |
99 int port_; | 98 int port_; |
100 QuicSocketAddress server_address_; | 99 QuicSocketAddress server_address_; |
101 TestQuicServer server_; | 100 TestQuicServer server_; |
102 }; | 101 }; |
103 | 102 |
104 // Tests that if dispatcher has CHLOs waiting for connection creation, EPOLLIN | 103 // Tests that if dispatcher has CHLOs waiting for connection creation, EPOLLIN |
105 // event should try to create connections for them. And set epoll mask with | 104 // event should try to create connections for them. And set epoll mask with |
106 // EPOLLIN if there are still CHLOs remaining at the end of epoll event. | 105 // EPOLLIN if there are still CHLOs remaining at the end of epoll event. |
107 TEST_F(QuicServerEpollInTest, ProcessBufferedCHLOsOnEpollin) { | 106 TEST_F(QuicServerEpollInTest, ProcessBufferedCHLOsOnEpollin) { |
108 FLAGS_quic_reloadable_flag_quic_limit_num_new_sessions_per_epoll_loop = true; | 107 FLAGS_quic_reloadable_flag_quic_limit_num_new_sessions_per_epoll_loop = true; |
(...skipping 26 matching lines...) Expand all Loading... |
135 reinterpret_cast<sockaddr*>(&storage), sizeof(storage)); | 134 reinterpret_cast<sockaddr*>(&storage), sizeof(storage)); |
136 if (rc < 0) { | 135 if (rc < 0) { |
137 QUIC_DLOG(INFO) << errno << " " << strerror(errno); | 136 QUIC_DLOG(INFO) << errno << " " << strerror(errno); |
138 } | 137 } |
139 | 138 |
140 while (more_chlos) { | 139 while (more_chlos) { |
141 server_.WaitForEvents(); | 140 server_.WaitForEvents(); |
142 } | 141 } |
143 } | 142 } |
144 | 143 |
145 class QuicServerDispatchPacketTest : public ::testing::Test { | 144 class QuicServerDispatchPacketTest : public QuicTest { |
146 public: | 145 public: |
147 QuicServerDispatchPacketTest() | 146 QuicServerDispatchPacketTest() |
148 : crypto_config_("blah", | 147 : crypto_config_("blah", |
149 QuicRandom::GetInstance(), | 148 QuicRandom::GetInstance(), |
150 crypto_test_utils::ProofSourceForTesting()), | 149 crypto_test_utils::ProofSourceForTesting()), |
151 version_manager_(AllSupportedVersions()), | 150 version_manager_(AllSupportedVersions()), |
152 dispatcher_( | 151 dispatcher_( |
153 config_, | 152 config_, |
154 &crypto_config_, | 153 &crypto_config_, |
155 &version_manager_, | 154 &version_manager_, |
(...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
198 reinterpret_cast<char*>(valid_packet), arraysize(valid_packet), | 197 reinterpret_cast<char*>(valid_packet), arraysize(valid_packet), |
199 QuicTime::Zero(), false); | 198 QuicTime::Zero(), false); |
200 | 199 |
201 EXPECT_CALL(dispatcher_, ProcessPacket(_, _, _)).Times(1); | 200 EXPECT_CALL(dispatcher_, ProcessPacket(_, _, _)).Times(1); |
202 DispatchPacket(encrypted_valid_packet); | 201 DispatchPacket(encrypted_valid_packet); |
203 } | 202 } |
204 | 203 |
205 } // namespace | 204 } // namespace |
206 } // namespace test | 205 } // namespace test |
207 } // namespace net | 206 } // namespace net |
OLD | NEW |