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/quic_reliable_client_stream.h" | 5 #include "net/quic/quic_chromium_client_stream.h" |
6 | 6 |
7 #include "base/macros.h" | 7 #include "base/macros.h" |
8 #include "net/base/net_errors.h" | 8 #include "net/base/net_errors.h" |
9 #include "net/base/test_completion_callback.h" | 9 #include "net/base/test_completion_callback.h" |
10 #include "net/quic/quic_chromium_client_session.h" | 10 #include "net/quic/quic_chromium_client_session.h" |
11 #include "net/quic/quic_utils.h" | 11 #include "net/quic/quic_utils.h" |
12 #include "net/quic/spdy_utils.h" | 12 #include "net/quic/spdy_utils.h" |
13 #include "net/quic/test_tools/crypto_test_utils.h" | 13 #include "net/quic/test_tools/crypto_test_utils.h" |
14 #include "net/quic/test_tools/quic_test_utils.h" | 14 #include "net/quic/test_tools/quic_test_utils.h" |
15 #include "testing/gmock/include/gmock/gmock.h" | 15 #include "testing/gmock/include/gmock/gmock.h" |
16 #include "testing/gmock_mutant.h" | 16 #include "testing/gmock_mutant.h" |
17 | 17 |
18 using testing::AnyNumber; | 18 using testing::AnyNumber; |
19 using testing::CreateFunctor; | 19 using testing::CreateFunctor; |
20 using testing::Invoke; | 20 using testing::Invoke; |
21 using testing::Return; | 21 using testing::Return; |
22 using testing::StrEq; | 22 using testing::StrEq; |
23 using testing::_; | 23 using testing::_; |
24 | 24 |
25 namespace net { | 25 namespace net { |
26 namespace test { | 26 namespace test { |
27 namespace { | 27 namespace { |
28 | 28 |
29 const QuicStreamId kTestStreamId = 5u; | 29 const QuicStreamId kTestStreamId = 5u; |
30 | 30 |
31 class MockDelegate : public QuicReliableClientStream::Delegate { | 31 class MockDelegate : public QuicChromiumClientStream::Delegate { |
32 public: | 32 public: |
33 MockDelegate() {} | 33 MockDelegate() {} |
34 | 34 |
35 MOCK_METHOD0(OnSendData, int()); | 35 MOCK_METHOD0(OnSendData, int()); |
36 MOCK_METHOD2(OnSendDataComplete, int(int, bool*)); | 36 MOCK_METHOD2(OnSendDataComplete, int(int, bool*)); |
37 MOCK_METHOD2(OnHeadersAvailable, void(const SpdyHeaderBlock&, size_t)); | 37 MOCK_METHOD2(OnHeadersAvailable, void(const SpdyHeaderBlock&, size_t)); |
38 MOCK_METHOD2(OnDataReceived, int(const char*, int)); | 38 MOCK_METHOD2(OnDataReceived, int(const char*, int)); |
39 MOCK_METHOD0(OnDataAvailable, void()); | 39 MOCK_METHOD0(OnDataAvailable, void()); |
40 MOCK_METHOD1(OnClose, void(QuicErrorCode)); | 40 MOCK_METHOD1(OnClose, void(QuicErrorCode)); |
41 MOCK_METHOD1(OnError, void(int)); | 41 MOCK_METHOD1(OnError, void(int)); |
42 MOCK_METHOD0(HasSendHeadersComplete, bool()); | 42 MOCK_METHOD0(HasSendHeadersComplete, bool()); |
43 | 43 |
44 private: | 44 private: |
45 DISALLOW_COPY_AND_ASSIGN(MockDelegate); | 45 DISALLOW_COPY_AND_ASSIGN(MockDelegate); |
46 }; | 46 }; |
47 | 47 |
48 class QuicReliableClientStreamTest | 48 class QuicChromiumClientStreamTest |
49 : public ::testing::TestWithParam<QuicVersion> { | 49 : public ::testing::TestWithParam<QuicVersion> { |
50 public: | 50 public: |
51 QuicReliableClientStreamTest() | 51 QuicChromiumClientStreamTest() |
52 : crypto_config_(CryptoTestUtils::ProofVerifierForTesting()), | 52 : crypto_config_(CryptoTestUtils::ProofVerifierForTesting()), |
53 session_(new MockConnection(&helper_, | 53 session_(new MockConnection(&helper_, |
54 Perspective::IS_CLIENT, | 54 Perspective::IS_CLIENT, |
55 SupportedVersions(GetParam()))) { | 55 SupportedVersions(GetParam()))) { |
56 stream_ = | 56 stream_ = |
57 new QuicReliableClientStream(kTestStreamId, &session_, BoundNetLog()); | 57 new QuicChromiumClientStream(kTestStreamId, &session_, BoundNetLog()); |
58 session_.ActivateStream(stream_); | 58 session_.ActivateStream(stream_); |
59 stream_->SetDelegate(&delegate_); | 59 stream_->SetDelegate(&delegate_); |
60 } | 60 } |
61 | 61 |
62 void InitializeHeaders() { | 62 void InitializeHeaders() { |
63 headers_[":host"] = "www.google.com"; | 63 headers_[":host"] = "www.google.com"; |
64 headers_[":path"] = "/index.hml"; | 64 headers_[":path"] = "/index.hml"; |
65 headers_[":scheme"] = "https"; | 65 headers_[":scheme"] = "https"; |
66 headers_["cookie"] = | 66 headers_["cookie"] = |
67 "__utma=208381060.1228362404.1372200928.1372200928.1372200928.1; " | 67 "__utma=208381060.1228362404.1372200928.1372200928.1372200928.1; " |
(...skipping 26 matching lines...) Expand all Loading... |
94 EXPECT_EQ(static_cast<int>(expected_data.length()), | 94 EXPECT_EQ(static_cast<int>(expected_data.length()), |
95 stream_->Read(buffer.get(), expected_data.length() + 1)); | 95 stream_->Read(buffer.get(), expected_data.length() + 1)); |
96 EXPECT_EQ(expected_data, | 96 EXPECT_EQ(expected_data, |
97 StringPiece(buffer->data(), expected_data.length())); | 97 StringPiece(buffer->data(), expected_data.length())); |
98 } | 98 } |
99 | 99 |
100 QuicCryptoClientConfig crypto_config_; | 100 QuicCryptoClientConfig crypto_config_; |
101 testing::StrictMock<MockDelegate> delegate_; | 101 testing::StrictMock<MockDelegate> delegate_; |
102 MockConnectionHelper helper_; | 102 MockConnectionHelper helper_; |
103 MockQuicSpdySession session_; | 103 MockQuicSpdySession session_; |
104 QuicReliableClientStream* stream_; | 104 QuicChromiumClientStream* stream_; |
105 SpdyHeaderBlock headers_; | 105 SpdyHeaderBlock headers_; |
106 }; | 106 }; |
107 | 107 |
108 INSTANTIATE_TEST_CASE_P(Version, | 108 INSTANTIATE_TEST_CASE_P(Version, |
109 QuicReliableClientStreamTest, | 109 QuicChromiumClientStreamTest, |
110 ::testing::ValuesIn(QuicSupportedVersions())); | 110 ::testing::ValuesIn(QuicSupportedVersions())); |
111 | 111 |
112 TEST_P(QuicReliableClientStreamTest, OnFinRead) { | 112 TEST_P(QuicChromiumClientStreamTest, OnFinRead) { |
113 InitializeHeaders(); | 113 InitializeHeaders(); |
114 std::string uncompressed_headers = | 114 std::string uncompressed_headers = |
115 SpdyUtils::SerializeUncompressedHeaders(headers_); | 115 SpdyUtils::SerializeUncompressedHeaders(headers_); |
116 QuicStreamOffset offset = 0; | 116 QuicStreamOffset offset = 0; |
117 stream_->OnStreamHeaders(uncompressed_headers); | 117 stream_->OnStreamHeaders(uncompressed_headers); |
118 stream_->OnStreamHeadersComplete(false, uncompressed_headers.length()); | 118 stream_->OnStreamHeadersComplete(false, uncompressed_headers.length()); |
119 | 119 |
120 EXPECT_CALL(delegate_, | 120 EXPECT_CALL(delegate_, |
121 OnHeadersAvailable(headers_, uncompressed_headers.length())); | 121 OnHeadersAvailable(headers_, uncompressed_headers.length())); |
122 base::MessageLoop::current()->RunUntilIdle(); | 122 base::MessageLoop::current()->RunUntilIdle(); |
123 EXPECT_TRUE(stream_->decompressed_headers().empty()); | 123 EXPECT_TRUE(stream_->decompressed_headers().empty()); |
124 | 124 |
125 QuicStreamFrame frame2(kTestStreamId, true, offset, StringPiece()); | 125 QuicStreamFrame frame2(kTestStreamId, true, offset, StringPiece()); |
126 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 126 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
127 stream_->OnStreamFrame(frame2); | 127 stream_->OnStreamFrame(frame2); |
128 } | 128 } |
129 | 129 |
130 TEST_P(QuicReliableClientStreamTest, OnDataAvailableBeforeHeaders) { | 130 TEST_P(QuicChromiumClientStreamTest, OnDataAvailableBeforeHeaders) { |
131 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 131 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
132 | 132 |
133 EXPECT_CALL(delegate_, OnDataAvailable()).Times(0); | 133 EXPECT_CALL(delegate_, OnDataAvailable()).Times(0); |
134 stream_->OnDataAvailable(); | 134 stream_->OnDataAvailable(); |
135 } | 135 } |
136 | 136 |
137 TEST_P(QuicReliableClientStreamTest, OnDataAvailable) { | 137 TEST_P(QuicChromiumClientStreamTest, OnDataAvailable) { |
138 InitializeHeaders(); | 138 InitializeHeaders(); |
139 std::string uncompressed_headers = | 139 std::string uncompressed_headers = |
140 SpdyUtils::SerializeUncompressedHeaders(headers_); | 140 SpdyUtils::SerializeUncompressedHeaders(headers_); |
141 stream_->OnStreamHeaders(uncompressed_headers); | 141 stream_->OnStreamHeaders(uncompressed_headers); |
142 stream_->OnStreamHeadersComplete(false, uncompressed_headers.length()); | 142 stream_->OnStreamHeadersComplete(false, uncompressed_headers.length()); |
143 | 143 |
144 EXPECT_CALL(delegate_, | 144 EXPECT_CALL(delegate_, |
145 OnHeadersAvailable(headers_, uncompressed_headers.length())); | 145 OnHeadersAvailable(headers_, uncompressed_headers.length())); |
146 base::MessageLoop::current()->RunUntilIdle(); | 146 base::MessageLoop::current()->RunUntilIdle(); |
147 EXPECT_TRUE(stream_->decompressed_headers().empty()); | 147 EXPECT_TRUE(stream_->decompressed_headers().empty()); |
148 | 148 |
149 const char data[] = "hello world!"; | 149 const char data[] = "hello world!"; |
150 stream_->OnStreamFrame(QuicStreamFrame(kTestStreamId, /*fin=*/false, | 150 stream_->OnStreamFrame(QuicStreamFrame(kTestStreamId, /*fin=*/false, |
151 /*offset=*/0, data)); | 151 /*offset=*/0, data)); |
152 | 152 |
153 EXPECT_CALL(delegate_, OnDataAvailable()) | 153 EXPECT_CALL(delegate_, OnDataAvailable()) |
154 .WillOnce(testing::Invoke( | 154 .WillOnce(testing::Invoke( |
155 CreateFunctor(this, &QuicReliableClientStreamTest::ReadData, | 155 CreateFunctor(this, &QuicChromiumClientStreamTest::ReadData, |
156 StringPiece(data, arraysize(data) - 1)))); | 156 StringPiece(data, arraysize(data) - 1)))); |
157 base::MessageLoop::current()->RunUntilIdle(); | 157 base::MessageLoop::current()->RunUntilIdle(); |
158 | 158 |
159 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 159 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
160 } | 160 } |
161 | 161 |
162 TEST_P(QuicReliableClientStreamTest, ProcessHeadersWithError) { | 162 TEST_P(QuicChromiumClientStreamTest, ProcessHeadersWithError) { |
163 std::string bad_headers = "..."; | 163 std::string bad_headers = "..."; |
164 stream_->OnStreamHeaders(StringPiece(bad_headers)); | 164 stream_->OnStreamHeaders(StringPiece(bad_headers)); |
165 stream_->OnStreamHeadersComplete(false, bad_headers.length()); | 165 stream_->OnStreamHeadersComplete(false, bad_headers.length()); |
166 | 166 |
167 EXPECT_CALL(session_, | 167 EXPECT_CALL(session_, |
168 SendRstStream(kTestStreamId, QUIC_BAD_APPLICATION_PAYLOAD, 0)); | 168 SendRstStream(kTestStreamId, QUIC_BAD_APPLICATION_PAYLOAD, 0)); |
169 base::MessageLoop::current()->RunUntilIdle(); | 169 base::MessageLoop::current()->RunUntilIdle(); |
170 | 170 |
171 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 171 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
172 } | 172 } |
173 | 173 |
174 TEST_P(QuicReliableClientStreamTest, OnDataAvailableWithError) { | 174 TEST_P(QuicChromiumClientStreamTest, OnDataAvailableWithError) { |
175 InitializeHeaders(); | 175 InitializeHeaders(); |
176 std::string uncompressed_headers = | 176 std::string uncompressed_headers = |
177 SpdyUtils::SerializeUncompressedHeaders(headers_); | 177 SpdyUtils::SerializeUncompressedHeaders(headers_); |
178 stream_->OnStreamHeaders(uncompressed_headers); | 178 stream_->OnStreamHeaders(uncompressed_headers); |
179 stream_->OnStreamHeadersComplete(false, uncompressed_headers.length()); | 179 stream_->OnStreamHeadersComplete(false, uncompressed_headers.length()); |
180 | 180 |
181 EXPECT_CALL(delegate_, | 181 EXPECT_CALL(delegate_, |
182 OnHeadersAvailable(headers_, uncompressed_headers.length())); | 182 OnHeadersAvailable(headers_, uncompressed_headers.length())); |
183 base::MessageLoop::current()->RunUntilIdle(); | 183 base::MessageLoop::current()->RunUntilIdle(); |
184 EXPECT_TRUE(stream_->decompressed_headers().empty()); | 184 EXPECT_TRUE(stream_->decompressed_headers().empty()); |
185 | 185 |
186 const char data[] = "hello world!"; | 186 const char data[] = "hello world!"; |
187 stream_->OnStreamFrame(QuicStreamFrame(kTestStreamId, /*fin=*/false, | 187 stream_->OnStreamFrame(QuicStreamFrame(kTestStreamId, /*fin=*/false, |
188 /*offset=*/0, data)); | 188 /*offset=*/0, data)); |
189 EXPECT_CALL(delegate_, OnDataAvailable()) | 189 EXPECT_CALL(delegate_, OnDataAvailable()) |
190 .WillOnce(testing::Invoke(CreateFunctor( | 190 .WillOnce(testing::Invoke(CreateFunctor( |
191 stream_, &QuicReliableClientStream::Reset, QUIC_STREAM_CANCELLED))); | 191 stream_, &QuicChromiumClientStream::Reset, QUIC_STREAM_CANCELLED))); |
192 base::MessageLoop::current()->RunUntilIdle(); | 192 base::MessageLoop::current()->RunUntilIdle(); |
193 | 193 |
194 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 194 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
195 } | 195 } |
196 | 196 |
197 TEST_P(QuicReliableClientStreamTest, OnError) { | 197 TEST_P(QuicChromiumClientStreamTest, OnError) { |
198 EXPECT_CALL(delegate_, OnError(ERR_INTERNET_DISCONNECTED)); | 198 EXPECT_CALL(delegate_, OnError(ERR_INTERNET_DISCONNECTED)); |
199 | 199 |
200 stream_->OnError(ERR_INTERNET_DISCONNECTED); | 200 stream_->OnError(ERR_INTERNET_DISCONNECTED); |
201 EXPECT_FALSE(stream_->GetDelegate()); | 201 EXPECT_FALSE(stream_->GetDelegate()); |
202 } | 202 } |
203 | 203 |
204 TEST_P(QuicReliableClientStreamTest, WriteStreamData) { | 204 TEST_P(QuicChromiumClientStreamTest, WriteStreamData) { |
205 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 205 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
206 | 206 |
207 const char kData1[] = "hello world"; | 207 const char kData1[] = "hello world"; |
208 const size_t kDataLen = arraysize(kData1); | 208 const size_t kDataLen = arraysize(kData1); |
209 | 209 |
210 // All data written. | 210 // All data written. |
211 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)) | 211 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)) |
212 .WillOnce(Return(QuicConsumedData(kDataLen, true))); | 212 .WillOnce(Return(QuicConsumedData(kDataLen, true))); |
213 TestCompletionCallback callback; | 213 TestCompletionCallback callback; |
214 EXPECT_EQ(OK, stream_->WriteStreamData(base::StringPiece(kData1, kDataLen), | 214 EXPECT_EQ(OK, stream_->WriteStreamData(base::StringPiece(kData1, kDataLen), |
215 true, callback.callback())); | 215 true, callback.callback())); |
216 } | 216 } |
217 | 217 |
218 TEST_P(QuicReliableClientStreamTest, WriteStreamDataAsync) { | 218 TEST_P(QuicChromiumClientStreamTest, WriteStreamDataAsync) { |
219 EXPECT_CALL(delegate_, HasSendHeadersComplete()).Times(AnyNumber()); | 219 EXPECT_CALL(delegate_, HasSendHeadersComplete()).Times(AnyNumber()); |
220 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 220 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
221 | 221 |
222 const char kData1[] = "hello world"; | 222 const char kData1[] = "hello world"; |
223 const size_t kDataLen = arraysize(kData1); | 223 const size_t kDataLen = arraysize(kData1); |
224 | 224 |
225 // No data written. | 225 // No data written. |
226 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)) | 226 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)) |
227 .WillOnce(Return(QuicConsumedData(0, false))); | 227 .WillOnce(Return(QuicConsumedData(0, false))); |
228 TestCompletionCallback callback; | 228 TestCompletionCallback callback; |
229 EXPECT_EQ(ERR_IO_PENDING, | 229 EXPECT_EQ(ERR_IO_PENDING, |
230 stream_->WriteStreamData(base::StringPiece(kData1, kDataLen), true, | 230 stream_->WriteStreamData(base::StringPiece(kData1, kDataLen), true, |
231 callback.callback())); | 231 callback.callback())); |
232 ASSERT_FALSE(callback.have_result()); | 232 ASSERT_FALSE(callback.have_result()); |
233 | 233 |
234 // All data written. | 234 // All data written. |
235 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)) | 235 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)) |
236 .WillOnce(Return(QuicConsumedData(kDataLen, true))); | 236 .WillOnce(Return(QuicConsumedData(kDataLen, true))); |
237 stream_->OnCanWrite(); | 237 stream_->OnCanWrite(); |
238 ASSERT_TRUE(callback.have_result()); | 238 ASSERT_TRUE(callback.have_result()); |
239 EXPECT_EQ(OK, callback.WaitForResult()); | 239 EXPECT_EQ(OK, callback.WaitForResult()); |
240 } | 240 } |
241 | 241 |
242 } // namespace | 242 } // namespace |
243 } // namespace test | 243 } // namespace test |
244 } // namespace net | 244 } // namespace net |
OLD | NEW |