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 |