| 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 |