| 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 408 matching lines...) Expand 10 before | Expand all | Expand 10 after  Loading... | 
| 419   void QuicConnection_OnError(QuicFramer* framer) { | 419   void QuicConnection_OnError(QuicFramer* framer) { | 
| 420     QuicConnection::OnError(framer); | 420     QuicConnection::OnError(framer); | 
| 421   } | 421   } | 
| 422 | 422 | 
| 423   void ReallyProcessUdpPacket(const QuicSocketAddress& self_address, | 423   void ReallyProcessUdpPacket(const QuicSocketAddress& self_address, | 
| 424                               const QuicSocketAddress& peer_address, | 424                               const QuicSocketAddress& peer_address, | 
| 425                               const QuicReceivedPacket& packet) { | 425                               const QuicReceivedPacket& packet) { | 
| 426     QuicConnection::ProcessUdpPacket(self_address, peer_address, packet); | 426     QuicConnection::ProcessUdpPacket(self_address, peer_address, packet); | 
| 427   } | 427   } | 
| 428 | 428 | 
| 429   bool OnProtocolVersionMismatch(QuicVersion version) override { return false; } | 429   bool OnProtocolVersionMismatch(QuicVersion version) override; | 
| 430 | 430 | 
| 431   void ReallySendGoAway(QuicErrorCode error, | 431   void ReallySendGoAway(QuicErrorCode error, | 
| 432                         QuicStreamId last_good_stream_id, | 432                         QuicStreamId last_good_stream_id, | 
| 433                         const std::string& reason) { | 433                         const std::string& reason) { | 
| 434     QuicConnection::SendGoAway(error, last_good_stream_id, reason); | 434     QuicConnection::SendGoAway(error, last_good_stream_id, reason); | 
| 435   } | 435   } | 
| 436 | 436 | 
| 437  private: | 437  private: | 
| 438   DISALLOW_COPY_AND_ASSIGN(MockQuicConnection); | 438   DISALLOW_COPY_AND_ASSIGN(MockQuicConnection); | 
| 439 }; | 439 }; | 
| (...skipping 18 matching lines...) Expand all  Loading... | 
| 458  private: | 458  private: | 
| 459   DISALLOW_COPY_AND_ASSIGN(PacketSavingConnection); | 459   DISALLOW_COPY_AND_ASSIGN(PacketSavingConnection); | 
| 460 }; | 460 }; | 
| 461 | 461 | 
| 462 class MockQuicSession : public QuicSession { | 462 class MockQuicSession : public QuicSession { | 
| 463  public: | 463  public: | 
| 464   // Takes ownership of |connection|. | 464   // Takes ownership of |connection|. | 
| 465   explicit MockQuicSession(QuicConnection* connection); | 465   explicit MockQuicSession(QuicConnection* connection); | 
| 466   ~MockQuicSession() override; | 466   ~MockQuicSession() override; | 
| 467 | 467 | 
| 468   QuicCryptoStream* GetMutableCryptoStream() override { | 468   QuicCryptoStream* GetMutableCryptoStream() override; | 
| 469     return crypto_stream_.get(); | 469   const QuicCryptoStream* GetCryptoStream() const override; | 
| 470   } |  | 
| 471 |  | 
| 472   const QuicCryptoStream* GetCryptoStream() const override { |  | 
| 473     return crypto_stream_.get(); |  | 
| 474   } |  | 
| 475 | 470 | 
| 476   MOCK_METHOD3(OnConnectionClosed, | 471   MOCK_METHOD3(OnConnectionClosed, | 
| 477                void(QuicErrorCode error, | 472                void(QuicErrorCode error, | 
| 478                     const std::string& error_details, | 473                     const std::string& error_details, | 
| 479                     ConnectionCloseSource source)); | 474                     ConnectionCloseSource source)); | 
| 480   MOCK_METHOD1(CreateIncomingDynamicStream, QuicStream*(QuicStreamId id)); | 475   MOCK_METHOD1(CreateIncomingDynamicStream, QuicStream*(QuicStreamId id)); | 
| 481   MOCK_METHOD1(CreateOutgoingDynamicStream, QuicStream*(SpdyPriority priority)); | 476   MOCK_METHOD1(CreateOutgoingDynamicStream, QuicStream*(SpdyPriority priority)); | 
| 482   MOCK_METHOD1(MaybeCreateIncomingDynamicStream, QuicStream*(QuicStreamId id)); | 477   MOCK_METHOD1(MaybeCreateIncomingDynamicStream, QuicStream*(QuicStreamId id)); | 
| 483   MOCK_METHOD1(MaybeCreateOutgoingDynamicStream, | 478   MOCK_METHOD1(MaybeCreateOutgoingDynamicStream, | 
| 484                QuicStream*(SpdyPriority priority)); | 479                QuicStream*(SpdyPriority priority)); | 
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after  Loading... | 
| 525 | 520 | 
| 526   DISALLOW_COPY_AND_ASSIGN(MockQuicSession); | 521   DISALLOW_COPY_AND_ASSIGN(MockQuicSession); | 
| 527 }; | 522 }; | 
| 528 | 523 | 
| 529 class MockQuicSpdySession : public QuicSpdySession { | 524 class MockQuicSpdySession : public QuicSpdySession { | 
| 530  public: | 525  public: | 
| 531   // Takes ownership of |connection|. | 526   // Takes ownership of |connection|. | 
| 532   explicit MockQuicSpdySession(QuicConnection* connection); | 527   explicit MockQuicSpdySession(QuicConnection* connection); | 
| 533   ~MockQuicSpdySession() override; | 528   ~MockQuicSpdySession() override; | 
| 534 | 529 | 
| 535   QuicCryptoStream* GetMutableCryptoStream() override { | 530   QuicCryptoStream* GetMutableCryptoStream() override; | 
| 536     return crypto_stream_.get(); | 531   const QuicCryptoStream* GetCryptoStream() const override; | 
| 537   } |  | 
| 538   const QuicCryptoStream* GetCryptoStream() const override { |  | 
| 539     return crypto_stream_.get(); |  | 
| 540   } |  | 
| 541   const SpdyHeaderBlock& GetWriteHeaders() { return write_headers_; } | 532   const SpdyHeaderBlock& GetWriteHeaders() { return write_headers_; } | 
| 542 | 533 | 
| 543   // From QuicSession. | 534   // From QuicSession. | 
| 544   MOCK_METHOD3(OnConnectionClosed, | 535   MOCK_METHOD3(OnConnectionClosed, | 
| 545                void(QuicErrorCode error, | 536                void(QuicErrorCode error, | 
| 546                     const std::string& error_details, | 537                     const std::string& error_details, | 
| 547                     ConnectionCloseSource source)); | 538                     ConnectionCloseSource source)); | 
| 548   MOCK_METHOD1(CreateIncomingDynamicStream, QuicSpdyStream*(QuicStreamId id)); | 539   MOCK_METHOD1(CreateIncomingDynamicStream, QuicSpdyStream*(QuicStreamId id)); | 
| 549   MOCK_METHOD1(CreateOutgoingDynamicStream, | 540   MOCK_METHOD1(CreateOutgoingDynamicStream, | 
| 550                QuicSpdyStream*(SpdyPriority priority)); | 541                QuicSpdyStream*(SpdyPriority priority)); | 
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after  Loading... | 
| 589                     QuicStreamId promised_stream_id, | 580                     QuicStreamId promised_stream_id, | 
| 590                     size_t frame_len, | 581                     size_t frame_len, | 
| 591                     const QuicHeaderList& header_list)); | 582                     const QuicHeaderList& header_list)); | 
| 592   // Methods taking non-copyable types like SpdyHeaderBlock by value cannot be | 583   // Methods taking non-copyable types like SpdyHeaderBlock by value cannot be | 
| 593   // mocked directly. | 584   // mocked directly. | 
| 594   size_t WriteHeaders(QuicStreamId id, | 585   size_t WriteHeaders(QuicStreamId id, | 
| 595                       SpdyHeaderBlock headers, | 586                       SpdyHeaderBlock headers, | 
| 596                       bool fin, | 587                       bool fin, | 
| 597                       SpdyPriority priority, | 588                       SpdyPriority priority, | 
| 598                       QuicReferenceCountedPointer<QuicAckListenerInterface> | 589                       QuicReferenceCountedPointer<QuicAckListenerInterface> | 
| 599                           ack_listener) override { | 590                           ack_listener) override; | 
| 600     write_headers_ = std::move(headers); |  | 
| 601     return WriteHeadersMock(id, write_headers_, fin, priority, ack_listener); |  | 
| 602   } |  | 
| 603   MOCK_METHOD5( | 591   MOCK_METHOD5( | 
| 604       WriteHeadersMock, | 592       WriteHeadersMock, | 
| 605       size_t(QuicStreamId id, | 593       size_t(QuicStreamId id, | 
| 606              const SpdyHeaderBlock& headers, | 594              const SpdyHeaderBlock& headers, | 
| 607              bool fin, | 595              bool fin, | 
| 608              SpdyPriority priority, | 596              SpdyPriority priority, | 
| 609              const QuicReferenceCountedPointer<QuicAckListenerInterface>& | 597              const QuicReferenceCountedPointer<QuicAckListenerInterface>& | 
| 610                  ack_listener)); | 598                  ack_listener)); | 
| 611   MOCK_METHOD1(OnHeadersHeadOfLineBlocking, void(QuicTime::Delta delta)); | 599   MOCK_METHOD1(OnHeadersHeadOfLineBlocking, void(QuicTime::Delta delta)); | 
| 612   MOCK_METHOD4( | 600   MOCK_METHOD4( | 
| (...skipping 398 matching lines...) Expand 10 before | Expand all | Expand 10 after  Loading... | 
| 1011 // Utilities that will adapt stream ids when http stream pairs are | 999 // Utilities that will adapt stream ids when http stream pairs are | 
| 1012 // enabled. | 1000 // enabled. | 
| 1013 QuicStreamId NextStreamId(QuicVersion version); | 1001 QuicStreamId NextStreamId(QuicVersion version); | 
| 1014 QuicStreamId GetNthClientInitiatedStreamId(QuicVersion version, int n); | 1002 QuicStreamId GetNthClientInitiatedStreamId(QuicVersion version, int n); | 
| 1015 QuicStreamId GetNthServerInitiatedStreamId(QuicVersion version, int n); | 1003 QuicStreamId GetNthServerInitiatedStreamId(QuicVersion version, int n); | 
| 1016 | 1004 | 
| 1017 }  // namespace test | 1005 }  // namespace test | 
| 1018 }  // namespace net | 1006 }  // namespace net | 
| 1019 | 1007 | 
| 1020 #endif  // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 1008 #endif  // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 
| OLD | NEW | 
|---|