| 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 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 46 // Returns QuicConfig set to default values. | 46 // Returns QuicConfig set to default values. |
| 47 QuicConfig DefaultQuicConfig(); | 47 QuicConfig DefaultQuicConfig(); |
| 48 | 48 |
| 49 class MockFramerVisitor : public QuicFramerVisitorInterface { | 49 class MockFramerVisitor : public QuicFramerVisitorInterface { |
| 50 public: | 50 public: |
| 51 MockFramerVisitor(); | 51 MockFramerVisitor(); |
| 52 ~MockFramerVisitor(); | 52 ~MockFramerVisitor(); |
| 53 | 53 |
| 54 MOCK_METHOD1(OnError, void(QuicFramer* framer)); | 54 MOCK_METHOD1(OnError, void(QuicFramer* framer)); |
| 55 // The constructor sets this up to return false by default. | 55 // The constructor sets this up to return false by default. |
| 56 MOCK_METHOD1(OnProtocolVersionMismatch, bool(QuicTag version)); | 56 MOCK_METHOD1(OnProtocolVersionMismatch, bool(QuicVersion version)); |
| 57 MOCK_METHOD0(OnPacket, void()); | 57 MOCK_METHOD0(OnPacket, void()); |
| 58 MOCK_METHOD1(OnPublicResetPacket, void(const QuicPublicResetPacket& header)); | 58 MOCK_METHOD1(OnPublicResetPacket, void(const QuicPublicResetPacket& header)); |
| 59 MOCK_METHOD1(OnVersionNegotiationPacket, | 59 MOCK_METHOD1(OnVersionNegotiationPacket, |
| 60 void(const QuicVersionNegotiationPacket& packet)); | 60 void(const QuicVersionNegotiationPacket& packet)); |
| 61 MOCK_METHOD0(OnRevivedPacket, void()); | 61 MOCK_METHOD0(OnRevivedPacket, void()); |
| 62 // The constructor sets this up to return true by default. | 62 // The constructor sets this up to return true by default. |
| 63 MOCK_METHOD1(OnPacketHeader, bool(const QuicPacketHeader& header)); | 63 MOCK_METHOD1(OnPacketHeader, bool(const QuicPacketHeader& header)); |
| 64 MOCK_METHOD1(OnFecProtectedPayload, void(base::StringPiece payload)); | 64 MOCK_METHOD1(OnFecProtectedPayload, void(base::StringPiece payload)); |
| 65 MOCK_METHOD1(OnStreamFrame, bool(const QuicStreamFrame& frame)); | 65 MOCK_METHOD1(OnStreamFrame, bool(const QuicStreamFrame& frame)); |
| 66 MOCK_METHOD1(OnAckFrame, bool(const QuicAckFrame& frame)); | 66 MOCK_METHOD1(OnAckFrame, bool(const QuicAckFrame& frame)); |
| (...skipping 14 matching lines...) Expand all Loading... |
| 81 public: | 81 public: |
| 82 NoOpFramerVisitor() {} | 82 NoOpFramerVisitor() {} |
| 83 | 83 |
| 84 virtual void OnError(QuicFramer* framer) OVERRIDE {} | 84 virtual void OnError(QuicFramer* framer) OVERRIDE {} |
| 85 virtual void OnPacket() OVERRIDE {} | 85 virtual void OnPacket() OVERRIDE {} |
| 86 virtual void OnPublicResetPacket( | 86 virtual void OnPublicResetPacket( |
| 87 const QuicPublicResetPacket& packet) OVERRIDE {} | 87 const QuicPublicResetPacket& packet) OVERRIDE {} |
| 88 virtual void OnVersionNegotiationPacket( | 88 virtual void OnVersionNegotiationPacket( |
| 89 const QuicVersionNegotiationPacket& packet) OVERRIDE {} | 89 const QuicVersionNegotiationPacket& packet) OVERRIDE {} |
| 90 virtual void OnRevivedPacket() OVERRIDE {} | 90 virtual void OnRevivedPacket() OVERRIDE {} |
| 91 virtual bool OnProtocolVersionMismatch(QuicTag version) OVERRIDE; | 91 virtual bool OnProtocolVersionMismatch(QuicVersion version) OVERRIDE; |
| 92 virtual bool OnPacketHeader(const QuicPacketHeader& header) OVERRIDE; | 92 virtual bool OnPacketHeader(const QuicPacketHeader& header) OVERRIDE; |
| 93 virtual void OnFecProtectedPayload(base::StringPiece payload) OVERRIDE {} | 93 virtual void OnFecProtectedPayload(base::StringPiece payload) OVERRIDE {} |
| 94 virtual bool OnStreamFrame(const QuicStreamFrame& frame) OVERRIDE; | 94 virtual bool OnStreamFrame(const QuicStreamFrame& frame) OVERRIDE; |
| 95 virtual bool OnAckFrame(const QuicAckFrame& frame) OVERRIDE; | 95 virtual bool OnAckFrame(const QuicAckFrame& frame) OVERRIDE; |
| 96 virtual bool OnCongestionFeedbackFrame( | 96 virtual bool OnCongestionFeedbackFrame( |
| 97 const QuicCongestionFeedbackFrame& frame) OVERRIDE; | 97 const QuicCongestionFeedbackFrame& frame) OVERRIDE; |
| 98 virtual void OnFecData(const QuicFecData& fec) OVERRIDE {} | 98 virtual void OnFecData(const QuicFecData& fec) OVERRIDE {} |
| 99 virtual bool OnRstStreamFrame(const QuicRstStreamFrame& frame) OVERRIDE; | 99 virtual bool OnRstStreamFrame(const QuicRstStreamFrame& frame) OVERRIDE; |
| 100 virtual bool OnConnectionCloseFrame( | 100 virtual bool OnConnectionCloseFrame( |
| 101 const QuicConnectionCloseFrame& frame) OVERRIDE; | 101 const QuicConnectionCloseFrame& frame) OVERRIDE; |
| (...skipping 136 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 238 QuicStreamId last_good_stream_id, | 238 QuicStreamId last_good_stream_id, |
| 239 const string& reason)); | 239 const string& reason)); |
| 240 MOCK_METHOD0(OnCanWrite, bool()); | 240 MOCK_METHOD0(OnCanWrite, bool()); |
| 241 | 241 |
| 242 void ProcessUdpPacketInternal(const IPEndPoint& self_address, | 242 void ProcessUdpPacketInternal(const IPEndPoint& self_address, |
| 243 const IPEndPoint& peer_address, | 243 const IPEndPoint& peer_address, |
| 244 const QuicEncryptedPacket& packet) { | 244 const QuicEncryptedPacket& packet) { |
| 245 QuicConnection::ProcessUdpPacket(self_address, peer_address, packet); | 245 QuicConnection::ProcessUdpPacket(self_address, peer_address, packet); |
| 246 } | 246 } |
| 247 | 247 |
| 248 virtual bool OnProtocolVersionMismatch(QuicTag version) OVERRIDE { | 248 virtual bool OnProtocolVersionMismatch(QuicVersion version) OVERRIDE { |
| 249 return false; | 249 return false; |
| 250 } | 250 } |
| 251 | 251 |
| 252 private: | 252 private: |
| 253 const bool has_mock_helper_; | 253 const bool has_mock_helper_; |
| 254 | 254 |
| 255 DISALLOW_COPY_AND_ASSIGN(MockConnection); | 255 DISALLOW_COPY_AND_ASSIGN(MockConnection); |
| 256 }; | 256 }; |
| 257 | 257 |
| 258 class PacketSavingConnection : public MockConnection { | 258 class PacketSavingConnection : public MockConnection { |
| (...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 331 void(QuicPacketSequenceNumber, QuicByteCount, QuicTime::Delta)); | 331 void(QuicPacketSequenceNumber, QuicByteCount, QuicTime::Delta)); |
| 332 MOCK_METHOD1(OnIncomingLoss, void(QuicTime)); | 332 MOCK_METHOD1(OnIncomingLoss, void(QuicTime)); |
| 333 MOCK_METHOD4(SentPacket, void(QuicTime sent_time, QuicPacketSequenceNumber, | 333 MOCK_METHOD4(SentPacket, void(QuicTime sent_time, QuicPacketSequenceNumber, |
| 334 QuicByteCount, Retransmission)); | 334 QuicByteCount, Retransmission)); |
| 335 MOCK_METHOD2(AbandoningPacket, void(QuicPacketSequenceNumber sequence_number, | 335 MOCK_METHOD2(AbandoningPacket, void(QuicPacketSequenceNumber sequence_number, |
| 336 QuicByteCount abandoned_bytes)); | 336 QuicByteCount abandoned_bytes)); |
| 337 MOCK_METHOD3(TimeUntilSend, QuicTime::Delta(QuicTime now, Retransmission, | 337 MOCK_METHOD3(TimeUntilSend, QuicTime::Delta(QuicTime now, Retransmission, |
| 338 HasRetransmittableData)); | 338 HasRetransmittableData)); |
| 339 MOCK_METHOD0(BandwidthEstimate, QuicBandwidth(void)); | 339 MOCK_METHOD0(BandwidthEstimate, QuicBandwidth(void)); |
| 340 MOCK_METHOD0(SmoothedRtt, QuicTime::Delta(void)); | 340 MOCK_METHOD0(SmoothedRtt, QuicTime::Delta(void)); |
| 341 MOCK_METHOD0(RetransmissionDelay, QuicTime::Delta(void)); |
| 341 | 342 |
| 342 private: | 343 private: |
| 343 DISALLOW_COPY_AND_ASSIGN(MockSendAlgorithm); | 344 DISALLOW_COPY_AND_ASSIGN(MockSendAlgorithm); |
| 344 }; | 345 }; |
| 345 | 346 |
| 346 class TestEntropyCalculator : | 347 class TestEntropyCalculator : |
| 347 public QuicReceivedEntropyHashCalculatorInterface { | 348 public QuicReceivedEntropyHashCalculatorInterface { |
| 348 public: | 349 public: |
| 349 TestEntropyCalculator() { } | 350 TestEntropyCalculator() { } |
| 350 virtual ~TestEntropyCalculator() { } | 351 virtual ~TestEntropyCalculator() { } |
| 351 | 352 |
| 352 virtual QuicPacketEntropyHash ReceivedEntropyHash( | 353 virtual QuicPacketEntropyHash EntropyHash( |
| 353 QuicPacketSequenceNumber sequence_number) const OVERRIDE; | 354 QuicPacketSequenceNumber sequence_number) const OVERRIDE; |
| 354 }; | 355 }; |
| 355 | 356 |
| 356 class TestDecompressorVisitor : public QuicSpdyDecompressor::Visitor { | 357 class TestDecompressorVisitor : public QuicSpdyDecompressor::Visitor { |
| 357 public: | 358 public: |
| 358 virtual ~TestDecompressorVisitor() {} | 359 virtual ~TestDecompressorVisitor() {} |
| 359 virtual bool OnDecompressedData(base::StringPiece data) OVERRIDE; | 360 virtual bool OnDecompressedData(base::StringPiece data) OVERRIDE; |
| 360 virtual void OnDecompressionError() OVERRIDE; | 361 virtual void OnDecompressionError() OVERRIDE; |
| 361 | 362 |
| 362 string data() { return data_; } | 363 string data() { return data_; } |
| 363 bool error() { return error_; } | 364 bool error() { return error_; } |
| 364 | 365 |
| 365 private: | 366 private: |
| 366 string data_; | 367 string data_; |
| 367 bool error_; | 368 bool error_; |
| 368 }; | 369 }; |
| 369 | 370 |
| 370 } // namespace test | 371 } // namespace test |
| 371 } // namespace net | 372 } // namespace net |
| 372 | 373 |
| 373 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 374 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| OLD | NEW |