| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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/core/quic_stream.h" | 5 #include "net/quic/core/quic_stream.h" |
| 6 | 6 |
| 7 #include <memory> | 7 #include <memory> |
| 8 | 8 |
| 9 #include "base/memory/ptr_util.h" | 9 #include "base/memory/ptr_util.h" |
| 10 #include "net/quic/core/quic_connection.h" | 10 #include "net/quic/core/quic_connection.h" |
| 11 #include "net/quic/core/quic_flags.h" | 11 #include "net/quic/core/quic_flags.h" |
| 12 #include "net/quic/core/quic_utils.h" | 12 #include "net/quic/core/quic_utils.h" |
| 13 #include "net/quic/core/quic_write_blocked_list.h" | 13 #include "net/quic/core/quic_write_blocked_list.h" |
| 14 #include "net/quic/core/spdy_utils.h" | 14 #include "net/quic/core/spdy_utils.h" |
| 15 #include "net/quic/platform/api/quic_logging.h" |
| 15 #include "net/quic/test_tools/quic_config_peer.h" | 16 #include "net/quic/test_tools/quic_config_peer.h" |
| 16 #include "net/quic/test_tools/quic_connection_peer.h" | 17 #include "net/quic/test_tools/quic_connection_peer.h" |
| 17 #include "net/quic/test_tools/quic_flow_controller_peer.h" | 18 #include "net/quic/test_tools/quic_flow_controller_peer.h" |
| 18 #include "net/quic/test_tools/quic_session_peer.h" | 19 #include "net/quic/test_tools/quic_session_peer.h" |
| 19 #include "net/quic/test_tools/quic_stream_peer.h" | 20 #include "net/quic/test_tools/quic_stream_peer.h" |
| 20 #include "net/quic/test_tools/quic_test_utils.h" | 21 #include "net/quic/test_tools/quic_test_utils.h" |
| 21 #include "net/test/gtest_util.h" | 22 #include "net/test/gtest_util.h" |
| 22 #include "testing/gmock/include/gmock/gmock.h" | 23 #include "testing/gmock/include/gmock/gmock.h" |
| 23 #include "testing/gmock_mutant.h" | 24 #include "testing/gmock_mutant.h" |
| 24 | 25 |
| (...skipping 22 matching lines...) Expand all Loading... |
| 47 | 48 |
| 48 class TestStream : public QuicStream { | 49 class TestStream : public QuicStream { |
| 49 public: | 50 public: |
| 50 TestStream(QuicStreamId id, QuicSession* session, bool should_process_data) | 51 TestStream(QuicStreamId id, QuicSession* session, bool should_process_data) |
| 51 : QuicStream(id, session), should_process_data_(should_process_data) {} | 52 : QuicStream(id, session), should_process_data_(should_process_data) {} |
| 52 | 53 |
| 53 void OnDataAvailable() override {} | 54 void OnDataAvailable() override {} |
| 54 | 55 |
| 55 uint32_t ProcessRawData(const char* data, uint32_t data_len) { | 56 uint32_t ProcessRawData(const char* data, uint32_t data_len) { |
| 56 EXPECT_NE(0u, data_len); | 57 EXPECT_NE(0u, data_len); |
| 57 DVLOG(1) << "ProcessData data_len: " << data_len; | 58 QUIC_DVLOG(1) << "ProcessData data_len: " << data_len; |
| 58 data_ += string(data, data_len); | 59 data_ += string(data, data_len); |
| 59 return should_process_data_ ? data_len : 0; | 60 return should_process_data_ ? data_len : 0; |
| 60 } | 61 } |
| 61 | 62 |
| 62 using QuicStream::WriteOrBufferData; | 63 using QuicStream::WriteOrBufferData; |
| 63 using QuicStream::CloseWriteSide; | 64 using QuicStream::CloseWriteSide; |
| 64 using QuicStream::OnClose; | 65 using QuicStream::OnClose; |
| 65 | 66 |
| 66 private: | 67 private: |
| 67 bool should_process_data_; | 68 bool should_process_data_; |
| (...skipping 644 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 712 // Receive remaining data and FIN for the request. | 713 // Receive remaining data and FIN for the request. |
| 713 QuicStreamFrame frame2(stream_->id(), true, 0, StringPiece("End")); | 714 QuicStreamFrame frame2(stream_->id(), true, 0, StringPiece("End")); |
| 714 stream_->OnStreamFrame(frame2); | 715 stream_->OnStreamFrame(frame2); |
| 715 EXPECT_TRUE(stream_->fin_received()); | 716 EXPECT_TRUE(stream_->fin_received()); |
| 716 EXPECT_TRUE(stream_->HasFinalReceivedByteOffset()); | 717 EXPECT_TRUE(stream_->HasFinalReceivedByteOffset()); |
| 717 } | 718 } |
| 718 | 719 |
| 719 } // namespace | 720 } // namespace |
| 720 } // namespace test | 721 } // namespace test |
| 721 } // namespace net | 722 } // namespace net |
| OLD | NEW |