OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 #include "net/quic/quic_headers_stream.h" | 5 #include "net/quic/quic_headers_stream.h" |
6 | 6 |
7 #include "net/quic/quic_utils.h" | 7 #include "net/quic/quic_utils.h" |
8 #include "net/quic/spdy_utils.h" | 8 #include "net/quic/spdy_utils.h" |
9 #include "net/quic/test_tools/quic_connection_peer.h" | 9 #include "net/quic/test_tools/quic_connection_peer.h" |
10 #include "net/quic/test_tools/quic_session_peer.h" | 10 #include "net/quic/test_tools/quic_session_peer.h" |
(...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
83 : connection_(new StrictMock<MockConnection>(is_server(), GetVersions())), | 83 : connection_(new StrictMock<MockConnection>(is_server(), GetVersions())), |
84 session_(connection_), | 84 session_(connection_), |
85 headers_stream_(QuicSessionPeer::GetHeadersStream(&session_)), | 85 headers_stream_(QuicSessionPeer::GetHeadersStream(&session_)), |
86 body_("hello world"), | 86 body_("hello world"), |
87 framer_(SPDY3) { | 87 framer_(SPDY3) { |
88 headers_[":version"] = "HTTP/1.1"; | 88 headers_[":version"] = "HTTP/1.1"; |
89 headers_[":status"] = "200 Ok"; | 89 headers_[":status"] = "200 Ok"; |
90 headers_["content-length"] = "11"; | 90 headers_["content-length"] = "11"; |
91 framer_.set_visitor(&visitor_); | 91 framer_.set_visitor(&visitor_); |
92 EXPECT_EQ(QuicVersionMax(), session_.connection()->version()); | 92 EXPECT_EQ(QuicVersionMax(), session_.connection()->version()); |
93 EXPECT_TRUE(headers_stream_ != NULL); | 93 EXPECT_TRUE(headers_stream_ != nullptr); |
94 } | 94 } |
95 | 95 |
96 QuicConsumedData SaveIov(const IOVector& data) { | 96 QuicConsumedData SaveIov(const IOVector& data) { |
97 const iovec* iov = data.iovec(); | 97 const iovec* iov = data.iovec(); |
98 int count = data.Capacity(); | 98 int count = data.Capacity(); |
99 for (int i = 0 ; i < count; ++i) { | 99 for (int i = 0 ; i < count; ++i) { |
100 saved_data_.append(static_cast<char*>(iov[i].iov_base), iov[i].iov_len); | 100 saved_data_.append(static_cast<char*>(iov[i].iov_base), iov[i].iov_len); |
101 } | 101 } |
102 return QuicConsumedData(saved_data_.length(), false); | 102 return QuicConsumedData(saved_data_.length(), false); |
103 } | 103 } |
(...skipping 16 matching lines...) Expand all Loading... |
120 void WriteHeadersAndExpectSynReply(QuicStreamId stream_id, | 120 void WriteHeadersAndExpectSynReply(QuicStreamId stream_id, |
121 bool fin) { | 121 bool fin) { |
122 WriteHeadersAndCheckData(stream_id, fin, 0, SYN_REPLY); | 122 WriteHeadersAndCheckData(stream_id, fin, 0, SYN_REPLY); |
123 } | 123 } |
124 | 124 |
125 void WriteHeadersAndCheckData(QuicStreamId stream_id, | 125 void WriteHeadersAndCheckData(QuicStreamId stream_id, |
126 bool fin, | 126 bool fin, |
127 QuicPriority priority, | 127 QuicPriority priority, |
128 SpdyFrameType type) { | 128 SpdyFrameType type) { |
129 // Write the headers and capture the outgoing data | 129 // Write the headers and capture the outgoing data |
130 EXPECT_CALL(session_, WritevData(kHeadersStreamId, _, _, false, _, NULL)) | 130 EXPECT_CALL(session_, WritevData(kHeadersStreamId, _, _, false, _, nullptr)) |
131 .WillOnce(WithArgs<1>(Invoke(this, &QuicHeadersStreamTest::SaveIov))); | 131 .WillOnce(WithArgs<1>(Invoke(this, &QuicHeadersStreamTest::SaveIov))); |
132 headers_stream_->WriteHeaders(stream_id, headers_, fin, NULL); | 132 headers_stream_->WriteHeaders(stream_id, headers_, fin, nullptr); |
133 | 133 |
134 // Parse the outgoing data and check that it matches was was written. | 134 // Parse the outgoing data and check that it matches was was written. |
135 if (type == SYN_STREAM) { | 135 if (type == SYN_STREAM) { |
136 EXPECT_CALL(visitor_, OnSynStream(stream_id, kNoAssociatedStream, 0, | 136 EXPECT_CALL(visitor_, OnSynStream(stream_id, kNoAssociatedStream, 0, |
137 // priority, | 137 // priority, |
138 fin, kNotUnidirectional)); | 138 fin, kNotUnidirectional)); |
139 } else { | 139 } else { |
140 EXPECT_CALL(visitor_, OnSynReply(stream_id, fin)); | 140 EXPECT_CALL(visitor_, OnSynReply(stream_id, fin)); |
141 } | 141 } |
142 EXPECT_CALL(visitor_, OnControlFrameHeaderData(stream_id, _, _)) | 142 EXPECT_CALL(visitor_, OnControlFrameHeaderData(stream_id, _, _)) |
143 .WillRepeatedly(WithArgs<1, 2>( | 143 .WillRepeatedly(WithArgs<1, 2>( |
144 Invoke(this, &QuicHeadersStreamTest::SaveHeaderData))); | 144 Invoke(this, &QuicHeadersStreamTest::SaveHeaderData))); |
145 if (fin) { | 145 if (fin) { |
146 EXPECT_CALL(visitor_, OnStreamFrameData(stream_id, NULL, 0, true)); | 146 EXPECT_CALL(visitor_, OnStreamFrameData(stream_id, nullptr, 0, true)); |
147 } | 147 } |
148 framer_.ProcessInput(saved_data_.data(), saved_data_.length()); | 148 framer_.ProcessInput(saved_data_.data(), saved_data_.length()); |
149 EXPECT_FALSE(framer_.HasError()) << framer_.error_code(); | 149 EXPECT_FALSE(framer_.HasError()) << framer_.error_code(); |
150 | 150 |
151 CheckHeaders(); | 151 CheckHeaders(); |
152 saved_data_.clear(); | 152 saved_data_.clear(); |
153 } | 153 } |
154 | 154 |
155 void CheckHeaders() { | 155 void CheckHeaders() { |
156 SpdyHeaderBlock headers; | 156 SpdyHeaderBlock headers; |
(...skipping 174 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
331 } else { | 331 } else { |
332 EXPECT_TRUE(headers_stream_->flow_controller()->IsEnabled()); | 332 EXPECT_TRUE(headers_stream_->flow_controller()->IsEnabled()); |
333 } | 333 } |
334 EXPECT_FALSE(ReliableQuicStreamPeer::StreamContributesToConnectionFlowControl( | 334 EXPECT_FALSE(ReliableQuicStreamPeer::StreamContributesToConnectionFlowControl( |
335 headers_stream_)); | 335 headers_stream_)); |
336 } | 336 } |
337 | 337 |
338 } // namespace | 338 } // namespace |
339 } // namespace test | 339 } // namespace test |
340 } // namespace net | 340 } // namespace net |
OLD | NEW |