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 <string> | 10 #include <string> |
(...skipping 208 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
219 | 219 |
220 class MockConnectionVisitor : public QuicConnectionVisitorInterface { | 220 class MockConnectionVisitor : public QuicConnectionVisitorInterface { |
221 public: | 221 public: |
222 MockConnectionVisitor(); | 222 MockConnectionVisitor(); |
223 virtual ~MockConnectionVisitor(); | 223 virtual ~MockConnectionVisitor(); |
224 | 224 |
225 MOCK_METHOD1(OnStreamFrames, bool(const std::vector<QuicStreamFrame>& frame)); | 225 MOCK_METHOD1(OnStreamFrames, bool(const std::vector<QuicStreamFrame>& frame)); |
226 MOCK_METHOD1(OnRstStream, void(const QuicRstStreamFrame& frame)); | 226 MOCK_METHOD1(OnRstStream, void(const QuicRstStreamFrame& frame)); |
227 MOCK_METHOD1(OnGoAway, void(const QuicGoAwayFrame& frame)); | 227 MOCK_METHOD1(OnGoAway, void(const QuicGoAwayFrame& frame)); |
228 MOCK_METHOD2(OnConnectionClosed, void(QuicErrorCode error, bool from_peer)); | 228 MOCK_METHOD2(OnConnectionClosed, void(QuicErrorCode error, bool from_peer)); |
| 229 MOCK_METHOD0(OnWriteBlocked, void()); |
229 MOCK_METHOD0(OnCanWrite, bool()); | 230 MOCK_METHOD0(OnCanWrite, bool()); |
230 MOCK_CONST_METHOD0(HasPendingHandshake, bool()); | 231 MOCK_CONST_METHOD0(HasPendingHandshake, bool()); |
231 MOCK_METHOD1(OnSuccessfulVersionNegotiation, | 232 MOCK_METHOD1(OnSuccessfulVersionNegotiation, |
232 void(const QuicVersion& version)); | 233 void(const QuicVersion& version)); |
233 MOCK_METHOD0(OnConfigNegotiated, void()); | 234 MOCK_METHOD0(OnConfigNegotiated, void()); |
234 | 235 |
235 private: | 236 private: |
236 DISALLOW_COPY_AND_ASSIGN(MockConnectionVisitor); | 237 DISALLOW_COPY_AND_ASSIGN(MockConnectionVisitor); |
237 }; | 238 }; |
238 | 239 |
(...skipping 107 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
346 MOCK_METHOD2(OnStreamHeaders, void(QuicStreamId stream_id, | 347 MOCK_METHOD2(OnStreamHeaders, void(QuicStreamId stream_id, |
347 base::StringPiece headers_data)); | 348 base::StringPiece headers_data)); |
348 MOCK_METHOD2(OnStreamHeadersPriority, void(QuicStreamId stream_id, | 349 MOCK_METHOD2(OnStreamHeadersPriority, void(QuicStreamId stream_id, |
349 QuicPriority priority)); | 350 QuicPriority priority)); |
350 MOCK_METHOD3(OnStreamHeadersComplete, void(QuicStreamId stream_id, | 351 MOCK_METHOD3(OnStreamHeadersComplete, void(QuicStreamId stream_id, |
351 bool fin, | 352 bool fin, |
352 size_t frame_len)); | 353 size_t frame_len)); |
353 MOCK_METHOD0(IsHandshakeComplete, bool()); | 354 MOCK_METHOD0(IsHandshakeComplete, bool()); |
354 MOCK_METHOD0(IsCryptoHandshakeConfirmed, bool()); | 355 MOCK_METHOD0(IsCryptoHandshakeConfirmed, bool()); |
355 | 356 |
| 357 using QuicSession::ActivateStream; |
| 358 |
356 private: | 359 private: |
357 DISALLOW_COPY_AND_ASSIGN(MockSession); | 360 DISALLOW_COPY_AND_ASSIGN(MockSession); |
358 }; | 361 }; |
359 | 362 |
360 class TestSession : public QuicSession { | 363 class TestSession : public QuicSession { |
361 public: | 364 public: |
362 TestSession(QuicConnection* connection, const QuicConfig& config); | 365 TestSession(QuicConnection* connection, const QuicConfig& config); |
363 virtual ~TestSession(); | 366 virtual ~TestSession(); |
364 | 367 |
365 MOCK_METHOD1(CreateIncomingDataStream, QuicDataStream*(QuicStreamId id)); | 368 MOCK_METHOD1(CreateIncomingDataStream, QuicDataStream*(QuicStreamId id)); |
(...skipping 28 matching lines...) Expand all Loading... |
394 public: | 397 public: |
395 MockSendAlgorithm(); | 398 MockSendAlgorithm(); |
396 virtual ~MockSendAlgorithm(); | 399 virtual ~MockSendAlgorithm(); |
397 | 400 |
398 MOCK_METHOD2(SetFromConfig, void(const QuicConfig& config, bool is_server)); | 401 MOCK_METHOD2(SetFromConfig, void(const QuicConfig& config, bool is_server)); |
399 MOCK_METHOD1(SetMaxPacketSize, void(QuicByteCount max_packet_size)); | 402 MOCK_METHOD1(SetMaxPacketSize, void(QuicByteCount max_packet_size)); |
400 MOCK_METHOD3(OnIncomingQuicCongestionFeedbackFrame, | 403 MOCK_METHOD3(OnIncomingQuicCongestionFeedbackFrame, |
401 void(const QuicCongestionFeedbackFrame&, | 404 void(const QuicCongestionFeedbackFrame&, |
402 QuicTime feedback_receive_time, | 405 QuicTime feedback_receive_time, |
403 const SentPacketsMap&)); | 406 const SentPacketsMap&)); |
404 MOCK_METHOD3(OnPacketAcked, | 407 MOCK_METHOD2(OnPacketAcked, |
405 void(QuicPacketSequenceNumber, QuicByteCount, QuicTime::Delta)); | 408 void(QuicPacketSequenceNumber, QuicByteCount)); |
406 MOCK_METHOD2(OnPacketLost, void(QuicPacketSequenceNumber, QuicTime)); | 409 MOCK_METHOD2(OnPacketLost, void(QuicPacketSequenceNumber, QuicTime)); |
407 MOCK_METHOD5(OnPacketSent, | 410 MOCK_METHOD5(OnPacketSent, |
408 bool(QuicTime sent_time, QuicPacketSequenceNumber, QuicByteCount, | 411 bool(QuicTime sent_time, QuicPacketSequenceNumber, QuicByteCount, |
409 TransmissionType, HasRetransmittableData)); | 412 TransmissionType, HasRetransmittableData)); |
410 MOCK_METHOD1(OnRetransmissionTimeout, void(bool)); | 413 MOCK_METHOD1(OnRetransmissionTimeout, void(bool)); |
411 MOCK_METHOD2(OnPacketAbandoned, void(QuicPacketSequenceNumber sequence_number, | 414 MOCK_METHOD2(OnPacketAbandoned, void(QuicPacketSequenceNumber sequence_number, |
412 QuicByteCount abandoned_bytes)); | 415 QuicByteCount abandoned_bytes)); |
413 MOCK_METHOD4(TimeUntilSend, QuicTime::Delta(QuicTime now, TransmissionType, | 416 MOCK_METHOD4(TimeUntilSend, QuicTime::Delta(QuicTime now, TransmissionType, |
414 HasRetransmittableData, | 417 HasRetransmittableData, |
415 IsHandshake)); | 418 IsHandshake)); |
416 MOCK_CONST_METHOD0(BandwidthEstimate, QuicBandwidth(void)); | 419 MOCK_CONST_METHOD0(BandwidthEstimate, QuicBandwidth(void)); |
| 420 MOCK_METHOD1(UpdateRtt, void(QuicTime::Delta rtt_sample)); |
417 MOCK_CONST_METHOD0(SmoothedRtt, QuicTime::Delta(void)); | 421 MOCK_CONST_METHOD0(SmoothedRtt, QuicTime::Delta(void)); |
418 MOCK_CONST_METHOD0(RetransmissionDelay, QuicTime::Delta(void)); | 422 MOCK_CONST_METHOD0(RetransmissionDelay, QuicTime::Delta(void)); |
419 MOCK_CONST_METHOD0(GetCongestionWindow, QuicByteCount()); | 423 MOCK_CONST_METHOD0(GetCongestionWindow, QuicByteCount()); |
420 | 424 |
421 private: | 425 private: |
422 DISALLOW_COPY_AND_ASSIGN(MockSendAlgorithm); | 426 DISALLOW_COPY_AND_ASSIGN(MockSendAlgorithm); |
423 }; | 427 }; |
424 | 428 |
425 class TestEntropyCalculator : | 429 class TestEntropyCalculator : |
426 public QuicReceivedEntropyHashCalculatorInterface { | 430 public QuicReceivedEntropyHashCalculatorInterface { |
(...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
464 | 468 |
465 protected: | 469 protected: |
466 // Object is ref counted. | 470 // Object is ref counted. |
467 virtual ~MockAckNotifierDelegate(); | 471 virtual ~MockAckNotifierDelegate(); |
468 }; | 472 }; |
469 | 473 |
470 } // namespace test | 474 } // namespace test |
471 } // namespace net | 475 } // namespace net |
472 | 476 |
473 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 477 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
OLD | NEW |