| Index: content/browser/loader/test_resource_handler.cc
|
| diff --git a/content/browser/loader/test_resource_handler.cc b/content/browser/loader/test_resource_handler.cc
|
| index b31ed8b984f5ac9f1e84aeacd69e21a26fc2485b..0ee72ae6744d140e22a9c3cf27a251259851c5b5 100644
|
| --- a/content/browser/loader/test_resource_handler.cc
|
| +++ b/content/browser/loader/test_resource_handler.cc
|
| @@ -5,18 +5,14 @@
|
| #include "content/browser/loader/test_resource_handler.h"
|
|
|
| #include "base/logging.h"
|
| -#include "content/public/browser/resource_controller.h"
|
| -#include "content/public/common/resource_response.h"
|
| +#include "net/url_request/url_request_status.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
|
|
| namespace content {
|
|
|
| TestResourceHandler::TestResourceHandler(net::URLRequestStatus* request_status,
|
| std::string* body)
|
| - : ResourceHandler(nullptr),
|
| - request_status_ptr_(request_status),
|
| - body_ptr_(body),
|
| - deferred_run_loop_(new base::RunLoop()) {
|
| + : ResourceHandler(nullptr), request_status_(request_status), body_(body) {
|
| SetBufferSize(2048);
|
| }
|
|
|
| @@ -25,122 +21,68 @@
|
|
|
| TestResourceHandler::~TestResourceHandler() {}
|
|
|
| -void TestResourceHandler::SetController(ResourceController* controller) {
|
| - controller_ = controller;
|
| -}
|
| +void TestResourceHandler::SetController(ResourceController* controller) {}
|
|
|
| bool TestResourceHandler::OnRequestRedirected(
|
| const net::RedirectInfo& redirect_info,
|
| ResourceResponse* response,
|
| bool* defer) {
|
| - EXPECT_FALSE(canceled_);
|
| - EXPECT_EQ(1, on_will_start_called_);
|
| - EXPECT_EQ(0, on_response_started_called_);
|
| - EXPECT_EQ(0, on_response_completed_called_);
|
| - ++on_request_redirected_called_;
|
| -
|
| - if (!on_request_redirected_result_) {
|
| - canceled_ = true;
|
| - return false;
|
| - }
|
| -
|
| - *defer = defer_on_request_redirected_;
|
| - defer_on_request_redirected_ = false;
|
| - if (*defer)
|
| - deferred_run_loop_->Quit();
|
| - return true;
|
| + NOTREACHED() << "Redirects are not supported by the TestResourceHandler.";
|
| + return false;
|
| }
|
|
|
| bool TestResourceHandler::OnResponseStarted(ResourceResponse* response,
|
| bool* defer) {
|
| - EXPECT_FALSE(canceled_);
|
| EXPECT_EQ(1, on_will_start_called_);
|
| EXPECT_EQ(0, on_response_started_called_);
|
| EXPECT_EQ(0, on_response_completed_called_);
|
| ++on_response_started_called_;
|
|
|
| - EXPECT_FALSE(resource_response_);
|
| - resource_response_ = response;
|
| -
|
| - if (!on_response_started_result_) {
|
| - canceled_ = true;
|
| + if (!on_response_started_result_)
|
| return false;
|
| - }
|
| -
|
| *defer = defer_on_response_started_;
|
| defer_on_response_started_ = false;
|
| - if (*defer)
|
| - deferred_run_loop_->Quit();
|
| return true;
|
| }
|
|
|
| bool TestResourceHandler::OnWillStart(const GURL& url, bool* defer) {
|
| - EXPECT_FALSE(canceled_);
|
| EXPECT_EQ(0, on_response_started_called_);
|
| EXPECT_EQ(0, on_will_start_called_);
|
| EXPECT_EQ(0, on_response_completed_called_);
|
| ++on_will_start_called_;
|
|
|
| - start_url_ = url;
|
| -
|
| - if (!on_will_start_result_) {
|
| - canceled_ = true;
|
| + if (!on_will_start_result_)
|
| return false;
|
| - }
|
|
|
| *defer = defer_on_will_start_;
|
| - if (*defer)
|
| - deferred_run_loop_->Quit();
|
| return true;
|
| }
|
|
|
| bool TestResourceHandler::OnWillRead(scoped_refptr<net::IOBuffer>* buf,
|
| int* buf_size,
|
| int min_size) {
|
| - EXPECT_FALSE(canceled_);
|
| - EXPECT_FALSE(expect_on_data_downloaded_);
|
| EXPECT_EQ(0, on_response_completed_called_);
|
| ++on_will_read_called_;
|
|
|
| *buf = buffer_;
|
| *buf_size = buffer_size_;
|
| memset(buffer_->data(), '\0', buffer_size_);
|
| - if (!on_will_read_result_)
|
| - canceled_ = true;
|
| return on_will_read_result_;
|
| }
|
|
|
| bool TestResourceHandler::OnReadCompleted(int bytes_read, bool* defer) {
|
| - EXPECT_FALSE(canceled_);
|
| - EXPECT_FALSE(expect_on_data_downloaded_);
|
| EXPECT_EQ(1, on_will_start_called_);
|
| EXPECT_EQ(1, on_response_started_called_);
|
| EXPECT_EQ(0, on_response_completed_called_);
|
| - EXPECT_EQ(0, on_read_eof_);
|
| -
|
| ++on_read_completed_called_;
|
| - if (bytes_read == 0)
|
| - ++on_read_eof_;
|
|
|
| EXPECT_LE(static_cast<size_t>(bytes_read), buffer_size_);
|
| - if (body_ptr_)
|
| - body_ptr_->append(buffer_->data(), bytes_read);
|
| - body_.append(buffer_->data(), bytes_read);
|
| -
|
| - if (!on_read_completed_result_ ||
|
| - (!on_on_read_eof_result_ && bytes_read == 0)) {
|
| - canceled_ = true;
|
| + if (body_)
|
| + body_->append(buffer_->data(), bytes_read);
|
| + if (!on_read_completed_result_)
|
| return false;
|
| - }
|
| -
|
| *defer = defer_on_read_completed_;
|
| defer_on_read_completed_ = false;
|
| - if (bytes_read == 0 && defer_on_read_eof_)
|
| - *defer = true;
|
| -
|
| - if (*defer)
|
| - deferred_run_loop_->Quit();
|
| -
|
| return true;
|
| }
|
|
|
| @@ -148,34 +90,16 @@
|
| const net::URLRequestStatus& status,
|
| bool* defer) {
|
| EXPECT_EQ(0, on_response_completed_called_);
|
| - if (status.is_success() && !expect_on_data_downloaded_ && expect_eof_read_)
|
| - EXPECT_EQ(1, on_read_eof_);
|
| -
|
| ++on_response_completed_called_;
|
|
|
| - if (request_status_ptr_)
|
| - *request_status_ptr_ = status;
|
| - final_status_ = status;
|
| + if (request_status_)
|
| + *request_status_ = status;
|
| *defer = defer_on_response_completed_;
|
| defer_on_response_completed_ = false;
|
| -
|
| - if (*defer)
|
| - deferred_run_loop_->Quit();
|
| - response_complete_run_loop_.Quit();
|
| }
|
|
|
| void TestResourceHandler::OnDataDownloaded(int bytes_downloaded) {
|
| - EXPECT_TRUE(expect_on_data_downloaded_);
|
| - total_bytes_downloaded_ += bytes_downloaded;
|
| -}
|
| -
|
| -void TestResourceHandler::Resume() {
|
| - controller_->Resume();
|
| -}
|
| -
|
| -void TestResourceHandler::CancelWithError(net::Error net_error) {
|
| - canceled_ = true;
|
| - controller_->CancelWithError(net_error);
|
| + NOTREACHED() << "Saving to file is not supported by the TestResourceHandler.";
|
| }
|
|
|
| void TestResourceHandler::SetBufferSize(int buffer_size) {
|
| @@ -184,13 +108,4 @@
|
| memset(buffer_->data(), '\0', buffer_size);
|
| }
|
|
|
| -void TestResourceHandler::WaitUntilDeferred() {
|
| - deferred_run_loop_->Run();
|
| - deferred_run_loop_.reset(new base::RunLoop());
|
| -}
|
| -
|
| -void TestResourceHandler::WaitUntilResponseComplete() {
|
| - response_complete_run_loop_.Run();
|
| -}
|
| -
|
| } // namespace content
|
|
|