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_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 7 #ifndef NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
8 #define NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 8 #define NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
9 | 9 |
10 #include "net/quic/congestion_control/quic_send_scheduler.h" | 10 #include "net/quic/congestion_control/quic_send_scheduler.h" |
(...skipping 14 matching lines...) Expand all Loading... |
25 const char* expected, | 25 const char* expected, |
26 const int expected_len); | 26 const int expected_len); |
27 | 27 |
28 void CompareQuicDataWithHexError(const std::string& description, | 28 void CompareQuicDataWithHexError(const std::string& description, |
29 QuicData* actual, | 29 QuicData* actual, |
30 QuicData* expected); | 30 QuicData* expected); |
31 | 31 |
32 // Constructs a basic crypto handshake message | 32 // Constructs a basic crypto handshake message |
33 QuicPacket* ConstructHandshakePacket(QuicGuid guid, CryptoTag tag); | 33 QuicPacket* ConstructHandshakePacket(QuicGuid guid, CryptoTag tag); |
34 | 34 |
| 35 // Constructs a ClientHello crypto handshake message |
| 36 QuicPacket* ConstructClientHelloPacket(QuicGuid guid, |
| 37 const QuicClock* clock, |
| 38 QuicRandom* random_generator); |
| 39 |
35 class MockFramerVisitor : public QuicFramerVisitorInterface { | 40 class MockFramerVisitor : public QuicFramerVisitorInterface { |
36 public: | 41 public: |
37 MockFramerVisitor(); | 42 MockFramerVisitor(); |
38 ~MockFramerVisitor(); | 43 ~MockFramerVisitor(); |
39 | 44 |
40 MOCK_METHOD1(OnError, void(QuicFramer* framer)); | 45 MOCK_METHOD1(OnError, void(QuicFramer* framer)); |
41 MOCK_METHOD2(OnPacket, void(const IPEndPoint& self_address, | 46 MOCK_METHOD2(OnPacket, void(const IPEndPoint& self_address, |
42 const IPEndPoint& peer_address)); | 47 const IPEndPoint& peer_address)); |
43 MOCK_METHOD0(OnRevivedPacket, void()); | 48 MOCK_METHOD0(OnRevivedPacket, void()); |
44 // The constructor set this up to return true by default. | 49 // The constructor set this up to return true by default. |
(...skipping 96 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
141 MOCK_METHOD0(IsSendAlarmSet, bool()); | 146 MOCK_METHOD0(IsSendAlarmSet, bool()); |
142 MOCK_METHOD0(UnregisterSendAlarmIfRegistered, void()); | 147 MOCK_METHOD0(UnregisterSendAlarmIfRegistered, void()); |
143 private: | 148 private: |
144 const MockClock clock_; | 149 const MockClock clock_; |
145 MockRandom random_generator_; | 150 MockRandom random_generator_; |
146 }; | 151 }; |
147 | 152 |
148 class MockConnection : public QuicConnection { | 153 class MockConnection : public QuicConnection { |
149 public: | 154 public: |
150 MockConnection(QuicGuid guid, IPEndPoint address); | 155 MockConnection(QuicGuid guid, IPEndPoint address); |
| 156 MockConnection(QuicGuid guid, |
| 157 IPEndPoint address, |
| 158 QuicConnectionHelperInterface* helper); |
151 virtual ~MockConnection(); | 159 virtual ~MockConnection(); |
152 | 160 |
153 MOCK_METHOD3(ProcessUdpPacket, void(const IPEndPoint& self_address, | 161 MOCK_METHOD3(ProcessUdpPacket, void(const IPEndPoint& self_address, |
154 const IPEndPoint& peer_address, | 162 const IPEndPoint& peer_address, |
155 const QuicEncryptedPacket& packet)); | 163 const QuicEncryptedPacket& packet)); |
156 MOCK_METHOD1(SendConnectionClose, void(QuicErrorCode error)); | 164 MOCK_METHOD1(SendConnectionClose, void(QuicErrorCode error)); |
157 | 165 |
158 MOCK_METHOD3(SendRstStream, void(QuicStreamId id, | 166 MOCK_METHOD3(SendRstStream, void(QuicStreamId id, |
159 QuicErrorCode error, | 167 QuicErrorCode error, |
160 QuicStreamOffset offset)); | 168 QuicStreamOffset offset)); |
(...skipping 29 matching lines...) Expand all Loading... |
190 | 198 |
191 MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, | 199 MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, |
192 const IPEndPoint& peer_address, | 200 const IPEndPoint& peer_address, |
193 const QuicPacketHeader& header, | 201 const QuicPacketHeader& header, |
194 const std::vector<QuicStreamFrame>& frame)); | 202 const std::vector<QuicStreamFrame>& frame)); |
195 MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); | 203 MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); |
196 MOCK_METHOD1(CreateIncomingReliableStream, | 204 MOCK_METHOD1(CreateIncomingReliableStream, |
197 ReliableQuicStream*(QuicStreamId id)); | 205 ReliableQuicStream*(QuicStreamId id)); |
198 MOCK_METHOD0(GetCryptoStream, QuicCryptoStream*()); | 206 MOCK_METHOD0(GetCryptoStream, QuicCryptoStream*()); |
199 MOCK_METHOD0(CreateOutgoingReliableStream, ReliableQuicStream*()); | 207 MOCK_METHOD0(CreateOutgoingReliableStream, ReliableQuicStream*()); |
200 MOCK_METHOD3(WriteData, | |
201 void(QuicStreamId id, base::StringPiece data, bool fin)); | |
202 MOCK_METHOD4(WriteData, QuicConsumedData(QuicStreamId id, | 208 MOCK_METHOD4(WriteData, QuicConsumedData(QuicStreamId id, |
203 base::StringPiece data, | 209 base::StringPiece data, |
204 QuicStreamOffset offset, bool fin)); | 210 QuicStreamOffset offset, bool fin)); |
205 MOCK_METHOD0(IsHandshakeComplete, bool()); | 211 MOCK_METHOD0(IsHandshakeComplete, bool()); |
206 | 212 |
207 private: | 213 private: |
208 DISALLOW_COPY_AND_ASSIGN(MockSession); | 214 DISALLOW_COPY_AND_ASSIGN(MockSession); |
209 }; | 215 }; |
210 | 216 |
211 class MockScheduler : public QuicSendScheduler { | 217 class MockScheduler : public QuicSendScheduler { |
212 public: | 218 public: |
213 MockScheduler(); | 219 MockScheduler(); |
214 virtual ~MockScheduler(); | 220 virtual ~MockScheduler(); |
215 | 221 |
216 MOCK_METHOD1(TimeUntilSend, QuicTime::Delta(bool)); | 222 MOCK_METHOD1(TimeUntilSend, QuicTime::Delta(bool)); |
217 MOCK_METHOD1(OnIncomingAckFrame, void(const QuicAckFrame&)); | 223 MOCK_METHOD1(OnIncomingAckFrame, void(const QuicAckFrame&)); |
218 MOCK_METHOD3(SentPacket, void(QuicPacketSequenceNumber, size_t, bool)); | 224 MOCK_METHOD3(SentPacket, void(QuicPacketSequenceNumber, size_t, bool)); |
219 | 225 |
220 private: | 226 private: |
221 DISALLOW_COPY_AND_ASSIGN(MockScheduler); | 227 DISALLOW_COPY_AND_ASSIGN(MockScheduler); |
222 }; | 228 }; |
223 | 229 |
224 } // namespace test | 230 } // namespace test |
225 | 231 |
226 } // namespace net | 232 } // namespace net |
227 | 233 |
228 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 234 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
OLD | NEW |