Index: net/tools/quic/quic_spdy_server_stream_test.cc |
diff --git a/net/tools/quic/quic_spdy_server_stream_test.cc b/net/tools/quic/quic_spdy_server_stream_test.cc |
index eaf6d905d1ff07e45d755ab84b3f860ab5c9be15..ad67f0fdc7aa7d53b63f5717809f9f5cd4b97f57 100644 |
--- a/net/tools/quic/quic_spdy_server_stream_test.cc |
+++ b/net/tools/quic/quic_spdy_server_stream_test.cc |
@@ -37,11 +37,25 @@ namespace net { |
namespace tools { |
namespace test { |
-class QuicSpdyServerStreamPeer { |
+class QuicSpdyServerStreamPeer : public QuicSpdyServerStream { |
public: |
- static BalsaHeaders* GetMutableHeaders( |
- QuicSpdyServerStream* stream) { |
- return &(stream->headers_); |
+ QuicSpdyServerStreamPeer(QuicStreamId stream_id, QuicSession* session) |
+ : QuicSpdyServerStream(stream_id, session) { |
+ } |
+ |
+ using QuicSpdyServerStream::SendResponse; |
+ using QuicSpdyServerStream::SendErrorResponse; |
+ |
+ const string& body() { |
+ return body_; |
+ } |
+ |
+ const BalsaHeaders& headers() { |
+ return headers_; |
+ } |
+ |
+ BalsaHeaders* mutable_headers() { |
+ return &headers_; |
} |
}; |
@@ -58,7 +72,7 @@ class QuicSpdyServerStreamTest : public ::testing::Test { |
request_headers.ReplaceOrAppendHeader("content-length", "11"); |
headers_string_ = SpdyUtils::SerializeRequestHeaders(request_headers); |
- stream_.reset(new QuicSpdyServerStream(3, &session_)); |
+ stream_.reset(new QuicSpdyServerStreamPeer(3, &session_)); |
} |
QuicConsumedData ValidateHeaders(const struct iovec* iov) { |
@@ -119,7 +133,7 @@ class QuicSpdyServerStreamTest : public ::testing::Test { |
BalsaHeaders response_headers_; |
EpollServer eps_; |
StrictMock<MockSession> session_; |
- scoped_ptr<QuicSpdyServerStream> stream_; |
+ scoped_ptr<QuicSpdyServerStreamPeer> stream_; |
string headers_string_; |
string body_; |
}; |
@@ -184,8 +198,7 @@ TEST_F(QuicSpdyServerStreamTest, TestFramingExtraData) { |
} |
TEST_F(QuicSpdyServerStreamTest, TestSendResponse) { |
- BalsaHeaders* request_headers = |
- QuicSpdyServerStreamPeer::GetMutableHeaders(stream_.get()); |
+ BalsaHeaders* request_headers = stream_->mutable_headers(); |
request_headers->SetRequestFirstlineFromStringPieces( |
"GET", |
"https://www.google.com/foo", |