| 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 480 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 491 void SendOrQueuePacket(SerializedPacket* packet) override; | 491 void SendOrQueuePacket(SerializedPacket* packet) override; |
| 492 | 492 |
| 493 std::vector<QuicEncryptedPacket*> encrypted_packets_; | 493 std::vector<QuicEncryptedPacket*> encrypted_packets_; |
| 494 | 494 |
| 495 private: | 495 private: |
| 496 DISALLOW_COPY_AND_ASSIGN(PacketSavingConnection); | 496 DISALLOW_COPY_AND_ASSIGN(PacketSavingConnection); |
| 497 }; | 497 }; |
| 498 | 498 |
| 499 class MockQuicSession : public QuicSession { | 499 class MockQuicSession : public QuicSession { |
| 500 public: | 500 public: |
| 501 // Takes ownership of |connection|. |
| 501 explicit MockQuicSession(QuicConnection* connection); | 502 explicit MockQuicSession(QuicConnection* connection); |
| 502 ~MockQuicSession() override; | 503 ~MockQuicSession() override; |
| 503 | 504 |
| 504 QuicCryptoStream* GetCryptoStream() override { return crypto_stream_.get(); } | 505 QuicCryptoStream* GetCryptoStream() override { return crypto_stream_.get(); } |
| 505 | 506 |
| 506 MOCK_METHOD3(OnConnectionClosed, | 507 MOCK_METHOD3(OnConnectionClosed, |
| 507 void(QuicErrorCode error, | 508 void(QuicErrorCode error, |
| 508 const std::string& error_details, | 509 const std::string& error_details, |
| 509 ConnectionCloseSource source)); | 510 ConnectionCloseSource source)); |
| 510 MOCK_METHOD1(CreateIncomingDynamicStream, | 511 MOCK_METHOD1(CreateIncomingDynamicStream, |
| (...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 547 QuicAckListenerInterface* ack_notifier_delegate); | 548 QuicAckListenerInterface* ack_notifier_delegate); |
| 548 | 549 |
| 549 private: | 550 private: |
| 550 std::unique_ptr<QuicCryptoStream> crypto_stream_; | 551 std::unique_ptr<QuicCryptoStream> crypto_stream_; |
| 551 | 552 |
| 552 DISALLOW_COPY_AND_ASSIGN(MockQuicSession); | 553 DISALLOW_COPY_AND_ASSIGN(MockQuicSession); |
| 553 }; | 554 }; |
| 554 | 555 |
| 555 class MockQuicSpdySession : public QuicSpdySession { | 556 class MockQuicSpdySession : public QuicSpdySession { |
| 556 public: | 557 public: |
| 558 // Takes ownership of |connection|. |
| 557 explicit MockQuicSpdySession(QuicConnection* connection); | 559 explicit MockQuicSpdySession(QuicConnection* connection); |
| 558 ~MockQuicSpdySession() override; | 560 ~MockQuicSpdySession() override; |
| 559 | 561 |
| 560 QuicCryptoStream* GetCryptoStream() override { return crypto_stream_.get(); } | 562 QuicCryptoStream* GetCryptoStream() override { return crypto_stream_.get(); } |
| 561 const SpdyHeaderBlock& GetWriteHeaders() { return write_headers_; } | 563 const SpdyHeaderBlock& GetWriteHeaders() { return write_headers_; } |
| 562 | 564 |
| 563 // From QuicSession. | 565 // From QuicSession. |
| 564 MOCK_METHOD3(OnConnectionClosed, | 566 MOCK_METHOD3(OnConnectionClosed, |
| 565 void(QuicErrorCode error, | 567 void(QuicErrorCode error, |
| 566 const std::string& error_details, | 568 const std::string& error_details, |
| (...skipping 66 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 633 | 635 |
| 634 private: | 636 private: |
| 635 std::unique_ptr<QuicCryptoStream> crypto_stream_; | 637 std::unique_ptr<QuicCryptoStream> crypto_stream_; |
| 636 SpdyHeaderBlock write_headers_; | 638 SpdyHeaderBlock write_headers_; |
| 637 | 639 |
| 638 DISALLOW_COPY_AND_ASSIGN(MockQuicSpdySession); | 640 DISALLOW_COPY_AND_ASSIGN(MockQuicSpdySession); |
| 639 }; | 641 }; |
| 640 | 642 |
| 641 class TestQuicSpdyServerSession : public QuicServerSessionBase { | 643 class TestQuicSpdyServerSession : public QuicServerSessionBase { |
| 642 public: | 644 public: |
| 645 // Takes ownership of |connection|. |
| 643 TestQuicSpdyServerSession(QuicConnection* connection, | 646 TestQuicSpdyServerSession(QuicConnection* connection, |
| 644 const QuicConfig& config, | 647 const QuicConfig& config, |
| 645 const QuicCryptoServerConfig* crypto_config, | 648 const QuicCryptoServerConfig* crypto_config, |
| 646 QuicCompressedCertsCache* compressed_certs_cache); | 649 QuicCompressedCertsCache* compressed_certs_cache); |
| 647 ~TestQuicSpdyServerSession() override; | 650 ~TestQuicSpdyServerSession() override; |
| 648 | 651 |
| 649 MOCK_METHOD1(CreateIncomingDynamicStream, QuicSpdyStream*(QuicStreamId id)); | 652 MOCK_METHOD1(CreateIncomingDynamicStream, QuicSpdyStream*(QuicStreamId id)); |
| 650 MOCK_METHOD1(CreateOutgoingDynamicStream, | 653 MOCK_METHOD1(CreateOutgoingDynamicStream, |
| 651 QuicSpdyStream*(SpdyPriority priority)); | 654 QuicSpdyStream*(SpdyPriority priority)); |
| 652 QuicCryptoServerStreamBase* CreateQuicCryptoServerStream( | 655 QuicCryptoServerStreamBase* CreateQuicCryptoServerStream( |
| (...skipping 365 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1018 TestQuicSpdyServerSession** server_session); | 1021 TestQuicSpdyServerSession** server_session); |
| 1019 | 1022 |
| 1020 // Helper to generate client side stream ids, generalizes | 1023 // Helper to generate client side stream ids, generalizes |
| 1021 // kClientDataStreamId1 etc. above. | 1024 // kClientDataStreamId1 etc. above. |
| 1022 QuicStreamId QuicClientDataStreamId(int i); | 1025 QuicStreamId QuicClientDataStreamId(int i); |
| 1023 | 1026 |
| 1024 } // namespace test | 1027 } // namespace test |
| 1025 } // namespace net | 1028 } // namespace net |
| 1026 | 1029 |
| 1027 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 1030 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| OLD | NEW |