Index: net/quic/chromium/quic_chromium_client_stream_test.cc |
diff --git a/net/quic/chromium/quic_chromium_client_stream_test.cc b/net/quic/chromium/quic_chromium_client_stream_test.cc |
index 89cba5f21028aaecc2a20c0cf65c148379c2ae16..764c2f408f1d9732f31852936fa9a6307235cfd4 100644 |
--- a/net/quic/chromium/quic_chromium_client_stream_test.cc |
+++ b/net/quic/chromium/quic_chromium_client_stream_test.cc |
@@ -16,10 +16,12 @@ |
#include "net/quic/core/quic_client_session_base.h" |
#include "net/quic/core/quic_utils.h" |
#include "net/quic/core/spdy_utils.h" |
+#include "net/quic/platform/api/quic_ptr_util.h" |
#include "net/quic/test_tools/crypto_test_utils.h" |
#include "net/quic/test_tools/quic_spdy_session_peer.h" |
#include "net/quic/test_tools/quic_test_utils.h" |
#include "net/test/gtest_util.h" |
+#include "net/tools/quic/quic_spdy_client_stream.h" |
#include "testing/gmock/include/gmock/gmock.h" |
#include "testing/gmock_mutant.h" |
@@ -61,6 +63,14 @@ class MockDelegate : public QuicChromiumClientStream::Delegate { |
DISALLOW_COPY_AND_ASSIGN(MockDelegate); |
}; |
+class TestQuicClientSessionBaseStream : public QuicSpdyStream { |
+ public: |
+ TestQuicClientSessionBaseStream(QuicStreamId id, QuicSpdySession* session) |
+ : QuicSpdyStream(id, session) {} |
+ |
+ void OnDataAvailable() override {} |
+}; |
+ |
class MockQuicClientSessionBase : public QuicClientSessionBase { |
public: |
explicit MockQuicClientSessionBase(QuicConnection* connection, |
@@ -130,6 +140,10 @@ class MockQuicClientSessionBase : public QuicClientSessionBase { |
ack_listener)); |
MOCK_METHOD1(OnHeadersHeadOfLineBlocking, void(QuicTime::Delta delta)); |
+ std::unique_ptr<QuicStream> CreateStream(QuicStreamId id) { |
+ return QuicMakeUnique<TestQuicClientSessionBaseStream>(id, this); |
+ } |
+ |
using QuicSession::ActivateStream; |
// Returns a QuicConsumedData that indicates all of |data| (and |fin| if set) |