Index: net/quic/test_tools/quic_test_utils.h |
diff --git a/net/quic/test_tools/quic_test_utils.h b/net/quic/test_tools/quic_test_utils.h |
index bb7423c8967213abfbf0fc5e9983417da572857e..f3c0cea7c8a29311a00e41ce7dbf725ad9dc6a65 100644 |
--- a/net/quic/test_tools/quic_test_utils.h |
+++ b/net/quic/test_tools/quic_test_utils.h |
@@ -24,6 +24,7 @@ |
#include "net/quic/core/quic_sent_packet_manager.h" |
#include "net/quic/core/quic_server_session_base.h" |
#include "net/quic/core/quic_simple_buffer_allocator.h" |
+#include "net/quic/platform/api/quic_ptr_util.h" |
#include "net/quic/platform/api/quic_string_piece.h" |
#include "net/quic/test_tools/mock_clock.h" |
#include "net/quic/test_tools/mock_random.h" |
@@ -478,6 +479,9 @@ class MockQuicSession : public QuicSession { |
ConnectionCloseSource source)); |
MOCK_METHOD1(CreateIncomingDynamicStream, QuicStream*(QuicStreamId id)); |
MOCK_METHOD1(CreateOutgoingDynamicStream, QuicStream*(SpdyPriority priority)); |
+ MOCK_METHOD1(MaybeCreateIncomingDynamicStream, QuicStream*(QuicStreamId id)); |
+ MOCK_METHOD1(MaybeCreateOutgoingDynamicStream, |
+ QuicStream*(SpdyPriority priority)); |
MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); |
MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); |
MOCK_METHOD6( |
@@ -501,6 +505,7 @@ class MockQuicSession : public QuicSession { |
MOCK_METHOD3(OnStreamHeadersComplete, |
void(QuicStreamId stream_id, bool fin, size_t frame_len)); |
MOCK_CONST_METHOD0(IsCryptoHandshakeConfirmed, bool()); |
+ MOCK_METHOD1(CreateStream, std::unique_ptr<QuicStream>(QuicStreamId id)); |
using QuicSession::ActivateStream; |
@@ -545,6 +550,7 @@ class MockQuicSpdySession : public QuicSpdySession { |
QuicSpdyStream*(SpdyPriority priority)); |
MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); |
MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); |
+ MOCK_METHOD1(CreateStream, std::unique_ptr<QuicStream>(QuicStreamId id)); |
MOCK_METHOD6( |
WritevData, |
QuicConsumedData( |
@@ -607,6 +613,24 @@ class MockQuicSpdySession : public QuicSpdySession { |
OnStreamFrameData, |
void(QuicStreamId stream_id, const char* data, size_t len, bool fin)); |
+ QuicSpdyStream* QuicSpdySessionMaybeCreateIncomingDynamicStream( |
+ QuicStreamId id) { |
+ return QuicSpdySession::MaybeCreateIncomingDynamicStream(id); |
+ } |
+ |
+ bool QuicSpdySessionShouldCreateIncomingDynamicStream2(QuicStreamId id) { |
+ return QuicSpdySession::ShouldCreateIncomingDynamicStream2(id); |
+ } |
+ |
+ QuicSpdyStream* QuicSpdySessionMaybeCreateOutgoingDynamicStream( |
+ SpdyPriority priority) { |
+ return QuicSpdySession::MaybeCreateOutgoingDynamicStream(priority); |
+ } |
+ |
+ bool QuicSpdySessionShouldCreateOutgoingDynamicStream2() { |
+ return QuicSpdySession::ShouldCreateOutgoingDynamicStream2(); |
+ } |
+ |
using QuicSession::ActivateStream; |
private: |
@@ -628,6 +652,7 @@ class TestQuicSpdyServerSession : public QuicServerSessionBase { |
MOCK_METHOD1(CreateIncomingDynamicStream, QuicSpdyStream*(QuicStreamId id)); |
MOCK_METHOD1(CreateOutgoingDynamicStream, |
QuicSpdyStream*(SpdyPriority priority)); |
+ MOCK_METHOD1(CreateStream, std::unique_ptr<QuicStream>(QuicStreamId id)); |
QuicCryptoServerStreamBase* CreateQuicCryptoServerStream( |
const QuicCryptoServerConfig* crypto_config, |
QuicCompressedCertsCache* compressed_certs_cache) override; |
@@ -690,6 +715,8 @@ class TestQuicSpdyClientSession : public QuicClientSessionBase { |
MOCK_METHOD1(ShouldCreateIncomingDynamicStream, bool(QuicStreamId id)); |
MOCK_METHOD0(ShouldCreateOutgoingDynamicStream, bool()); |
+ MOCK_METHOD1(CreateStream, std::unique_ptr<QuicStream>(QuicStreamId id)); |
+ |
QuicCryptoClientStream* GetMutableCryptoStream() override; |
const QuicCryptoClientStream* GetCryptoStream() const override; |