| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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/url_request/url_request_test_util.h" | 5 #include "net/url_request/url_request_test_util.h" |
| 6 | 6 |
| 7 #include "base/compiler_specific.h" | 7 #include "base/compiler_specific.h" |
| 8 #include "base/logging.h" | 8 #include "base/logging.h" |
| 9 #include "base/message_loop.h" | 9 #include "base/message_loop.h" |
| 10 #include "base/threading/thread.h" | 10 #include "base/threading/thread.h" |
| (...skipping 321 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 332 | 332 |
| 333 void TestNetworkDelegate::InitRequestStatesIfNew(int request_id) { | 333 void TestNetworkDelegate::InitRequestStatesIfNew(int request_id) { |
| 334 if (next_states_.find(request_id) == next_states_.end()) { | 334 if (next_states_.find(request_id) == next_states_.end()) { |
| 335 next_states_[request_id] = kStageBeforeURLRequest; | 335 next_states_[request_id] = kStageBeforeURLRequest; |
| 336 event_order_[request_id] = ""; | 336 event_order_[request_id] = ""; |
| 337 } | 337 } |
| 338 } | 338 } |
| 339 | 339 |
| 340 int TestNetworkDelegate::OnBeforeURLRequest( | 340 int TestNetworkDelegate::OnBeforeURLRequest( |
| 341 net::URLRequest* request, | 341 net::URLRequest* request, |
| 342 net::OldCompletionCallback* callback, | 342 const net::CompletionCallback& callback, |
| 343 GURL* new_url ) { | 343 GURL* new_url ) { |
| 344 int req_id = request->identifier(); | 344 int req_id = request->identifier(); |
| 345 InitRequestStatesIfNew(req_id); | 345 InitRequestStatesIfNew(req_id); |
| 346 event_order_[req_id] += "OnBeforeURLRequest\n"; | 346 event_order_[req_id] += "OnBeforeURLRequest\n"; |
| 347 EXPECT_TRUE(next_states_[req_id] & kStageBeforeURLRequest) << | 347 EXPECT_TRUE(next_states_[req_id] & kStageBeforeURLRequest) << |
| 348 event_order_[req_id]; | 348 event_order_[req_id]; |
| 349 next_states_[req_id] = | 349 next_states_[req_id] = |
| 350 kStageBeforeSendHeaders | | 350 kStageBeforeSendHeaders | |
| 351 kStageResponseStarted | // data: URLs do not trigger sending headers | 351 kStageResponseStarted | // data: URLs do not trigger sending headers |
| 352 kStageBeforeRedirect | // a delegate can trigger a redirection | 352 kStageBeforeRedirect | // a delegate can trigger a redirection |
| 353 kStageCompletedError | // request canceled by delegate | 353 kStageCompletedError | // request canceled by delegate |
| 354 kStageAuthRequired; // Auth can come next for FTP requests | 354 kStageAuthRequired; // Auth can come next for FTP requests |
| 355 created_requests_++; | 355 created_requests_++; |
| 356 return net::OK; | 356 return net::OK; |
| 357 } | 357 } |
| 358 | 358 |
| 359 int TestNetworkDelegate::OnBeforeSendHeaders( | 359 int TestNetworkDelegate::OnBeforeSendHeaders( |
| 360 net::URLRequest* request, | 360 net::URLRequest* request, |
| 361 net::OldCompletionCallback* callback, | 361 const net::CompletionCallback& callback, |
| 362 net::HttpRequestHeaders* headers) { | 362 net::HttpRequestHeaders* headers) { |
| 363 int req_id = request->identifier(); | 363 int req_id = request->identifier(); |
| 364 InitRequestStatesIfNew(req_id); | 364 InitRequestStatesIfNew(req_id); |
| 365 event_order_[req_id] += "OnBeforeSendHeaders\n"; | 365 event_order_[req_id] += "OnBeforeSendHeaders\n"; |
| 366 EXPECT_TRUE(next_states_[req_id] & kStageBeforeSendHeaders) << | 366 EXPECT_TRUE(next_states_[req_id] & kStageBeforeSendHeaders) << |
| 367 event_order_[req_id]; | 367 event_order_[req_id]; |
| 368 next_states_[req_id] = | 368 next_states_[req_id] = |
| 369 kStageSendHeaders | | 369 kStageSendHeaders | |
| 370 kStageCompletedError; // request canceled by delegate | 370 kStageCompletedError; // request canceled by delegate |
| 371 | 371 |
| 372 return net::OK; | 372 return net::OK; |
| 373 } | 373 } |
| 374 | 374 |
| 375 void TestNetworkDelegate::OnSendHeaders( | 375 void TestNetworkDelegate::OnSendHeaders( |
| 376 net::URLRequest* request, | 376 net::URLRequest* request, |
| 377 const net::HttpRequestHeaders& headers) { | 377 const net::HttpRequestHeaders& headers) { |
| 378 int req_id = request->identifier(); | 378 int req_id = request->identifier(); |
| 379 InitRequestStatesIfNew(req_id); | 379 InitRequestStatesIfNew(req_id); |
| 380 event_order_[req_id] += "OnSendHeaders\n"; | 380 event_order_[req_id] += "OnSendHeaders\n"; |
| 381 EXPECT_TRUE(next_states_[req_id] & kStageSendHeaders) << | 381 EXPECT_TRUE(next_states_[req_id] & kStageSendHeaders) << |
| 382 event_order_[req_id]; | 382 event_order_[req_id]; |
| 383 next_states_[req_id] = | 383 next_states_[req_id] = |
| 384 kStageHeadersReceived | | 384 kStageHeadersReceived | |
| 385 kStageCompletedError; | 385 kStageCompletedError; |
| 386 } | 386 } |
| 387 | 387 |
| 388 int TestNetworkDelegate::OnHeadersReceived( | 388 int TestNetworkDelegate::OnHeadersReceived( |
| 389 net::URLRequest* request, | 389 net::URLRequest* request, |
| 390 net::OldCompletionCallback* callback, | 390 const net::CompletionCallback& callback, |
| 391 net::HttpResponseHeaders* original_response_headers, | 391 net::HttpResponseHeaders* original_response_headers, |
| 392 scoped_refptr<net::HttpResponseHeaders>* override_response_headers) { | 392 scoped_refptr<net::HttpResponseHeaders>* override_response_headers) { |
| 393 int req_id = request->identifier(); | 393 int req_id = request->identifier(); |
| 394 event_order_[req_id] += "OnHeadersReceived\n"; | 394 event_order_[req_id] += "OnHeadersReceived\n"; |
| 395 InitRequestStatesIfNew(req_id); | 395 InitRequestStatesIfNew(req_id); |
| 396 EXPECT_TRUE(next_states_[req_id] & kStageHeadersReceived) << | 396 EXPECT_TRUE(next_states_[req_id] & kStageHeadersReceived) << |
| 397 event_order_[req_id]; | 397 event_order_[req_id]; |
| 398 next_states_[req_id] = | 398 next_states_[req_id] = |
| 399 kStageBeforeRedirect | | 399 kStageBeforeRedirect | |
| 400 kStageResponseStarted | | 400 kStageResponseStarted | |
| (...skipping 88 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 489 event_order_[req_id] += "OnAuthRequired\n"; | 489 event_order_[req_id] += "OnAuthRequired\n"; |
| 490 EXPECT_TRUE(next_states_[req_id] & kStageAuthRequired) << | 490 EXPECT_TRUE(next_states_[req_id] & kStageAuthRequired) << |
| 491 event_order_[req_id]; | 491 event_order_[req_id]; |
| 492 next_states_[req_id] = kStageBeforeSendHeaders | | 492 next_states_[req_id] = kStageBeforeSendHeaders | |
| 493 kStageHeadersReceived | // Request canceled by delegate simulates empty | 493 kStageHeadersReceived | // Request canceled by delegate simulates empty |
| 494 // response. | 494 // response. |
| 495 kStageResponseStarted | // data: URLs do not trigger sending headers | 495 kStageResponseStarted | // data: URLs do not trigger sending headers |
| 496 kStageBeforeRedirect; // a delegate can trigger a redirection | 496 kStageBeforeRedirect; // a delegate can trigger a redirection |
| 497 return net::NetworkDelegate::AUTH_REQUIRED_RESPONSE_NO_ACTION; | 497 return net::NetworkDelegate::AUTH_REQUIRED_RESPONSE_NO_ACTION; |
| 498 } | 498 } |
| OLD | NEW |