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 146 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
191 MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, | 196 MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, |
192 const IPEndPoint& peer_address, | 197 const IPEndPoint& peer_address, |
193 const QuicPacketHeader& header, | 198 const QuicPacketHeader& header, |
194 const std::vector<QuicStreamFrame>& frame)); | 199 const std::vector<QuicStreamFrame>& frame)); |
195 MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); | 200 MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); |
196 MOCK_METHOD1(CreateIncomingReliableStream, | 201 MOCK_METHOD1(CreateIncomingReliableStream, |
197 ReliableQuicStream*(QuicStreamId id)); | 202 ReliableQuicStream*(QuicStreamId id)); |
198 MOCK_METHOD0(GetCryptoStream, QuicCryptoStream*()); | 203 MOCK_METHOD0(GetCryptoStream, QuicCryptoStream*()); |
199 MOCK_METHOD0(CreateOutgoingReliableStream, ReliableQuicStream*()); | 204 MOCK_METHOD0(CreateOutgoingReliableStream, ReliableQuicStream*()); |
200 MOCK_METHOD3(WriteData, | 205 MOCK_METHOD3(WriteData, |
201 void(QuicStreamId id, base::StringPiece data, bool fin)); | 206 void(QuicStreamId id, base::StringPiece data, bool fin)); |
wtc
2013/01/15 23:11:03
I removed this mock WriteData method because QuicS
Ryan Hamilton
2013/01/16 00:29:36
Oh nice.
| |
202 MOCK_METHOD4(WriteData, int(QuicStreamId id, base::StringPiece data, | 207 MOCK_METHOD4(WriteData, int(QuicStreamId id, base::StringPiece data, |
203 QuicStreamOffset offset, bool fin)); | 208 QuicStreamOffset offset, bool fin)); |
204 MOCK_METHOD0(IsHandshakeComplete, bool()); | 209 MOCK_METHOD0(IsHandshakeComplete, bool()); |
205 | 210 |
206 private: | 211 private: |
207 DISALLOW_COPY_AND_ASSIGN(MockSession); | 212 DISALLOW_COPY_AND_ASSIGN(MockSession); |
208 }; | 213 }; |
209 | 214 |
210 class MockScheduler : public QuicSendScheduler { | 215 class MockScheduler : public QuicSendScheduler { |
211 public: | 216 public: |
212 MockScheduler(); | 217 MockScheduler(); |
213 virtual ~MockScheduler(); | 218 virtual ~MockScheduler(); |
214 | 219 |
215 MOCK_METHOD1(TimeUntilSend, QuicTime::Delta(bool)); | 220 MOCK_METHOD1(TimeUntilSend, QuicTime::Delta(bool)); |
216 MOCK_METHOD1(OnIncomingAckFrame, void(const QuicAckFrame&)); | 221 MOCK_METHOD1(OnIncomingAckFrame, void(const QuicAckFrame&)); |
217 MOCK_METHOD3(SentPacket, void(QuicPacketSequenceNumber, size_t, bool)); | 222 MOCK_METHOD3(SentPacket, void(QuicPacketSequenceNumber, size_t, bool)); |
218 | 223 |
219 private: | 224 private: |
220 DISALLOW_COPY_AND_ASSIGN(MockScheduler); | 225 DISALLOW_COPY_AND_ASSIGN(MockScheduler); |
221 }; | 226 }; |
222 | 227 |
223 } // namespace test | 228 } // namespace test |
224 | 229 |
225 } // namespace net | 230 } // namespace net |
226 | 231 |
227 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 232 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
OLD | NEW |