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 |