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, IPEndPoint address, | |
157 QuicConnectionHelperInterface* helper); | |
wtc
2013/01/15 23:11:03
This formatting is allowed by the current Style Gu
Ryan Hamilton
2013/01/16 00:29:36
Are you sure? jar complains about this every time
wtc
2013/01/16 00:43:49
Yes, I am sure. See the second code example at
htt
| |
151 virtual ~MockConnection(); | 158 virtual ~MockConnection(); |
152 | 159 |
153 MOCK_METHOD3(ProcessUdpPacket, void(const IPEndPoint& self_address, | 160 MOCK_METHOD3(ProcessUdpPacket, void(const IPEndPoint& self_address, |
154 const IPEndPoint& peer_address, | 161 const IPEndPoint& peer_address, |
155 const QuicEncryptedPacket& packet)); | 162 const QuicEncryptedPacket& packet)); |
156 MOCK_METHOD1(SendConnectionClose, void(QuicErrorCode error)); | 163 MOCK_METHOD1(SendConnectionClose, void(QuicErrorCode error)); |
157 | 164 |
158 MOCK_METHOD3(SendRstStream, void(QuicStreamId id, | 165 MOCK_METHOD3(SendRstStream, void(QuicStreamId id, |
159 QuicErrorCode error, | 166 QuicErrorCode error, |
160 QuicStreamOffset offset)); | 167 QuicStreamOffset offset)); |
(...skipping 29 matching lines...) Expand all Loading... | |
190 | 197 |
191 MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, | 198 MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, |
192 const IPEndPoint& peer_address, | 199 const IPEndPoint& peer_address, |
193 const QuicPacketHeader& header, | 200 const QuicPacketHeader& header, |
194 const std::vector<QuicStreamFrame>& frame)); | 201 const std::vector<QuicStreamFrame>& frame)); |
195 MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); | 202 MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); |
196 MOCK_METHOD1(CreateIncomingReliableStream, | 203 MOCK_METHOD1(CreateIncomingReliableStream, |
197 ReliableQuicStream*(QuicStreamId id)); | 204 ReliableQuicStream*(QuicStreamId id)); |
198 MOCK_METHOD0(GetCryptoStream, QuicCryptoStream*()); | 205 MOCK_METHOD0(GetCryptoStream, QuicCryptoStream*()); |
199 MOCK_METHOD0(CreateOutgoingReliableStream, ReliableQuicStream*()); | 206 MOCK_METHOD0(CreateOutgoingReliableStream, ReliableQuicStream*()); |
200 MOCK_METHOD3(WriteData, | |
201 void(QuicStreamId id, base::StringPiece data, bool fin)); | |
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 |