| 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 19 matching lines...) Expand all Loading... |
| 30 MOCK_METHOD3(OnDataFrameHeader, void(SpdyStreamId stream_id, | 30 MOCK_METHOD3(OnDataFrameHeader, void(SpdyStreamId stream_id, |
| 31 size_t length, | 31 size_t length, |
| 32 bool fin)); | 32 bool fin)); |
| 33 MOCK_METHOD4(OnStreamFrameData, void(SpdyStreamId stream_id, | 33 MOCK_METHOD4(OnStreamFrameData, void(SpdyStreamId stream_id, |
| 34 const char* data, | 34 const char* data, |
| 35 size_t len, | 35 size_t len, |
| 36 bool fin)); | 36 bool fin)); |
| 37 MOCK_METHOD3(OnControlFrameHeaderData, bool(SpdyStreamId stream_id, | 37 MOCK_METHOD3(OnControlFrameHeaderData, bool(SpdyStreamId stream_id, |
| 38 const char* header_data, | 38 const char* header_data, |
| 39 size_t len)); | 39 size_t len)); |
| 40 MOCK_METHOD6(OnSynStream, void(SpdyStreamId stream_id, | 40 MOCK_METHOD5(OnSynStream, void(SpdyStreamId stream_id, |
| 41 SpdyStreamId associated_stream_id, | 41 SpdyStreamId associated_stream_id, |
| 42 SpdyPriority priority, | 42 SpdyPriority priority, |
| 43 uint8 slot, | |
| 44 bool fin, | 43 bool fin, |
| 45 bool unidirectional)); | 44 bool unidirectional)); |
| 46 MOCK_METHOD2(OnSynReply, void(SpdyStreamId stream_id, bool fin)); | 45 MOCK_METHOD2(OnSynReply, void(SpdyStreamId stream_id, bool fin)); |
| 47 MOCK_METHOD2(OnRstStream, void(SpdyStreamId stream_id, | 46 MOCK_METHOD2(OnRstStream, void(SpdyStreamId stream_id, |
| 48 SpdyRstStreamStatus status)); | 47 SpdyRstStreamStatus status)); |
| 49 MOCK_METHOD1(OnSettings, void(bool clear_persisted)); | 48 MOCK_METHOD1(OnSettings, void(bool clear_persisted)); |
| 50 MOCK_METHOD3(OnSetting, void(SpdySettingsIds id, uint8 flags, uint32 value)); | 49 MOCK_METHOD3(OnSetting, void(SpdySettingsIds id, uint8 flags, uint32 value)); |
| 51 MOCK_METHOD1(OnPing, void(uint32 unique_id)); | 50 MOCK_METHOD1(OnPing, void(uint32 unique_id)); |
| 52 MOCK_METHOD2(OnGoAway, void(SpdyStreamId last_accepted_stream_id, | 51 MOCK_METHOD2(OnGoAway, void(SpdyStreamId last_accepted_stream_id, |
| 53 SpdyGoAwayStatus status)); | 52 SpdyGoAwayStatus status)); |
| (...skipping 63 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 117 // Write the headers and capture the outgoing data | 116 // Write the headers and capture the outgoing data |
| 118 EXPECT_CALL(session_, WritevData(kHeadersStreamId, _, _, _, false, NULL)) | 117 EXPECT_CALL(session_, WritevData(kHeadersStreamId, _, _, _, false, NULL)) |
| 119 .WillOnce(WithArgs<1, 2>( | 118 .WillOnce(WithArgs<1, 2>( |
| 120 Invoke(this, &QuicHeadersStreamTest::SaveIov))); | 119 Invoke(this, &QuicHeadersStreamTest::SaveIov))); |
| 121 headers_stream_->WriteHeaders(stream_id, headers_, fin); | 120 headers_stream_->WriteHeaders(stream_id, headers_, fin); |
| 122 | 121 |
| 123 // Parse the outgoing data and check that it matches was was written. | 122 // Parse the outgoing data and check that it matches was was written. |
| 124 if (type == SYN_STREAM) { | 123 if (type == SYN_STREAM) { |
| 125 EXPECT_CALL(visitor_, OnSynStream(stream_id, kNoAssociatedStream, 0, | 124 EXPECT_CALL(visitor_, OnSynStream(stream_id, kNoAssociatedStream, 0, |
| 126 // priority, | 125 // priority, |
| 127 _, fin, kNotUnidirectional)); | 126 fin, kNotUnidirectional)); |
| 128 } else { | 127 } else { |
| 129 EXPECT_CALL(visitor_, OnSynReply(stream_id, fin)); | 128 EXPECT_CALL(visitor_, OnSynReply(stream_id, fin)); |
| 130 } | 129 } |
| 131 EXPECT_CALL(visitor_, OnControlFrameHeaderData(stream_id, _, _)) | 130 EXPECT_CALL(visitor_, OnControlFrameHeaderData(stream_id, _, _)) |
| 132 .WillRepeatedly(WithArgs<1, 2>( | 131 .WillRepeatedly(WithArgs<1, 2>( |
| 133 Invoke(this, &QuicHeadersStreamTest::SaveHeaderData))); | 132 Invoke(this, &QuicHeadersStreamTest::SaveHeaderData))); |
| 134 if (fin) { | 133 if (fin) { |
| 135 EXPECT_CALL(visitor_, OnStreamFrameData(stream_id, NULL, 0, true)); | 134 EXPECT_CALL(visitor_, OnStreamFrameData(stream_id, NULL, 0, true)); |
| 136 } | 135 } |
| 137 framer_.ProcessInput(saved_data_.data(), saved_data_.length()); | 136 framer_.ProcessInput(saved_data_.data(), saved_data_.length()); |
| (...skipping 170 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 308 QUIC_INVALID_HEADERS_STREAM_DATA, | 307 QUIC_INVALID_HEADERS_STREAM_DATA, |
| 309 "SPDY WINDOW_UPDATE frame recevied.")) | 308 "SPDY WINDOW_UPDATE frame recevied.")) |
| 310 .WillOnce(InvokeWithoutArgs(this, | 309 .WillOnce(InvokeWithoutArgs(this, |
| 311 &QuicHeadersStreamTest::CloseConnection)); | 310 &QuicHeadersStreamTest::CloseConnection)); |
| 312 headers_stream_->ProcessRawData(frame->data(), frame->size()); | 311 headers_stream_->ProcessRawData(frame->data(), frame->size()); |
| 313 } | 312 } |
| 314 | 313 |
| 315 } // namespace | 314 } // namespace |
| 316 } // namespace test | 315 } // namespace test |
| 317 } // namespace net | 316 } // namespace net |
| OLD | NEW |