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 <stddef.h> | 10 #include <stddef.h> |
(...skipping 494 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
505 QuicCryptoStream* GetCryptoStream() override { return crypto_stream_.get(); } | 505 QuicCryptoStream* GetCryptoStream() override { return crypto_stream_.get(); } |
506 | 506 |
507 MOCK_METHOD3(OnConnectionClosed, | 507 MOCK_METHOD3(OnConnectionClosed, |
508 void(QuicErrorCode error, | 508 void(QuicErrorCode error, |
509 const std::string& error_details, | 509 const std::string& error_details, |
510 ConnectionCloseSource source)); | 510 ConnectionCloseSource source)); |
511 MOCK_METHOD1(CreateIncomingDynamicStream, QuicStream*(QuicStreamId id)); | 511 MOCK_METHOD1(CreateIncomingDynamicStream, QuicStream*(QuicStreamId id)); |
512 MOCK_METHOD1(CreateOutgoingDynamicStream, QuicStream*(SpdyPriority priority)); | 512 MOCK_METHOD1(CreateOutgoingDynamicStream, QuicStream*(SpdyPriority priority)); |
513 MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); | 513 MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); |
514 MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); | 514 MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); |
515 MOCK_METHOD6(WritevData, | 515 MOCK_METHOD6( |
516 QuicConsumedData(QuicStream* stream, | 516 WritevData, |
517 QuicStreamId id, | 517 QuicConsumedData(QuicStream* stream, |
518 QuicIOVector data, | 518 QuicStreamId id, |
519 QuicStreamOffset offset, | 519 QuicIOVector data, |
520 bool fin, | 520 QuicStreamOffset offset, |
521 QuicAckListenerInterface*)); | 521 bool fin, |
| 522 const scoped_refptr<QuicAckListenerInterface>&)); |
522 | 523 |
523 MOCK_METHOD3(SendRstStream, | 524 MOCK_METHOD3(SendRstStream, |
524 void(QuicStreamId stream_id, | 525 void(QuicStreamId stream_id, |
525 QuicRstStreamErrorCode error, | 526 QuicRstStreamErrorCode error, |
526 QuicStreamOffset bytes_written)); | 527 QuicStreamOffset bytes_written)); |
527 | 528 |
528 MOCK_METHOD2(OnStreamHeaders, | 529 MOCK_METHOD2(OnStreamHeaders, |
529 void(QuicStreamId stream_id, base::StringPiece headers_data)); | 530 void(QuicStreamId stream_id, base::StringPiece headers_data)); |
530 MOCK_METHOD2(OnStreamHeadersPriority, | 531 MOCK_METHOD2(OnStreamHeadersPriority, |
531 void(QuicStreamId stream_id, SpdyPriority priority)); | 532 void(QuicStreamId stream_id, SpdyPriority priority)); |
532 MOCK_METHOD3(OnStreamHeadersComplete, | 533 MOCK_METHOD3(OnStreamHeadersComplete, |
533 void(QuicStreamId stream_id, bool fin, size_t frame_len)); | 534 void(QuicStreamId stream_id, bool fin, size_t frame_len)); |
534 MOCK_METHOD0(IsCryptoHandshakeConfirmed, bool()); | 535 MOCK_METHOD0(IsCryptoHandshakeConfirmed, bool()); |
535 | 536 |
536 using QuicSession::ActivateStream; | 537 using QuicSession::ActivateStream; |
537 | 538 |
538 // Returns a QuicConsumedData that indicates all of |data| (and |fin| if set) | 539 // Returns a QuicConsumedData that indicates all of |data| (and |fin| if set) |
539 // has been consumed. | 540 // has been consumed. |
540 static QuicConsumedData ConsumeAllData( | 541 static QuicConsumedData ConsumeAllData( |
541 QuicStream* stream, | 542 QuicStream* stream, |
542 QuicStreamId id, | 543 QuicStreamId id, |
543 const QuicIOVector& data, | 544 const QuicIOVector& data, |
544 QuicStreamOffset offset, | 545 QuicStreamOffset offset, |
545 bool fin, | 546 bool fin, |
546 QuicAckListenerInterface* ack_notifier_delegate); | 547 const scoped_refptr<QuicAckListenerInterface>& ack_notifier_delegate); |
547 | 548 |
548 private: | 549 private: |
549 std::unique_ptr<QuicCryptoStream> crypto_stream_; | 550 std::unique_ptr<QuicCryptoStream> crypto_stream_; |
550 | 551 |
551 DISALLOW_COPY_AND_ASSIGN(MockQuicSession); | 552 DISALLOW_COPY_AND_ASSIGN(MockQuicSession); |
552 }; | 553 }; |
553 | 554 |
554 class MockQuicSpdySession : public QuicSpdySession { | 555 class MockQuicSpdySession : public QuicSpdySession { |
555 public: | 556 public: |
556 // Takes ownership of |connection|. | 557 // Takes ownership of |connection|. |
(...skipping 12 matching lines...) Expand all Loading... |
569 MOCK_METHOD1(CreateOutgoingDynamicStream, | 570 MOCK_METHOD1(CreateOutgoingDynamicStream, |
570 QuicSpdyStream*(SpdyPriority priority)); | 571 QuicSpdyStream*(SpdyPriority priority)); |
571 MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); | 572 MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); |
572 MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); | 573 MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); |
573 MOCK_METHOD6(WritevData, | 574 MOCK_METHOD6(WritevData, |
574 QuicConsumedData(QuicStream* stream, | 575 QuicConsumedData(QuicStream* stream, |
575 QuicStreamId id, | 576 QuicStreamId id, |
576 QuicIOVector data, | 577 QuicIOVector data, |
577 QuicStreamOffset offset, | 578 QuicStreamOffset offset, |
578 bool fin, | 579 bool fin, |
579 QuicAckListenerInterface*)); | 580 const scoped_refptr<QuicAckListenerInterface>& |
| 581 ack_listener)); |
580 | 582 |
581 MOCK_METHOD3(SendRstStream, | 583 MOCK_METHOD3(SendRstStream, |
582 void(QuicStreamId stream_id, | 584 void(QuicStreamId stream_id, |
583 QuicRstStreamErrorCode error, | 585 QuicRstStreamErrorCode error, |
584 QuicStreamOffset bytes_written)); | 586 QuicStreamOffset bytes_written)); |
585 | 587 |
586 MOCK_METHOD2(OnStreamHeaders, | 588 MOCK_METHOD2(OnStreamHeaders, |
587 void(QuicStreamId stream_id, StringPiece headers_data)); | 589 void(QuicStreamId stream_id, StringPiece headers_data)); |
588 MOCK_METHOD2(OnStreamHeadersPriority, | 590 MOCK_METHOD2(OnStreamHeadersPriority, |
589 void(QuicStreamId stream_id, SpdyPriority priority)); | 591 void(QuicStreamId stream_id, SpdyPriority priority)); |
(...skipping 11 matching lines...) Expand all Loading... |
601 void(QuicStreamId stream_id, | 603 void(QuicStreamId stream_id, |
602 QuicStreamId promised_stream_id, | 604 QuicStreamId promised_stream_id, |
603 size_t frame_len)); | 605 size_t frame_len)); |
604 MOCK_METHOD4(OnPromiseHeaderList, | 606 MOCK_METHOD4(OnPromiseHeaderList, |
605 void(QuicStreamId stream_id, | 607 void(QuicStreamId stream_id, |
606 QuicStreamId promised_stream_id, | 608 QuicStreamId promised_stream_id, |
607 size_t frame_len, | 609 size_t frame_len, |
608 const QuicHeaderList& header_list)); | 610 const QuicHeaderList& header_list)); |
609 // Methods taking non-copyable types like SpdyHeaderBlock by value cannot be | 611 // Methods taking non-copyable types like SpdyHeaderBlock by value cannot be |
610 // mocked directly. | 612 // mocked directly. |
611 size_t WriteHeaders( | 613 size_t WriteHeaders(QuicStreamId id, |
612 QuicStreamId id, | 614 SpdyHeaderBlock headers, |
613 SpdyHeaderBlock headers, | 615 bool fin, |
614 bool fin, | 616 SpdyPriority priority, |
615 SpdyPriority priority, | 617 const scoped_refptr<QuicAckListenerInterface>& |
616 QuicAckListenerInterface* ack_notifier_delegate) override { | 618 ack_notifier_delegate) override { |
617 write_headers_ = std::move(headers); | 619 write_headers_ = std::move(headers); |
618 return WriteHeadersMock(id, write_headers_, fin, priority, | 620 return WriteHeadersMock(id, write_headers_, fin, priority, |
619 ack_notifier_delegate); | 621 ack_notifier_delegate); |
620 } | 622 } |
621 MOCK_METHOD5(WriteHeadersMock, | 623 MOCK_METHOD5(WriteHeadersMock, |
622 size_t(QuicStreamId id, | 624 size_t(QuicStreamId id, |
623 const SpdyHeaderBlock& headers, | 625 const SpdyHeaderBlock& headers, |
624 bool fin, | 626 bool fin, |
625 SpdyPriority priority, | 627 SpdyPriority priority, |
626 QuicAckListenerInterface* ack_notifier_delegate)); | 628 const scoped_refptr<QuicAckListenerInterface>& |
| 629 ack_notifier_delegate)); |
627 MOCK_METHOD1(OnHeadersHeadOfLineBlocking, void(QuicTime::Delta delta)); | 630 MOCK_METHOD1(OnHeadersHeadOfLineBlocking, void(QuicTime::Delta delta)); |
628 MOCK_METHOD4( | 631 MOCK_METHOD4( |
629 OnStreamFrameData, | 632 OnStreamFrameData, |
630 void(QuicStreamId stream_id, const char* data, size_t len, bool fin)); | 633 void(QuicStreamId stream_id, const char* data, size_t len, bool fin)); |
631 | 634 |
632 using QuicSession::ActivateStream; | 635 using QuicSession::ActivateStream; |
633 | 636 |
634 private: | 637 private: |
635 std::unique_ptr<QuicCryptoStream> crypto_stream_; | 638 std::unique_ptr<QuicCryptoStream> crypto_stream_; |
636 SpdyHeaderBlock write_headers_; | 639 SpdyHeaderBlock write_headers_; |
(...skipping 417 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1054 iov->iov_base = const_cast<char*>(str.data()); | 1057 iov->iov_base = const_cast<char*>(str.data()); |
1055 iov->iov_len = static_cast<size_t>(str.size()); | 1058 iov->iov_len = static_cast<size_t>(str.size()); |
1056 QuicIOVector quic_iov(iov, 1, str.size()); | 1059 QuicIOVector quic_iov(iov, 1, str.size()); |
1057 return quic_iov; | 1060 return quic_iov; |
1058 } | 1061 } |
1059 | 1062 |
1060 } // namespace test | 1063 } // namespace test |
1061 } // namespace net | 1064 } // namespace net |
1062 | 1065 |
1063 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 1066 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
OLD | NEW |