| 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 166 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 177 | 177 |
| 178 MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, | 178 MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, |
| 179 const IPEndPoint& peer_address, | 179 const IPEndPoint& peer_address, |
| 180 const QuicPacketHeader& header, | 180 const QuicPacketHeader& header, |
| 181 const std::vector<QuicStreamFrame>& frame)); | 181 const std::vector<QuicStreamFrame>& frame)); |
| 182 MOCK_METHOD1(OnRstStream, void(const QuicRstStreamFrame& frame)); | 182 MOCK_METHOD1(OnRstStream, void(const QuicRstStreamFrame& frame)); |
| 183 MOCK_METHOD1(OnGoAway, void(const QuicGoAwayFrame& frame)); | 183 MOCK_METHOD1(OnGoAway, void(const QuicGoAwayFrame& frame)); |
| 184 MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); | 184 MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); |
| 185 MOCK_METHOD1(OnAck, void(const SequenceNumberSet& acked_packets)); | 185 MOCK_METHOD1(OnAck, void(const SequenceNumberSet& acked_packets)); |
| 186 MOCK_METHOD0(OnCanWrite, bool()); | 186 MOCK_METHOD0(OnCanWrite, bool()); |
| 187 MOCK_METHOD1(OnSuccessfulVersionNegotiation, |
| 188 void(const QuicVersion& version)); |
| 187 | 189 |
| 188 private: | 190 private: |
| 189 DISALLOW_COPY_AND_ASSIGN(MockConnectionVisitor); | 191 DISALLOW_COPY_AND_ASSIGN(MockConnectionVisitor); |
| 190 }; | 192 }; |
| 191 | 193 |
| 192 class MockHelper : public QuicConnectionHelperInterface { | 194 class MockHelper : public QuicConnectionHelperInterface { |
| 193 public: | 195 public: |
| 194 MockHelper(); | 196 MockHelper(); |
| 195 virtual ~MockHelper(); | 197 virtual ~MockHelper(); |
| 196 | 198 |
| (...skipping 81 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 278 | 280 |
| 279 MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, | 281 MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, |
| 280 const IPEndPoint& peer_address, | 282 const IPEndPoint& peer_address, |
| 281 const QuicPacketHeader& header, | 283 const QuicPacketHeader& header, |
| 282 const std::vector<QuicStreamFrame>& frame)); | 284 const std::vector<QuicStreamFrame>& frame)); |
| 283 MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); | 285 MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); |
| 284 MOCK_METHOD1(CreateIncomingReliableStream, | 286 MOCK_METHOD1(CreateIncomingReliableStream, |
| 285 ReliableQuicStream*(QuicStreamId id)); | 287 ReliableQuicStream*(QuicStreamId id)); |
| 286 MOCK_METHOD0(GetCryptoStream, QuicCryptoStream*()); | 288 MOCK_METHOD0(GetCryptoStream, QuicCryptoStream*()); |
| 287 MOCK_METHOD0(CreateOutgoingReliableStream, ReliableQuicStream*()); | 289 MOCK_METHOD0(CreateOutgoingReliableStream, ReliableQuicStream*()); |
| 288 MOCK_METHOD4(WriteData, QuicConsumedData(QuicStreamId id, | 290 MOCK_METHOD5(WritevData, QuicConsumedData(QuicStreamId id, |
| 289 base::StringPiece data, | 291 const struct iovec* iov, |
| 290 QuicStreamOffset offset, | 292 int count, |
| 291 bool fin)); | 293 QuicStreamOffset offset, |
| 294 bool fin)); |
| 292 MOCK_METHOD0(IsHandshakeComplete, bool()); | 295 MOCK_METHOD0(IsHandshakeComplete, bool()); |
| 293 | 296 |
| 294 private: | 297 private: |
| 295 DISALLOW_COPY_AND_ASSIGN(MockSession); | 298 DISALLOW_COPY_AND_ASSIGN(MockSession); |
| 296 }; | 299 }; |
| 297 | 300 |
| 298 class TestSession : public QuicSession { | 301 class TestSession : public QuicSession { |
| 299 public: | 302 public: |
| 300 TestSession(QuicConnection* connection, | 303 TestSession(QuicConnection* connection, |
| 301 const QuicConfig& config, | 304 const QuicConfig& config, |
| (...skipping 18 matching lines...) Expand all Loading... |
| 320 MockSendAlgorithm(); | 323 MockSendAlgorithm(); |
| 321 virtual ~MockSendAlgorithm(); | 324 virtual ~MockSendAlgorithm(); |
| 322 | 325 |
| 323 MOCK_METHOD3(OnIncomingQuicCongestionFeedbackFrame, | 326 MOCK_METHOD3(OnIncomingQuicCongestionFeedbackFrame, |
| 324 void(const QuicCongestionFeedbackFrame&, | 327 void(const QuicCongestionFeedbackFrame&, |
| 325 QuicTime feedback_receive_time, | 328 QuicTime feedback_receive_time, |
| 326 const SentPacketsMap&)); | 329 const SentPacketsMap&)); |
| 327 MOCK_METHOD3(OnIncomingAck, | 330 MOCK_METHOD3(OnIncomingAck, |
| 328 void(QuicPacketSequenceNumber, QuicByteCount, QuicTime::Delta)); | 331 void(QuicPacketSequenceNumber, QuicByteCount, QuicTime::Delta)); |
| 329 MOCK_METHOD1(OnIncomingLoss, void(QuicTime)); | 332 MOCK_METHOD1(OnIncomingLoss, void(QuicTime)); |
| 330 MOCK_METHOD4(SentPacket, void(QuicTime sent_time, QuicPacketSequenceNumber, | 333 MOCK_METHOD5(SentPacket, |
| 331 QuicByteCount, Retransmission)); | 334 bool(QuicTime sent_time, QuicPacketSequenceNumber, QuicByteCount, |
| 335 Retransmission, HasRetransmittableData)); |
| 332 MOCK_METHOD2(AbandoningPacket, void(QuicPacketSequenceNumber sequence_number, | 336 MOCK_METHOD2(AbandoningPacket, void(QuicPacketSequenceNumber sequence_number, |
| 333 QuicByteCount abandoned_bytes)); | 337 QuicByteCount abandoned_bytes)); |
| 334 MOCK_METHOD4(TimeUntilSend, QuicTime::Delta(QuicTime now, Retransmission, | 338 MOCK_METHOD4(TimeUntilSend, QuicTime::Delta(QuicTime now, Retransmission, |
| 335 HasRetransmittableData, | 339 HasRetransmittableData, |
| 336 IsHandshake)); | 340 IsHandshake)); |
| 337 MOCK_METHOD0(BandwidthEstimate, QuicBandwidth(void)); | 341 MOCK_METHOD0(BandwidthEstimate, QuicBandwidth(void)); |
| 338 MOCK_METHOD0(SmoothedRtt, QuicTime::Delta(void)); | 342 MOCK_METHOD0(SmoothedRtt, QuicTime::Delta(void)); |
| 339 MOCK_METHOD0(RetransmissionDelay, QuicTime::Delta(void)); | 343 MOCK_METHOD0(RetransmissionDelay, QuicTime::Delta(void)); |
| 340 | 344 |
| 341 private: | 345 private: |
| (...skipping 29 matching lines...) Expand all Loading... |
| 371 MockAckNotifierDelegate(); | 375 MockAckNotifierDelegate(); |
| 372 virtual ~MockAckNotifierDelegate(); | 376 virtual ~MockAckNotifierDelegate(); |
| 373 | 377 |
| 374 MOCK_METHOD0(OnAckNotification, void()); | 378 MOCK_METHOD0(OnAckNotification, void()); |
| 375 }; | 379 }; |
| 376 | 380 |
| 377 } // namespace test | 381 } // namespace test |
| 378 } // namespace net | 382 } // namespace net |
| 379 | 383 |
| 380 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 384 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| OLD | NEW |