| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 // Common utilities for Quic tests | 5 // Common utilities for Quic tests |
| 6 | 6 |
| 7 #ifndef NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 7 #ifndef NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| 8 #define NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 8 #define NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| 9 | 9 |
| 10 #include <string> | 10 #include <string> |
| (...skipping 136 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 147 void Unregister(PerConnectionPacketWriter* writer); | 147 void Unregister(PerConnectionPacketWriter* writer); |
| 148 | 148 |
| 149 PerConnectionPacketWriter* current_writer_; | 149 PerConnectionPacketWriter* current_writer_; |
| 150 }; | 150 }; |
| 151 | 151 |
| 152 class MockTimeWaitListManager : public QuicTimeWaitListManager { | 152 class MockTimeWaitListManager : public QuicTimeWaitListManager { |
| 153 public: | 153 public: |
| 154 MockTimeWaitListManager(QuicPacketWriter* writer, | 154 MockTimeWaitListManager(QuicPacketWriter* writer, |
| 155 QuicServerSessionVisitor* visitor, | 155 QuicServerSessionVisitor* visitor, |
| 156 QuicConnectionHelperInterface* helper); | 156 QuicConnectionHelperInterface* helper); |
| 157 | |
| 158 ~MockTimeWaitListManager() override; | 157 ~MockTimeWaitListManager() override; |
| 159 | 158 |
| 160 MOCK_METHOD3(AddConnectionIdToTimeWait, | 159 MOCK_METHOD4(AddConnectionIdToTimeWait, |
| 161 void(QuicConnectionId connection_id, | 160 void(QuicConnectionId connection_id, |
| 162 QuicVersion version, | 161 QuicVersion version, |
| 162 bool connection_rejected_statelessly, |
| 163 QuicEncryptedPacket* close_packet)); | 163 QuicEncryptedPacket* close_packet)); |
| 164 | 164 |
| 165 void QuicTimeWaitListManager_AddConnectionIdToTimeWait( | 165 void QuicTimeWaitListManager_AddConnectionIdToTimeWait( |
| 166 QuicConnectionId connection_id, | 166 QuicConnectionId connection_id, |
| 167 QuicVersion version, | 167 QuicVersion version, |
| 168 bool connection_rejected_statelessly, |
| 168 QuicEncryptedPacket* close_packet) { | 169 QuicEncryptedPacket* close_packet) { |
| 169 QuicTimeWaitListManager::AddConnectionIdToTimeWait(connection_id, version, | 170 QuicTimeWaitListManager::AddConnectionIdToTimeWait( |
| 170 close_packet); | 171 connection_id, version, connection_rejected_statelessly, close_packet); |
| 171 } | 172 } |
| 172 | 173 |
| 173 MOCK_METHOD5(ProcessPacket, | 174 MOCK_METHOD5(ProcessPacket, |
| 174 void(const IPEndPoint& server_address, | 175 void(const IPEndPoint& server_address, |
| 175 const IPEndPoint& client_address, | 176 const IPEndPoint& client_address, |
| 176 QuicConnectionId connection_id, | 177 QuicConnectionId connection_id, |
| 177 QuicPacketSequenceNumber sequence_number, | 178 QuicPacketSequenceNumber sequence_number, |
| 178 const QuicEncryptedPacket& packet)); | 179 const QuicEncryptedPacket& packet)); |
| 179 }; | 180 }; |
| 180 | 181 |
| 181 } // namespace test | 182 } // namespace test |
| 182 } // namespace tools | 183 } // namespace tools |
| 183 } // namespace net | 184 } // namespace net |
| 184 | 185 |
| 185 #endif // NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 186 #endif // NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| OLD | NEW |