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