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 |