| 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 105 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 364 | 364 |
| 365 private: | 365 private: |
| 366 string data_; | 366 string data_; |
| 367 bool error_; | 367 bool error_; |
| 368 }; | 368 }; |
| 369 | 369 |
| 370 } // namespace test | 370 } // namespace test |
| 371 } // namespace net | 371 } // namespace net |
| 372 | 372 |
| 373 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 373 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| OLD | NEW |