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 |