| 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 153 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 164 // The constructor sets this up to return true by default. | 164 // The constructor sets this up to return true by default. |
| 165 MOCK_METHOD1(OnUnauthenticatedHeader, bool(const QuicPacketHeader& header)); | 165 MOCK_METHOD1(OnUnauthenticatedHeader, bool(const QuicPacketHeader& header)); |
| 166 // The constructor sets this up to return true by default. | 166 // The constructor sets this up to return true by default. |
| 167 MOCK_METHOD1(OnUnauthenticatedPublicHeader, bool( | 167 MOCK_METHOD1(OnUnauthenticatedPublicHeader, bool( |
| 168 const QuicPacketPublicHeader& header)); | 168 const QuicPacketPublicHeader& header)); |
| 169 MOCK_METHOD1(OnDecryptedPacket, void(EncryptionLevel level)); | 169 MOCK_METHOD1(OnDecryptedPacket, void(EncryptionLevel level)); |
| 170 MOCK_METHOD1(OnPacketHeader, bool(const QuicPacketHeader& header)); | 170 MOCK_METHOD1(OnPacketHeader, bool(const QuicPacketHeader& header)); |
| 171 MOCK_METHOD1(OnFecProtectedPayload, void(base::StringPiece payload)); | 171 MOCK_METHOD1(OnFecProtectedPayload, void(base::StringPiece payload)); |
| 172 MOCK_METHOD1(OnStreamFrame, bool(const QuicStreamFrame& frame)); | 172 MOCK_METHOD1(OnStreamFrame, bool(const QuicStreamFrame& frame)); |
| 173 MOCK_METHOD1(OnAckFrame, bool(const QuicAckFrame& frame)); | 173 MOCK_METHOD1(OnAckFrame, bool(const QuicAckFrame& frame)); |
| 174 MOCK_METHOD1(OnCongestionFeedbackFrame, | |
| 175 bool(const QuicCongestionFeedbackFrame& frame)); | |
| 176 MOCK_METHOD1(OnStopWaitingFrame, bool(const QuicStopWaitingFrame& frame)); | 174 MOCK_METHOD1(OnStopWaitingFrame, bool(const QuicStopWaitingFrame& frame)); |
| 177 MOCK_METHOD1(OnPingFrame, bool(const QuicPingFrame& frame)); | 175 MOCK_METHOD1(OnPingFrame, bool(const QuicPingFrame& frame)); |
| 178 MOCK_METHOD1(OnFecData, void(const QuicFecData& fec)); | 176 MOCK_METHOD1(OnFecData, void(const QuicFecData& fec)); |
| 179 MOCK_METHOD1(OnRstStreamFrame, bool(const QuicRstStreamFrame& frame)); | 177 MOCK_METHOD1(OnRstStreamFrame, bool(const QuicRstStreamFrame& frame)); |
| 180 MOCK_METHOD1(OnConnectionCloseFrame, | 178 MOCK_METHOD1(OnConnectionCloseFrame, |
| 181 bool(const QuicConnectionCloseFrame& frame)); | 179 bool(const QuicConnectionCloseFrame& frame)); |
| 182 MOCK_METHOD1(OnGoAwayFrame, bool(const QuicGoAwayFrame& frame)); | 180 MOCK_METHOD1(OnGoAwayFrame, bool(const QuicGoAwayFrame& frame)); |
| 183 MOCK_METHOD1(OnWindowUpdateFrame, bool(const QuicWindowUpdateFrame& frame)); | 181 MOCK_METHOD1(OnWindowUpdateFrame, bool(const QuicWindowUpdateFrame& frame)); |
| 184 MOCK_METHOD1(OnBlockedFrame, bool(const QuicBlockedFrame& frame)); | 182 MOCK_METHOD1(OnBlockedFrame, bool(const QuicBlockedFrame& frame)); |
| 185 MOCK_METHOD0(OnPacketComplete, void()); | 183 MOCK_METHOD0(OnPacketComplete, void()); |
| (...skipping 14 matching lines...) Expand all Loading... |
| 200 void OnRevivedPacket() override {} | 198 void OnRevivedPacket() override {} |
| 201 bool OnProtocolVersionMismatch(QuicVersion version) override; | 199 bool OnProtocolVersionMismatch(QuicVersion version) override; |
| 202 bool OnUnauthenticatedHeader(const QuicPacketHeader& header) override; | 200 bool OnUnauthenticatedHeader(const QuicPacketHeader& header) override; |
| 203 bool OnUnauthenticatedPublicHeader( | 201 bool OnUnauthenticatedPublicHeader( |
| 204 const QuicPacketPublicHeader& header) override; | 202 const QuicPacketPublicHeader& header) override; |
| 205 void OnDecryptedPacket(EncryptionLevel level) override {} | 203 void OnDecryptedPacket(EncryptionLevel level) override {} |
| 206 bool OnPacketHeader(const QuicPacketHeader& header) override; | 204 bool OnPacketHeader(const QuicPacketHeader& header) override; |
| 207 void OnFecProtectedPayload(base::StringPiece payload) override {} | 205 void OnFecProtectedPayload(base::StringPiece payload) override {} |
| 208 bool OnStreamFrame(const QuicStreamFrame& frame) override; | 206 bool OnStreamFrame(const QuicStreamFrame& frame) override; |
| 209 bool OnAckFrame(const QuicAckFrame& frame) override; | 207 bool OnAckFrame(const QuicAckFrame& frame) override; |
| 210 bool OnCongestionFeedbackFrame( | |
| 211 const QuicCongestionFeedbackFrame& frame) override; | |
| 212 bool OnStopWaitingFrame(const QuicStopWaitingFrame& frame) override; | 208 bool OnStopWaitingFrame(const QuicStopWaitingFrame& frame) override; |
| 213 bool OnPingFrame(const QuicPingFrame& frame) override; | 209 bool OnPingFrame(const QuicPingFrame& frame) override; |
| 214 void OnFecData(const QuicFecData& fec) override {} | 210 void OnFecData(const QuicFecData& fec) override {} |
| 215 bool OnRstStreamFrame(const QuicRstStreamFrame& frame) override; | 211 bool OnRstStreamFrame(const QuicRstStreamFrame& frame) override; |
| 216 bool OnConnectionCloseFrame(const QuicConnectionCloseFrame& frame) override; | 212 bool OnConnectionCloseFrame(const QuicConnectionCloseFrame& frame) override; |
| 217 bool OnGoAwayFrame(const QuicGoAwayFrame& frame) override; | 213 bool OnGoAwayFrame(const QuicGoAwayFrame& frame) override; |
| 218 bool OnWindowUpdateFrame(const QuicWindowUpdateFrame& frame) override; | 214 bool OnWindowUpdateFrame(const QuicWindowUpdateFrame& frame) override; |
| 219 bool OnBlockedFrame(const QuicBlockedFrame& frame) override; | 215 bool OnBlockedFrame(const QuicBlockedFrame& frame) override; |
| 220 void OnPacketComplete() override {} | 216 void OnPacketComplete() override {} |
| 221 | 217 |
| (...skipping 231 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 453 class MockSendAlgorithm : public SendAlgorithmInterface { | 449 class MockSendAlgorithm : public SendAlgorithmInterface { |
| 454 public: | 450 public: |
| 455 MockSendAlgorithm(); | 451 MockSendAlgorithm(); |
| 456 ~MockSendAlgorithm() override; | 452 ~MockSendAlgorithm() override; |
| 457 | 453 |
| 458 MOCK_METHOD3(SetFromConfig, void(const QuicConfig& config, | 454 MOCK_METHOD3(SetFromConfig, void(const QuicConfig& config, |
| 459 bool is_server, | 455 bool is_server, |
| 460 bool using_pacing)); | 456 bool using_pacing)); |
| 461 MOCK_METHOD1(SetNumEmulatedConnections, void(int num_connections)); | 457 MOCK_METHOD1(SetNumEmulatedConnections, void(int num_connections)); |
| 462 MOCK_METHOD1(SetMaxPacketSize, void(QuicByteCount max_packet_size)); | 458 MOCK_METHOD1(SetMaxPacketSize, void(QuicByteCount max_packet_size)); |
| 463 MOCK_METHOD2(OnIncomingQuicCongestionFeedbackFrame, | |
| 464 void(const QuicCongestionFeedbackFrame&, | |
| 465 QuicTime feedback_receive_time)); | |
| 466 MOCK_METHOD4(OnCongestionEvent, void(bool rtt_updated, | 459 MOCK_METHOD4(OnCongestionEvent, void(bool rtt_updated, |
| 467 QuicByteCount bytes_in_flight, | 460 QuicByteCount bytes_in_flight, |
| 468 const CongestionVector& acked_packets, | 461 const CongestionVector& acked_packets, |
| 469 const CongestionVector& lost_packets)); | 462 const CongestionVector& lost_packets)); |
| 470 MOCK_METHOD5(OnPacketSent, | 463 MOCK_METHOD5(OnPacketSent, |
| 471 bool(QuicTime, QuicByteCount, QuicPacketSequenceNumber, | 464 bool(QuicTime, QuicByteCount, QuicPacketSequenceNumber, |
| 472 QuicByteCount, HasRetransmittableData)); | 465 QuicByteCount, HasRetransmittableData)); |
| 473 MOCK_METHOD1(OnRetransmissionTimeout, void(bool)); | 466 MOCK_METHOD1(OnRetransmissionTimeout, void(bool)); |
| 474 MOCK_METHOD0(RevertRetransmissionTimeout, void()); | 467 MOCK_METHOD0(RevertRetransmissionTimeout, void()); |
| 475 MOCK_CONST_METHOD3(TimeUntilSend, | 468 MOCK_CONST_METHOD3(TimeUntilSend, |
| (...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 532 QuicPacketEntropyHash(QuicPacketSequenceNumber sequence_number)); | 525 QuicPacketEntropyHash(QuicPacketSequenceNumber sequence_number)); |
| 533 | 526 |
| 534 private: | 527 private: |
| 535 DISALLOW_COPY_AND_ASSIGN(MockEntropyCalculator); | 528 DISALLOW_COPY_AND_ASSIGN(MockEntropyCalculator); |
| 536 }; | 529 }; |
| 537 | 530 |
| 538 class MockAckNotifierDelegate : public QuicAckNotifier::DelegateInterface { | 531 class MockAckNotifierDelegate : public QuicAckNotifier::DelegateInterface { |
| 539 public: | 532 public: |
| 540 MockAckNotifierDelegate(); | 533 MockAckNotifierDelegate(); |
| 541 | 534 |
| 542 MOCK_METHOD5(OnAckNotification, void(int num_original_packets, | 535 MOCK_METHOD3(OnAckNotification, |
| 543 int num_original_bytes, | 536 void(int num_retransmitted_packets, |
| 544 int num_retransmitted_packets, | 537 int num_retransmitted_bytes, |
| 545 int num_retransmitted_bytes, | 538 QuicTime::Delta delta_largest_observed)); |
| 546 QuicTime::Delta delta_largest_observed)); | |
| 547 | 539 |
| 548 protected: | 540 protected: |
| 549 // Object is ref counted. | 541 // Object is ref counted. |
| 550 ~MockAckNotifierDelegate() override; | 542 ~MockAckNotifierDelegate() override; |
| 551 | 543 |
| 552 private: | 544 private: |
| 553 DISALLOW_COPY_AND_ASSIGN(MockAckNotifierDelegate); | 545 DISALLOW_COPY_AND_ASSIGN(MockAckNotifierDelegate); |
| 554 }; | 546 }; |
| 555 | 547 |
| 556 class MockNetworkChangeVisitor : | 548 class MockNetworkChangeVisitor : |
| (...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 602 // clears the pointer to it to prevent use-after-free. | 594 // clears the pointer to it to prevent use-after-free. |
| 603 void Unregister(PerConnectionPacketWriter* writer); | 595 void Unregister(PerConnectionPacketWriter* writer); |
| 604 | 596 |
| 605 PerConnectionPacketWriter* current_writer_; | 597 PerConnectionPacketWriter* current_writer_; |
| 606 }; | 598 }; |
| 607 | 599 |
| 608 } // namespace test | 600 } // namespace test |
| 609 } // namespace net | 601 } // namespace net |
| 610 | 602 |
| 611 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 603 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| OLD | NEW |