| 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 <stddef.h> | 10 #include <stddef.h> |
| (...skipping 202 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 213 MockFramerVisitor(); | 213 MockFramerVisitor(); |
| 214 ~MockFramerVisitor() override; | 214 ~MockFramerVisitor() override; |
| 215 | 215 |
| 216 MOCK_METHOD1(OnError, void(QuicFramer* framer)); | 216 MOCK_METHOD1(OnError, void(QuicFramer* framer)); |
| 217 // The constructor sets this up to return false by default. | 217 // The constructor sets this up to return false by default. |
| 218 MOCK_METHOD1(OnProtocolVersionMismatch, bool(QuicVersion version)); | 218 MOCK_METHOD1(OnProtocolVersionMismatch, bool(QuicVersion version)); |
| 219 MOCK_METHOD0(OnPacket, void()); | 219 MOCK_METHOD0(OnPacket, void()); |
| 220 MOCK_METHOD1(OnPublicResetPacket, void(const QuicPublicResetPacket& header)); | 220 MOCK_METHOD1(OnPublicResetPacket, void(const QuicPublicResetPacket& header)); |
| 221 MOCK_METHOD1(OnVersionNegotiationPacket, | 221 MOCK_METHOD1(OnVersionNegotiationPacket, |
| 222 void(const QuicVersionNegotiationPacket& packet)); | 222 void(const QuicVersionNegotiationPacket& packet)); |
| 223 MOCK_METHOD0(OnRevivedPacket, void()); | |
| 224 // The constructor sets this up to return true by default. | 223 // The constructor sets this up to return true by default. |
| 225 MOCK_METHOD1(OnUnauthenticatedHeader, bool(const QuicPacketHeader& header)); | 224 MOCK_METHOD1(OnUnauthenticatedHeader, bool(const QuicPacketHeader& header)); |
| 226 // The constructor sets this up to return true by default. | 225 // The constructor sets this up to return true by default. |
| 227 MOCK_METHOD1(OnUnauthenticatedPublicHeader, | 226 MOCK_METHOD1(OnUnauthenticatedPublicHeader, |
| 228 bool(const QuicPacketPublicHeader& header)); | 227 bool(const QuicPacketPublicHeader& header)); |
| 229 MOCK_METHOD1(OnDecryptedPacket, void(EncryptionLevel level)); | 228 MOCK_METHOD1(OnDecryptedPacket, void(EncryptionLevel level)); |
| 230 MOCK_METHOD1(OnPacketHeader, bool(const QuicPacketHeader& header)); | 229 MOCK_METHOD1(OnPacketHeader, bool(const QuicPacketHeader& header)); |
| 231 MOCK_METHOD1(OnStreamFrame, bool(const QuicStreamFrame& frame)); | 230 MOCK_METHOD1(OnStreamFrame, bool(const QuicStreamFrame& frame)); |
| 232 MOCK_METHOD1(OnAckFrame, bool(const QuicAckFrame& frame)); | 231 MOCK_METHOD1(OnAckFrame, bool(const QuicAckFrame& frame)); |
| 233 MOCK_METHOD1(OnStopWaitingFrame, bool(const QuicStopWaitingFrame& frame)); | 232 MOCK_METHOD1(OnStopWaitingFrame, bool(const QuicStopWaitingFrame& frame)); |
| (...skipping 496 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 730 MOCK_METHOD1(OnStopWaitingFrame, void(const QuicStopWaitingFrame&)); | 729 MOCK_METHOD1(OnStopWaitingFrame, void(const QuicStopWaitingFrame&)); |
| 731 | 730 |
| 732 MOCK_METHOD1(OnRstStreamFrame, void(const QuicRstStreamFrame&)); | 731 MOCK_METHOD1(OnRstStreamFrame, void(const QuicRstStreamFrame&)); |
| 733 | 732 |
| 734 MOCK_METHOD1(OnConnectionCloseFrame, void(const QuicConnectionCloseFrame&)); | 733 MOCK_METHOD1(OnConnectionCloseFrame, void(const QuicConnectionCloseFrame&)); |
| 735 | 734 |
| 736 MOCK_METHOD1(OnPublicResetPacket, void(const QuicPublicResetPacket&)); | 735 MOCK_METHOD1(OnPublicResetPacket, void(const QuicPublicResetPacket&)); |
| 737 | 736 |
| 738 MOCK_METHOD1(OnVersionNegotiationPacket, | 737 MOCK_METHOD1(OnVersionNegotiationPacket, |
| 739 void(const QuicVersionNegotiationPacket&)); | 738 void(const QuicVersionNegotiationPacket&)); |
| 740 | |
| 741 MOCK_METHOD2(OnRevivedPacket, | |
| 742 void(const QuicPacketHeader&, StringPiece payload)); | |
| 743 }; | 739 }; |
| 744 | 740 |
| 745 class MockReceivedPacketManager : public QuicReceivedPacketManager { | 741 class MockReceivedPacketManager : public QuicReceivedPacketManager { |
| 746 public: | 742 public: |
| 747 explicit MockReceivedPacketManager(QuicConnectionStats* stats); | 743 explicit MockReceivedPacketManager(QuicConnectionStats* stats); |
| 748 ~MockReceivedPacketManager() override; | 744 ~MockReceivedPacketManager() override; |
| 749 | 745 |
| 750 MOCK_METHOD3(RecordPacketReceived, | 746 MOCK_METHOD3(RecordPacketReceived, |
| 751 void(QuicByteCount bytes, | 747 void(QuicByteCount bytes, |
| 752 const QuicPacketHeader& header, | 748 const QuicPacketHeader& header, |
| 753 QuicTime receipt_time)); | 749 QuicTime receipt_time)); |
| 754 MOCK_METHOD1(RecordPacketRevived, void(QuicPacketNumber packet_number)); | |
| 755 MOCK_METHOD1(IsMissing, bool(QuicPacketNumber packet_number)); | 750 MOCK_METHOD1(IsMissing, bool(QuicPacketNumber packet_number)); |
| 756 MOCK_METHOD1(IsAwaitingPacket, bool(QuicPacketNumber packet_number)); | 751 MOCK_METHOD1(IsAwaitingPacket, bool(QuicPacketNumber packet_number)); |
| 757 MOCK_METHOD1(UpdatePacketInformationSentByPeer, | 752 MOCK_METHOD1(UpdatePacketInformationSentByPeer, |
| 758 void(const QuicStopWaitingFrame& stop_waiting)); | 753 void(const QuicStopWaitingFrame& stop_waiting)); |
| 759 MOCK_CONST_METHOD0(HasNewMissingPackets, bool(void)); | 754 MOCK_CONST_METHOD0(HasNewMissingPackets, bool(void)); |
| 760 MOCK_CONST_METHOD0(ack_frame_updated, bool(void)); | 755 MOCK_CONST_METHOD0(ack_frame_updated, bool(void)); |
| 761 }; | 756 }; |
| 762 | 757 |
| 763 // Creates a client session for testing. | 758 // Creates a client session for testing. |
| 764 // | 759 // |
| (...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 811 TestQuicSpdyServerSession** server_session); | 806 TestQuicSpdyServerSession** server_session); |
| 812 | 807 |
| 813 // Helper to generate client side stream ids, generalizes | 808 // Helper to generate client side stream ids, generalizes |
| 814 // kClientDataStreamId1 etc. above. | 809 // kClientDataStreamId1 etc. above. |
| 815 QuicStreamId QuicClientDataStreamId(int i); | 810 QuicStreamId QuicClientDataStreamId(int i); |
| 816 | 811 |
| 817 } // namespace test | 812 } // namespace test |
| 818 } // namespace net | 813 } // namespace net |
| 819 | 814 |
| 820 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 815 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| OLD | NEW |