| 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 495 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 506 // Takes ownership of |connection|. | 506 // Takes ownership of |connection|. |
| 507 explicit MockQuicSession(QuicConnection* connection); | 507 explicit MockQuicSession(QuicConnection* connection); |
| 508 ~MockQuicSession() override; | 508 ~MockQuicSession() override; |
| 509 | 509 |
| 510 QuicCryptoStream* GetCryptoStream() override { return crypto_stream_.get(); } | 510 QuicCryptoStream* GetCryptoStream() override { return crypto_stream_.get(); } |
| 511 | 511 |
| 512 MOCK_METHOD3(OnConnectionClosed, | 512 MOCK_METHOD3(OnConnectionClosed, |
| 513 void(QuicErrorCode error, | 513 void(QuicErrorCode error, |
| 514 const std::string& error_details, | 514 const std::string& error_details, |
| 515 ConnectionCloseSource source)); | 515 ConnectionCloseSource source)); |
| 516 MOCK_METHOD1(CreateIncomingDynamicStream, | 516 MOCK_METHOD1(CreateIncomingDynamicStream, QuicStream*(QuicStreamId id)); |
| 517 ReliableQuicStream*(QuicStreamId id)); | 517 MOCK_METHOD1(CreateOutgoingDynamicStream, QuicStream*(SpdyPriority priority)); |
| 518 MOCK_METHOD1(CreateOutgoingDynamicStream, | |
| 519 ReliableQuicStream*(SpdyPriority priority)); | |
| 520 MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); | 518 MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); |
| 521 MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); | 519 MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); |
| 522 MOCK_METHOD6(WritevData, | 520 MOCK_METHOD6(WritevData, |
| 523 QuicConsumedData(ReliableQuicStream* stream, | 521 QuicConsumedData(QuicStream* stream, |
| 524 QuicStreamId id, | 522 QuicStreamId id, |
| 525 QuicIOVector data, | 523 QuicIOVector data, |
| 526 QuicStreamOffset offset, | 524 QuicStreamOffset offset, |
| 527 bool fin, | 525 bool fin, |
| 528 QuicAckListenerInterface*)); | 526 QuicAckListenerInterface*)); |
| 529 | 527 |
| 530 MOCK_METHOD3(SendRstStream, | 528 MOCK_METHOD3(SendRstStream, |
| 531 void(QuicStreamId stream_id, | 529 void(QuicStreamId stream_id, |
| 532 QuicRstStreamErrorCode error, | 530 QuicRstStreamErrorCode error, |
| 533 QuicStreamOffset bytes_written)); | 531 QuicStreamOffset bytes_written)); |
| 534 | 532 |
| 535 MOCK_METHOD2(OnStreamHeaders, | 533 MOCK_METHOD2(OnStreamHeaders, |
| 536 void(QuicStreamId stream_id, base::StringPiece headers_data)); | 534 void(QuicStreamId stream_id, base::StringPiece headers_data)); |
| 537 MOCK_METHOD2(OnStreamHeadersPriority, | 535 MOCK_METHOD2(OnStreamHeadersPriority, |
| 538 void(QuicStreamId stream_id, SpdyPriority priority)); | 536 void(QuicStreamId stream_id, SpdyPriority priority)); |
| 539 MOCK_METHOD3(OnStreamHeadersComplete, | 537 MOCK_METHOD3(OnStreamHeadersComplete, |
| 540 void(QuicStreamId stream_id, bool fin, size_t frame_len)); | 538 void(QuicStreamId stream_id, bool fin, size_t frame_len)); |
| 541 MOCK_METHOD0(IsCryptoHandshakeConfirmed, bool()); | 539 MOCK_METHOD0(IsCryptoHandshakeConfirmed, bool()); |
| 542 | 540 |
| 543 using QuicSession::ActivateStream; | 541 using QuicSession::ActivateStream; |
| 544 | 542 |
| 545 // Returns a QuicConsumedData that indicates all of |data| (and |fin| if set) | 543 // Returns a QuicConsumedData that indicates all of |data| (and |fin| if set) |
| 546 // has been consumed. | 544 // has been consumed. |
| 547 static QuicConsumedData ConsumeAllData( | 545 static QuicConsumedData ConsumeAllData( |
| 548 ReliableQuicStream* stream, | 546 QuicStream* stream, |
| 549 QuicStreamId id, | 547 QuicStreamId id, |
| 550 const QuicIOVector& data, | 548 const QuicIOVector& data, |
| 551 QuicStreamOffset offset, | 549 QuicStreamOffset offset, |
| 552 bool fin, | 550 bool fin, |
| 553 QuicAckListenerInterface* ack_notifier_delegate); | 551 QuicAckListenerInterface* ack_notifier_delegate); |
| 554 | 552 |
| 555 private: | 553 private: |
| 556 std::unique_ptr<QuicCryptoStream> crypto_stream_; | 554 std::unique_ptr<QuicCryptoStream> crypto_stream_; |
| 557 | 555 |
| 558 DISALLOW_COPY_AND_ASSIGN(MockQuicSession); | 556 DISALLOW_COPY_AND_ASSIGN(MockQuicSession); |
| (...skipping 12 matching lines...) Expand all Loading... |
| 571 MOCK_METHOD3(OnConnectionClosed, | 569 MOCK_METHOD3(OnConnectionClosed, |
| 572 void(QuicErrorCode error, | 570 void(QuicErrorCode error, |
| 573 const std::string& error_details, | 571 const std::string& error_details, |
| 574 ConnectionCloseSource source)); | 572 ConnectionCloseSource source)); |
| 575 MOCK_METHOD1(CreateIncomingDynamicStream, QuicSpdyStream*(QuicStreamId id)); | 573 MOCK_METHOD1(CreateIncomingDynamicStream, QuicSpdyStream*(QuicStreamId id)); |
| 576 MOCK_METHOD1(CreateOutgoingDynamicStream, | 574 MOCK_METHOD1(CreateOutgoingDynamicStream, |
| 577 QuicSpdyStream*(SpdyPriority priority)); | 575 QuicSpdyStream*(SpdyPriority priority)); |
| 578 MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); | 576 MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); |
| 579 MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); | 577 MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); |
| 580 MOCK_METHOD6(WritevData, | 578 MOCK_METHOD6(WritevData, |
| 581 QuicConsumedData(ReliableQuicStream* stream, | 579 QuicConsumedData(QuicStream* stream, |
| 582 QuicStreamId id, | 580 QuicStreamId id, |
| 583 QuicIOVector data, | 581 QuicIOVector data, |
| 584 QuicStreamOffset offset, | 582 QuicStreamOffset offset, |
| 585 bool fin, | 583 bool fin, |
| 586 QuicAckListenerInterface*)); | 584 QuicAckListenerInterface*)); |
| 587 | 585 |
| 588 MOCK_METHOD3(SendRstStream, | 586 MOCK_METHOD3(SendRstStream, |
| 589 void(QuicStreamId stream_id, | 587 void(QuicStreamId stream_id, |
| 590 QuicRstStreamErrorCode error, | 588 QuicRstStreamErrorCode error, |
| 591 QuicStreamOffset bytes_written)); | 589 QuicStreamOffset bytes_written)); |
| (...skipping 465 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1057 l.OnHeader(p.first, p.second); | 1055 l.OnHeader(p.first, p.second); |
| 1058 } | 1056 } |
| 1059 l.OnHeaderBlockEnd(total_size); | 1057 l.OnHeaderBlockEnd(total_size); |
| 1060 return l; | 1058 return l; |
| 1061 } | 1059 } |
| 1062 | 1060 |
| 1063 } // namespace test | 1061 } // namespace test |
| 1064 } // namespace net | 1062 } // namespace net |
| 1065 | 1063 |
| 1066 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 1064 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| OLD | NEW |