Chromium Code Reviews| 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 |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..323759f9d6dc3221f5bc06c7cbfc927a22108fc0 |
| --- /dev/null |
| +++ b/content/browser/loader/test_resource_handler.cc |
| @@ -0,0 +1,111 @@ |
| +// Copyright 2016 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#include "content/browser/loader/test_resource_handler.h" |
| + |
| +#include "base/logging.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_(request_status), body_(body) { |
| + SetBufferSize(2048); |
| +} |
| + |
| +TestResourceHandler::TestResourceHandler() |
| + : TestResourceHandler(nullptr, nullptr) {} |
| + |
| +TestResourceHandler::~TestResourceHandler() {} |
| + |
| +void TestResourceHandler::SetController(ResourceController* controller) {} |
| + |
| +bool TestResourceHandler::OnRequestRedirected( |
| + const net::RedirectInfo& redirect_info, |
| + ResourceResponse* response, |
| + bool* defer) { |
| + NOTREACHED(); |
|
Charlie Harrison
2016/10/26 14:05:19
Can you NOTREACHED() << "Redirects are not support
mmenke
2016/10/26 15:27:29
Added the text, kept NOTREACHED(). Experimenting
|
| + return false; |
| +} |
| + |
| +bool TestResourceHandler::OnResponseStarted(ResourceResponse* response, |
| + bool* defer) { |
| + 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_; |
| + |
| + if (!on_response_started_result_) |
| + return false; |
| + *defer = defer_on_response_started_; |
| + defer_on_response_started_ = false; |
| + return true; |
| +} |
| + |
| +bool TestResourceHandler::OnWillStart(const GURL& url, bool* defer) { |
|
Charlie Harrison
2016/10/26 14:05:19
include gurl.h
mmenke
2016/10/26 15:27:29
I don't think it's needed - we never dereference i
|
| + 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_; |
| + |
| + if (!on_will_start_result_) |
| + return false; |
| + |
| + *defer = defer_on_will_start_; |
| + return true; |
| +} |
| + |
| +bool TestResourceHandler::OnWillRead(scoped_refptr<net::IOBuffer>* buf, |
| + int* buf_size, |
| + int min_size) { |
| + EXPECT_EQ(0, on_response_completed_called_); |
| + ++on_will_read_called_; |
| + |
| + *buf = buffer_; |
| + *buf_size = buffer_size_; |
| + memset(buffer_->data(), '\0', buffer_size_); |
| + return on_will_read_result_; |
| +} |
| + |
| +bool TestResourceHandler::OnReadCompleted(int bytes_read, bool* defer) { |
| + EXPECT_EQ(1, on_will_start_called_); |
| + EXPECT_EQ(1, on_response_started_called_); |
| + EXPECT_EQ(0, on_response_completed_called_); |
| + ++on_read_completed_called_; |
| + |
| + EXPECT_LE(static_cast<size_t>(bytes_read), buffer_size_); |
| + 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; |
| + return true; |
| +} |
| + |
| +void TestResourceHandler::OnResponseCompleted( |
| + const net::URLRequestStatus& status, |
| + bool* defer) { |
| + EXPECT_EQ(0, on_response_completed_called_); |
| + ++on_response_completed_called_; |
| + |
| + if (request_status_) |
| + *request_status_ = status; |
| + *defer = defer_on_response_completed_; |
| + defer_on_response_completed_ = false; |
| +} |
| + |
| +void TestResourceHandler::OnDataDownloaded(int bytes_downloaded) { |
| + NOTREACHED(); |
|
Charlie Harrison
2016/10/26 14:05:19
ditto
mmenke
2016/10/26 15:27:29
Done.
|
| +} |
| + |
| +void TestResourceHandler::SetBufferSize(int buffer_size) { |
| + buffer_ = new net::IOBuffer(buffer_size); |
| + buffer_size_ = buffer_size; |
| + memset(buffer_->data(), '\0', buffer_size); |
| +} |
| + |
| +} // namespace content |