| 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/tools/quic/quic_spdy_client_stream.h" | 5 #include "net/tools/quic/quic_spdy_client_stream.h" |
| 6 | 6 |
| 7 #include <memory> | 7 #include <memory> |
| 8 | 8 |
| 9 #include "base/macros.h" | 9 #include "base/macros.h" |
| 10 #include "base/strings/string_number_conversions.h" | 10 #include "base/strings/string_number_conversions.h" |
| (...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 56 QuicCryptoClientConfig crypto_config_; | 56 QuicCryptoClientConfig crypto_config_; |
| 57 | 57 |
| 58 DISALLOW_COPY_AND_ASSIGN(MockQuicClientSession); | 58 DISALLOW_COPY_AND_ASSIGN(MockQuicClientSession); |
| 59 }; | 59 }; |
| 60 | 60 |
| 61 class QuicSpdyClientStreamTest : public ::testing::Test { | 61 class QuicSpdyClientStreamTest : public ::testing::Test { |
| 62 public: | 62 public: |
| 63 class StreamVisitor; | 63 class StreamVisitor; |
| 64 | 64 |
| 65 QuicSpdyClientStreamTest() | 65 QuicSpdyClientStreamTest() |
| 66 : connection_( | 66 : connection_(new StrictMock<MockConnection>(&helper_, |
| 67 new StrictMock<MockConnection>(&helper_, Perspective::IS_CLIENT)), | 67 &alarm_factory_, |
| 68 Perspective::IS_CLIENT)), |
| 68 session_(connection_, &push_promise_index_), | 69 session_(connection_, &push_promise_index_), |
| 69 body_("hello world") { | 70 body_("hello world") { |
| 70 session_.Initialize(); | 71 session_.Initialize(); |
| 71 | 72 |
| 72 headers_.SetResponseFirstlineFromStringPieces("HTTP/1.1", "200", "Ok"); | 73 headers_.SetResponseFirstlineFromStringPieces("HTTP/1.1", "200", "Ok"); |
| 73 headers_.ReplaceOrAppendHeader("content-length", "11"); | 74 headers_.ReplaceOrAppendHeader("content-length", "11"); |
| 74 | 75 |
| 75 headers_string_ = net::SpdyBalsaUtils::SerializeResponseHeaders(headers_); | 76 headers_string_ = net::SpdyBalsaUtils::SerializeResponseHeaders(headers_); |
| 76 | 77 |
| 77 stream_.reset(new QuicSpdyClientStream(kClientDataStreamId1, &session_)); | 78 stream_.reset(new QuicSpdyClientStream(kClientDataStreamId1, &session_)); |
| 78 stream_visitor_.reset(new StreamVisitor()); | 79 stream_visitor_.reset(new StreamVisitor()); |
| 79 stream_->set_visitor(stream_visitor_.get()); | 80 stream_->set_visitor(stream_visitor_.get()); |
| 80 } | 81 } |
| 81 | 82 |
| 82 class StreamVisitor : public QuicSpdyClientStream::Visitor { | 83 class StreamVisitor : public QuicSpdyClientStream::Visitor { |
| 83 void OnClose(QuicSpdyStream* stream) override { | 84 void OnClose(QuicSpdyStream* stream) override { |
| 84 DVLOG(1) << "stream " << stream->id(); | 85 DVLOG(1) << "stream " << stream->id(); |
| 85 } | 86 } |
| 86 }; | 87 }; |
| 87 | 88 |
| 88 MockConnectionHelper helper_; | 89 MockConnectionHelper helper_; |
| 90 MockAlarmFactory alarm_factory_; |
| 89 StrictMock<MockConnection>* connection_; | 91 StrictMock<MockConnection>* connection_; |
| 90 QuicClientPushPromiseIndex push_promise_index_; | 92 QuicClientPushPromiseIndex push_promise_index_; |
| 91 | 93 |
| 92 MockQuicClientSession session_; | 94 MockQuicClientSession session_; |
| 93 std::unique_ptr<QuicSpdyClientStream> stream_; | 95 std::unique_ptr<QuicSpdyClientStream> stream_; |
| 94 std::unique_ptr<StreamVisitor> stream_visitor_; | 96 std::unique_ptr<StreamVisitor> stream_visitor_; |
| 95 BalsaHeaders headers_; | 97 BalsaHeaders headers_; |
| 96 string headers_string_; | 98 string headers_string_; |
| 97 string body_; | 99 string body_; |
| 98 }; | 100 }; |
| (...skipping 75 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 174 // Now send the body, which should close the stream as the FIN has been | 176 // Now send the body, which should close the stream as the FIN has been |
| 175 // received, as well as all data. | 177 // received, as well as all data. |
| 176 EXPECT_CALL(session_, CloseStream(stream_->id())); | 178 EXPECT_CALL(session_, CloseStream(stream_->id())); |
| 177 stream_->OnStreamFrame( | 179 stream_->OnStreamFrame( |
| 178 QuicStreamFrame(stream_->id(), /*fin=*/false, /*offset=*/0, body_)); | 180 QuicStreamFrame(stream_->id(), /*fin=*/false, /*offset=*/0, body_)); |
| 179 } | 181 } |
| 180 | 182 |
| 181 } // namespace | 183 } // namespace |
| 182 } // namespace test | 184 } // namespace test |
| 183 } // namespace net | 185 } // namespace net |
| OLD | NEW |