| 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 |