Merge TestResourceHandlers used by two different test fixtures.

content/browser/loader/test_resource_handler.cc

+// 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() << "Redirects are not supported by the TestResourceHandler.";
+  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) {
+  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() << "Saving to file is not supported by the TestResourceHandler.";
+}
+
+void TestResourceHandler::SetBufferSize(int buffer_size) {
+  buffer_ = new net::IOBuffer(buffer_size);
+  buffer_size_ = buffer_size;
+  memset(buffer_->data(), '\0', buffer_size);
+}
+
+}  // namespace content