OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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_transaction_test_util.h" | 5 #include "net/http/http_transaction_test_util.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/message_loop/message_loop.h" | 10 #include "base/message_loop/message_loop.h" |
11 #include "base/strings/stringprintf.h" | 11 #include "base/strings/stringprintf.h" |
| 12 #include "base/time/clock.h" |
12 #include "base/time/time.h" | 13 #include "base/time/time.h" |
13 #include "net/base/load_flags.h" | 14 #include "net/base/load_flags.h" |
14 #include "net/base/load_timing_info.h" | 15 #include "net/base/load_timing_info.h" |
15 #include "net/base/net_errors.h" | 16 #include "net/base/net_errors.h" |
16 #include "net/disk_cache/disk_cache.h" | 17 #include "net/disk_cache/disk_cache.h" |
17 #include "net/http/http_cache.h" | 18 #include "net/http/http_cache.h" |
18 #include "net/http/http_request_info.h" | 19 #include "net/http/http_request_info.h" |
19 #include "net/http/http_response_info.h" | 20 #include "net/http/http_response_info.h" |
20 #include "net/http/http_transaction.h" | 21 #include "net/http/http_transaction.h" |
21 #include "testing/gtest/include/gtest/gtest.h" | 22 #include "testing/gtest/include/gtest/gtest.h" |
(...skipping 370 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
392 std::string resp_headers = t->response_headers; | 393 std::string resp_headers = t->response_headers; |
393 std::string resp_data = t->data; | 394 std::string resp_data = t->data; |
394 received_bytes_ = resp_status.size() + resp_headers.size() + resp_data.size(); | 395 received_bytes_ = resp_status.size() + resp_headers.size() + resp_data.size(); |
395 if (t->handler) | 396 if (t->handler) |
396 (t->handler)(request, &resp_status, &resp_headers, &resp_data); | 397 (t->handler)(request, &resp_status, &resp_headers, &resp_data); |
397 | 398 |
398 std::string header_data = base::StringPrintf( | 399 std::string header_data = base::StringPrintf( |
399 "%s\n%s\n", resp_status.c_str(), resp_headers.c_str()); | 400 "%s\n%s\n", resp_status.c_str(), resp_headers.c_str()); |
400 std::replace(header_data.begin(), header_data.end(), '\n', '\0'); | 401 std::replace(header_data.begin(), header_data.end(), '\n', '\0'); |
401 | 402 |
402 response_.request_time = base::Time::Now(); | 403 response_.request_time = transaction_factory_->Now(); |
403 if (!t->request_time.is_null()) | 404 if (!t->request_time.is_null()) |
404 response_.request_time = t->request_time; | 405 response_.request_time = t->request_time; |
405 | 406 |
406 response_.was_cached = false; | 407 response_.was_cached = false; |
407 response_.network_accessed = true; | 408 response_.network_accessed = true; |
408 | 409 |
409 response_.response_time = base::Time::Now(); | 410 response_.response_time = transaction_factory_->Now(); |
410 if (!t->response_time.is_null()) | 411 if (!t->response_time.is_null()) |
411 response_.response_time = t->response_time; | 412 response_.response_time = t->response_time; |
412 | 413 |
413 response_.headers = new net::HttpResponseHeaders(header_data); | 414 response_.headers = new net::HttpResponseHeaders(header_data); |
414 response_.vary_data.Init(*request, *response_.headers.get()); | 415 response_.vary_data.Init(*request, *response_.headers.get()); |
415 response_.ssl_info.cert_status = t->cert_status; | 416 response_.ssl_info.cert_status = t->cert_status; |
416 data_ = resp_data; | 417 data_ = resp_data; |
417 | 418 |
418 if (net_log.net_log()) | 419 if (net_log.net_log()) |
419 socket_log_id_ = net_log.net_log()->NextID(); | 420 socket_log_id_ = net_log.net_log()->NextID(); |
420 | 421 |
| 422 if (request_->load_flags & net::LOAD_PREFETCH) |
| 423 response_.unused_since_prefetch = true; |
| 424 |
421 if (test_mode_ & TEST_MODE_SYNC_NET_START) | 425 if (test_mode_ & TEST_MODE_SYNC_NET_START) |
422 return net::OK; | 426 return net::OK; |
423 | 427 |
424 CallbackLater(callback, net::OK); | 428 CallbackLater(callback, net::OK); |
425 return net::ERR_IO_PENDING; | 429 return net::ERR_IO_PENDING; |
426 } | 430 } |
427 | 431 |
428 void MockNetworkTransaction::SetBeforeNetworkStartCallback( | 432 void MockNetworkTransaction::SetBeforeNetworkStartCallback( |
429 const BeforeNetworkStartCallback& callback) { | 433 const BeforeNetworkStartCallback& callback) { |
430 } | 434 } |
(...skipping 16 matching lines...) Expand all Loading... |
447 | 451 |
448 void MockNetworkTransaction::RunCallback( | 452 void MockNetworkTransaction::RunCallback( |
449 const net::CompletionCallback& callback, int result) { | 453 const net::CompletionCallback& callback, int result) { |
450 callback.Run(result); | 454 callback.Run(result); |
451 } | 455 } |
452 | 456 |
453 MockNetworkLayer::MockNetworkLayer() | 457 MockNetworkLayer::MockNetworkLayer() |
454 : transaction_count_(0), | 458 : transaction_count_(0), |
455 done_reading_called_(false), | 459 done_reading_called_(false), |
456 stop_caching_called_(false), | 460 stop_caching_called_(false), |
457 last_create_transaction_priority_(net::DEFAULT_PRIORITY) {} | 461 last_create_transaction_priority_(net::DEFAULT_PRIORITY), |
| 462 clock_(nullptr) { |
| 463 } |
458 | 464 |
459 MockNetworkLayer::~MockNetworkLayer() {} | 465 MockNetworkLayer::~MockNetworkLayer() {} |
460 | 466 |
461 void MockNetworkLayer::TransactionDoneReading() { | 467 void MockNetworkLayer::TransactionDoneReading() { |
462 done_reading_called_ = true; | 468 done_reading_called_ = true; |
463 } | 469 } |
464 | 470 |
465 void MockNetworkLayer::TransactionStopCaching() { | 471 void MockNetworkLayer::TransactionStopCaching() { |
466 stop_caching_called_ = true; | 472 stop_caching_called_ = true; |
467 } | 473 } |
(...skipping 11 matching lines...) Expand all Loading... |
479 } | 485 } |
480 | 486 |
481 net::HttpCache* MockNetworkLayer::GetCache() { | 487 net::HttpCache* MockNetworkLayer::GetCache() { |
482 return NULL; | 488 return NULL; |
483 } | 489 } |
484 | 490 |
485 net::HttpNetworkSession* MockNetworkLayer::GetSession() { | 491 net::HttpNetworkSession* MockNetworkLayer::GetSession() { |
486 return NULL; | 492 return NULL; |
487 } | 493 } |
488 | 494 |
| 495 void MockNetworkLayer::SetClock(base::Clock* clock) { |
| 496 DCHECK(!clock_); |
| 497 clock_ = clock; |
| 498 } |
| 499 |
| 500 base::Time MockNetworkLayer::Now() { |
| 501 if (clock_) |
| 502 return clock_->Now(); |
| 503 return base::Time::Now(); |
| 504 } |
| 505 |
489 //----------------------------------------------------------------------------- | 506 //----------------------------------------------------------------------------- |
490 // helpers | 507 // helpers |
491 | 508 |
492 int ReadTransaction(net::HttpTransaction* trans, std::string* result) { | 509 int ReadTransaction(net::HttpTransaction* trans, std::string* result) { |
493 int rv; | 510 int rv; |
494 | 511 |
495 net::TestCompletionCallback callback; | 512 net::TestCompletionCallback callback; |
496 | 513 |
497 std::string content; | 514 std::string content; |
498 do { | 515 do { |
499 scoped_refptr<net::IOBuffer> buf(new net::IOBuffer(256)); | 516 scoped_refptr<net::IOBuffer> buf(new net::IOBuffer(256)); |
500 rv = trans->Read(buf.get(), 256, callback.callback()); | 517 rv = trans->Read(buf.get(), 256, callback.callback()); |
501 if (rv == net::ERR_IO_PENDING) | 518 if (rv == net::ERR_IO_PENDING) |
502 rv = callback.WaitForResult(); | 519 rv = callback.WaitForResult(); |
503 | 520 |
504 if (rv > 0) | 521 if (rv > 0) |
505 content.append(buf->data(), rv); | 522 content.append(buf->data(), rv); |
506 else if (rv < 0) | 523 else if (rv < 0) |
507 return rv; | 524 return rv; |
508 } while (rv > 0); | 525 } while (rv > 0); |
509 | 526 |
510 result->swap(content); | 527 result->swap(content); |
511 return net::OK; | 528 return net::OK; |
512 } | 529 } |
OLD | NEW |