| 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/http/http_response_body_drainer.h" | 5 #include "net/http/http_response_body_drainer.h" |
| 6 | 6 |
| 7 #include <cstring> | 7 #include <cstring> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/compiler_specific.h" | 10 #include "base/compiler_specific.h" |
| (...skipping 179 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 190 std::memset(buf->data(), 1, buf_len); | 190 std::memset(buf->data(), 1, buf_len); |
| 191 } | 191 } |
| 192 num_chunks_--; | 192 num_chunks_--; |
| 193 if (!num_chunks_) | 193 if (!num_chunks_) |
| 194 is_complete_ = true; | 194 is_complete_ = true; |
| 195 | 195 |
| 196 return buf_len; | 196 return buf_len; |
| 197 } | 197 } |
| 198 | 198 |
| 199 void MockHttpStream::CompleteRead() { | 199 void MockHttpStream::CompleteRead() { |
| 200 int result = ReadResponseBodyImpl(user_buf_, buf_len_); | 200 int result = ReadResponseBodyImpl(user_buf_.get(), buf_len_); |
| 201 user_buf_ = NULL; | 201 user_buf_ = NULL; |
| 202 CompletionCallback callback = callback_; | 202 CompletionCallback callback = callback_; |
| 203 callback_.Reset(); | 203 callback_.Reset(); |
| 204 callback.Run(result); | 204 callback.Run(result); |
| 205 } | 205 } |
| 206 | 206 |
| 207 class HttpResponseBodyDrainerTest : public testing::Test { | 207 class HttpResponseBodyDrainerTest : public testing::Test { |
| 208 protected: | 208 protected: |
| 209 HttpResponseBodyDrainerTest() | 209 HttpResponseBodyDrainerTest() |
| 210 : proxy_service_(ProxyService::CreateDirect()), | 210 : proxy_service_(ProxyService::CreateDirect()), |
| 211 ssl_config_service_(new SSLConfigServiceDefaults), | 211 ssl_config_service_(new SSLConfigServiceDefaults), |
| 212 http_server_properties_(new HttpServerPropertiesImpl), | 212 http_server_properties_(new HttpServerPropertiesImpl), |
| 213 session_(CreateNetworkSession()), | 213 session_(CreateNetworkSession()), |
| 214 mock_stream_(new MockHttpStream(&result_waiter_)), | 214 mock_stream_(new MockHttpStream(&result_waiter_)), |
| 215 drainer_(new HttpResponseBodyDrainer(mock_stream_)) {} | 215 drainer_(new HttpResponseBodyDrainer(mock_stream_)) {} |
| 216 | 216 |
| 217 virtual ~HttpResponseBodyDrainerTest() {} | 217 virtual ~HttpResponseBodyDrainerTest() {} |
| 218 | 218 |
| 219 HttpNetworkSession* CreateNetworkSession() const { | 219 HttpNetworkSession* CreateNetworkSession() const { |
| 220 HttpNetworkSession::Params params; | 220 HttpNetworkSession::Params params; |
| 221 params.proxy_service = proxy_service_.get(); | 221 params.proxy_service = proxy_service_.get(); |
| 222 params.ssl_config_service = ssl_config_service_; | 222 params.ssl_config_service = ssl_config_service_.get(); |
| 223 params.http_server_properties = http_server_properties_.get(); | 223 params.http_server_properties = http_server_properties_.get(); |
| 224 return new HttpNetworkSession(params); | 224 return new HttpNetworkSession(params); |
| 225 } | 225 } |
| 226 | 226 |
| 227 scoped_ptr<ProxyService> proxy_service_; | 227 scoped_ptr<ProxyService> proxy_service_; |
| 228 scoped_refptr<SSLConfigService> ssl_config_service_; | 228 scoped_refptr<SSLConfigService> ssl_config_service_; |
| 229 scoped_ptr<HttpServerPropertiesImpl> http_server_properties_; | 229 scoped_ptr<HttpServerPropertiesImpl> http_server_properties_; |
| 230 const scoped_refptr<HttpNetworkSession> session_; | 230 const scoped_refptr<HttpNetworkSession> session_; |
| 231 CloseResultWaiter result_waiter_; | 231 CloseResultWaiter result_waiter_; |
| 232 MockHttpStream* const mock_stream_; // Owned by |drainer_|. | 232 MockHttpStream* const mock_stream_; // Owned by |drainer_|. |
| 233 HttpResponseBodyDrainer* const drainer_; // Deletes itself. | 233 HttpResponseBodyDrainer* const drainer_; // Deletes itself. |
| 234 }; | 234 }; |
| 235 | 235 |
| 236 TEST_F(HttpResponseBodyDrainerTest, DrainBodySyncSingleOK) { | 236 TEST_F(HttpResponseBodyDrainerTest, DrainBodySyncSingleOK) { |
| 237 mock_stream_->set_num_chunks(1); | 237 mock_stream_->set_num_chunks(1); |
| 238 mock_stream_->set_sync(); | 238 mock_stream_->set_sync(); |
| 239 drainer_->Start(session_); | 239 drainer_->Start(session_.get()); |
| 240 EXPECT_FALSE(result_waiter_.WaitForResult()); | 240 EXPECT_FALSE(result_waiter_.WaitForResult()); |
| 241 } | 241 } |
| 242 | 242 |
| 243 TEST_F(HttpResponseBodyDrainerTest, DrainBodySyncOK) { | 243 TEST_F(HttpResponseBodyDrainerTest, DrainBodySyncOK) { |
| 244 mock_stream_->set_num_chunks(3); | 244 mock_stream_->set_num_chunks(3); |
| 245 mock_stream_->set_sync(); | 245 mock_stream_->set_sync(); |
| 246 drainer_->Start(session_); | 246 drainer_->Start(session_.get()); |
| 247 EXPECT_FALSE(result_waiter_.WaitForResult()); | 247 EXPECT_FALSE(result_waiter_.WaitForResult()); |
| 248 } | 248 } |
| 249 | 249 |
| 250 TEST_F(HttpResponseBodyDrainerTest, DrainBodyAsyncOK) { | 250 TEST_F(HttpResponseBodyDrainerTest, DrainBodyAsyncOK) { |
| 251 mock_stream_->set_num_chunks(3); | 251 mock_stream_->set_num_chunks(3); |
| 252 drainer_->Start(session_); | 252 drainer_->Start(session_.get()); |
| 253 EXPECT_FALSE(result_waiter_.WaitForResult()); | 253 EXPECT_FALSE(result_waiter_.WaitForResult()); |
| 254 } | 254 } |
| 255 | 255 |
| 256 // Test the case when the final chunk is 0 bytes. This can happen when | 256 // Test the case when the final chunk is 0 bytes. This can happen when |
| 257 // the final 0-byte chunk of a chunk-encoded http response is read in a last | 257 // the final 0-byte chunk of a chunk-encoded http response is read in a last |
| 258 // call to ReadResponseBody, after all data were returned from HttpStream. | 258 // call to ReadResponseBody, after all data were returned from HttpStream. |
| 259 TEST_F(HttpResponseBodyDrainerTest, DrainBodyAsyncEmptyChunk) { | 259 TEST_F(HttpResponseBodyDrainerTest, DrainBodyAsyncEmptyChunk) { |
| 260 mock_stream_->set_num_chunks(4); | 260 mock_stream_->set_num_chunks(4); |
| 261 mock_stream_->set_is_last_chunk_zero_size(); | 261 mock_stream_->set_is_last_chunk_zero_size(); |
| 262 drainer_->Start(session_); | 262 drainer_->Start(session_.get()); |
| 263 EXPECT_FALSE(result_waiter_.WaitForResult()); | 263 EXPECT_FALSE(result_waiter_.WaitForResult()); |
| 264 } | 264 } |
| 265 | 265 |
| 266 TEST_F(HttpResponseBodyDrainerTest, DrainBodySyncEmptyChunk) { | 266 TEST_F(HttpResponseBodyDrainerTest, DrainBodySyncEmptyChunk) { |
| 267 mock_stream_->set_num_chunks(4); | 267 mock_stream_->set_num_chunks(4); |
| 268 mock_stream_->set_sync(); | 268 mock_stream_->set_sync(); |
| 269 mock_stream_->set_is_last_chunk_zero_size(); | 269 mock_stream_->set_is_last_chunk_zero_size(); |
| 270 drainer_->Start(session_); | 270 drainer_->Start(session_.get()); |
| 271 EXPECT_FALSE(result_waiter_.WaitForResult()); | 271 EXPECT_FALSE(result_waiter_.WaitForResult()); |
| 272 } | 272 } |
| 273 | 273 |
| 274 TEST_F(HttpResponseBodyDrainerTest, DrainBodySizeEqualsDrainBuffer) { | 274 TEST_F(HttpResponseBodyDrainerTest, DrainBodySizeEqualsDrainBuffer) { |
| 275 mock_stream_->set_num_chunks( | 275 mock_stream_->set_num_chunks( |
| 276 HttpResponseBodyDrainer::kDrainBodyBufferSize / kMagicChunkSize); | 276 HttpResponseBodyDrainer::kDrainBodyBufferSize / kMagicChunkSize); |
| 277 drainer_->Start(session_); | 277 drainer_->Start(session_.get()); |
| 278 EXPECT_FALSE(result_waiter_.WaitForResult()); | 278 EXPECT_FALSE(result_waiter_.WaitForResult()); |
| 279 } | 279 } |
| 280 | 280 |
| 281 TEST_F(HttpResponseBodyDrainerTest, DrainBodyTimeOut) { | 281 TEST_F(HttpResponseBodyDrainerTest, DrainBodyTimeOut) { |
| 282 mock_stream_->set_num_chunks(2); | 282 mock_stream_->set_num_chunks(2); |
| 283 mock_stream_->set_stall_reads_forever(); | 283 mock_stream_->set_stall_reads_forever(); |
| 284 drainer_->Start(session_); | 284 drainer_->Start(session_.get()); |
| 285 EXPECT_TRUE(result_waiter_.WaitForResult()); | 285 EXPECT_TRUE(result_waiter_.WaitForResult()); |
| 286 } | 286 } |
| 287 | 287 |
| 288 TEST_F(HttpResponseBodyDrainerTest, CancelledBySession) { | 288 TEST_F(HttpResponseBodyDrainerTest, CancelledBySession) { |
| 289 mock_stream_->set_num_chunks(2); | 289 mock_stream_->set_num_chunks(2); |
| 290 mock_stream_->set_stall_reads_forever(); | 290 mock_stream_->set_stall_reads_forever(); |
| 291 drainer_->Start(session_); | 291 drainer_->Start(session_.get()); |
| 292 // HttpNetworkSession should delete |drainer_|. | 292 // HttpNetworkSession should delete |drainer_|. |
| 293 } | 293 } |
| 294 | 294 |
| 295 TEST_F(HttpResponseBodyDrainerTest, DrainBodyTooLarge) { | 295 TEST_F(HttpResponseBodyDrainerTest, DrainBodyTooLarge) { |
| 296 int too_many_chunks = | 296 int too_many_chunks = |
| 297 HttpResponseBodyDrainer::kDrainBodyBufferSize / kMagicChunkSize; | 297 HttpResponseBodyDrainer::kDrainBodyBufferSize / kMagicChunkSize; |
| 298 too_many_chunks += 1; // Now it's too large. | 298 too_many_chunks += 1; // Now it's too large. |
| 299 | 299 |
| 300 mock_stream_->set_num_chunks(too_many_chunks); | 300 mock_stream_->set_num_chunks(too_many_chunks); |
| 301 drainer_->Start(session_); | 301 drainer_->Start(session_.get()); |
| 302 EXPECT_TRUE(result_waiter_.WaitForResult()); | 302 EXPECT_TRUE(result_waiter_.WaitForResult()); |
| 303 } | 303 } |
| 304 | 304 |
| 305 TEST_F(HttpResponseBodyDrainerTest, StartBodyTooLarge) { | 305 TEST_F(HttpResponseBodyDrainerTest, StartBodyTooLarge) { |
| 306 int too_many_chunks = | 306 int too_many_chunks = |
| 307 HttpResponseBodyDrainer::kDrainBodyBufferSize / kMagicChunkSize; | 307 HttpResponseBodyDrainer::kDrainBodyBufferSize / kMagicChunkSize; |
| 308 too_many_chunks += 1; // Now it's too large. | 308 too_many_chunks += 1; // Now it's too large. |
| 309 | 309 |
| 310 mock_stream_->set_num_chunks(0); | 310 mock_stream_->set_num_chunks(0); |
| 311 drainer_->StartWithSize(session_, too_many_chunks * kMagicChunkSize); | 311 drainer_->StartWithSize(session_.get(), too_many_chunks * kMagicChunkSize); |
| 312 EXPECT_TRUE(result_waiter_.WaitForResult()); | 312 EXPECT_TRUE(result_waiter_.WaitForResult()); |
| 313 } | 313 } |
| 314 | 314 |
| 315 TEST_F(HttpResponseBodyDrainerTest, StartWithNothingToDo) { | 315 TEST_F(HttpResponseBodyDrainerTest, StartWithNothingToDo) { |
| 316 mock_stream_->set_num_chunks(0); | 316 mock_stream_->set_num_chunks(0); |
| 317 drainer_->StartWithSize(session_, 0); | 317 drainer_->StartWithSize(session_.get(), 0); |
| 318 EXPECT_FALSE(result_waiter_.WaitForResult()); | 318 EXPECT_FALSE(result_waiter_.WaitForResult()); |
| 319 } | 319 } |
| 320 | 320 |
| 321 } // namespace | 321 } // namespace |
| 322 | 322 |
| 323 } // namespace net | 323 } // namespace net |
| OLD | NEW |