| 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 <cstdint> | 10 #include <cstdint> |
| (...skipping 480 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 491 MOCK_METHOD1(CreateIncomingDynamicStream, QuicStream*(QuicStreamId id)); | 491 MOCK_METHOD1(CreateIncomingDynamicStream, QuicStream*(QuicStreamId id)); |
| 492 MOCK_METHOD1(CreateOutgoingDynamicStream, QuicStream*(SpdyPriority priority)); | 492 MOCK_METHOD1(CreateOutgoingDynamicStream, QuicStream*(SpdyPriority priority)); |
| 493 MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); | 493 MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); |
| 494 MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); | 494 MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); |
| 495 MOCK_METHOD6( | 495 MOCK_METHOD6( |
| 496 WritevData, | 496 WritevData, |
| 497 QuicConsumedData(QuicStream* stream, | 497 QuicConsumedData(QuicStream* stream, |
| 498 QuicStreamId id, | 498 QuicStreamId id, |
| 499 QuicIOVector data, | 499 QuicIOVector data, |
| 500 QuicStreamOffset offset, | 500 QuicStreamOffset offset, |
| 501 bool fin, | 501 StreamSendingState state, |
| 502 QuicReferenceCountedPointer<QuicAckListenerInterface>)); | 502 QuicReferenceCountedPointer<QuicAckListenerInterface>)); |
| 503 | 503 |
| 504 MOCK_METHOD3(SendRstStream, | 504 MOCK_METHOD3(SendRstStream, |
| 505 void(QuicStreamId stream_id, | 505 void(QuicStreamId stream_id, |
| 506 QuicRstStreamErrorCode error, | 506 QuicRstStreamErrorCode error, |
| 507 QuicStreamOffset bytes_written)); | 507 QuicStreamOffset bytes_written)); |
| 508 | 508 |
| 509 MOCK_METHOD2(OnStreamHeaders, | 509 MOCK_METHOD2(OnStreamHeaders, |
| 510 void(QuicStreamId stream_id, QuicStringPiece headers_data)); | 510 void(QuicStreamId stream_id, QuicStringPiece headers_data)); |
| 511 MOCK_METHOD2(OnStreamHeadersPriority, | 511 MOCK_METHOD2(OnStreamHeadersPriority, |
| 512 void(QuicStreamId stream_id, SpdyPriority priority)); | 512 void(QuicStreamId stream_id, SpdyPriority priority)); |
| 513 MOCK_METHOD3(OnStreamHeadersComplete, | 513 MOCK_METHOD3(OnStreamHeadersComplete, |
| 514 void(QuicStreamId stream_id, bool fin, size_t frame_len)); | 514 void(QuicStreamId stream_id, bool fin, size_t frame_len)); |
| 515 MOCK_CONST_METHOD0(IsCryptoHandshakeConfirmed, bool()); | 515 MOCK_CONST_METHOD0(IsCryptoHandshakeConfirmed, bool()); |
| 516 | 516 |
| 517 using QuicSession::ActivateStream; | 517 using QuicSession::ActivateStream; |
| 518 | 518 |
| 519 // Returns a QuicConsumedData that indicates all of |data| (and |fin| if set) | 519 // Returns a QuicConsumedData that indicates all of |data| (and |fin| if set) |
| 520 // has been consumed. | 520 // has been consumed. |
| 521 static QuicConsumedData ConsumeAllData( | 521 static QuicConsumedData ConsumeAllData( |
| 522 QuicStream* stream, | 522 QuicStream* stream, |
| 523 QuicStreamId id, | 523 QuicStreamId id, |
| 524 const QuicIOVector& data, | 524 const QuicIOVector& data, |
| 525 QuicStreamOffset offset, | 525 QuicStreamOffset offset, |
| 526 bool fin, | 526 StreamSendingState state, |
| 527 const QuicReferenceCountedPointer<QuicAckListenerInterface>& | 527 const QuicReferenceCountedPointer<QuicAckListenerInterface>& |
| 528 ack_listener); | 528 ack_listener); |
| 529 | 529 |
| 530 private: | 530 private: |
| 531 std::unique_ptr<QuicCryptoStream> crypto_stream_; | 531 std::unique_ptr<QuicCryptoStream> crypto_stream_; |
| 532 | 532 |
| 533 DISALLOW_COPY_AND_ASSIGN(MockQuicSession); | 533 DISALLOW_COPY_AND_ASSIGN(MockQuicSession); |
| 534 }; | 534 }; |
| 535 | 535 |
| 536 class MockQuicSpdySession : public QuicSpdySession { | 536 class MockQuicSpdySession : public QuicSpdySession { |
| (...skipping 20 matching lines...) Expand all Loading... |
| 557 QuicSpdyStream*(SpdyPriority priority)); | 557 QuicSpdyStream*(SpdyPriority priority)); |
| 558 MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); | 558 MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); |
| 559 MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); | 559 MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); |
| 560 MOCK_METHOD6( | 560 MOCK_METHOD6( |
| 561 WritevData, | 561 WritevData, |
| 562 QuicConsumedData( | 562 QuicConsumedData( |
| 563 QuicStream* stream, | 563 QuicStream* stream, |
| 564 QuicStreamId id, | 564 QuicStreamId id, |
| 565 QuicIOVector data, | 565 QuicIOVector data, |
| 566 QuicStreamOffset offset, | 566 QuicStreamOffset offset, |
| 567 bool fin, | 567 StreamSendingState state, |
| 568 QuicReferenceCountedPointer<QuicAckListenerInterface> ack_listener)); | 568 QuicReferenceCountedPointer<QuicAckListenerInterface> ack_listener)); |
| 569 | 569 |
| 570 MOCK_METHOD3(SendRstStream, | 570 MOCK_METHOD3(SendRstStream, |
| 571 void(QuicStreamId stream_id, | 571 void(QuicStreamId stream_id, |
| 572 QuicRstStreamErrorCode error, | 572 QuicRstStreamErrorCode error, |
| 573 QuicStreamOffset bytes_written)); | 573 QuicStreamOffset bytes_written)); |
| 574 | 574 |
| 575 MOCK_METHOD2(OnStreamHeaders, | 575 MOCK_METHOD2(OnStreamHeaders, |
| 576 void(QuicStreamId stream_id, QuicStringPiece headers_data)); | 576 void(QuicStreamId stream_id, QuicStringPiece headers_data)); |
| 577 MOCK_METHOD2(OnStreamHeadersPriority, | 577 MOCK_METHOD2(OnStreamHeadersPriority, |
| (...skipping 417 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 995 iov->iov_base = const_cast<char*>(str.data()); | 995 iov->iov_base = const_cast<char*>(str.data()); |
| 996 iov->iov_len = static_cast<size_t>(str.size()); | 996 iov->iov_len = static_cast<size_t>(str.size()); |
| 997 QuicIOVector quic_iov(iov, 1, str.size()); | 997 QuicIOVector quic_iov(iov, 1, str.size()); |
| 998 return quic_iov; | 998 return quic_iov; |
| 999 } | 999 } |
| 1000 | 1000 |
| 1001 } // namespace test | 1001 } // namespace test |
| 1002 } // namespace net | 1002 } // namespace net |
| 1003 | 1003 |
| 1004 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 1004 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| OLD | NEW |