| Index: net/test/url_request/url_request_mock_http_job.cc
|
| diff --git a/content/test/net/url_request_mock_http_job.cc b/net/test/url_request/url_request_mock_http_job.cc
|
| similarity index 66%
|
| rename from content/test/net/url_request_mock_http_job.cc
|
| rename to net/test/url_request/url_request_mock_http_job.cc
|
| index 50fb1c836a369662fae1196e2a5ecb6931c2319d..afe8e4977db1fbf893bcb93fa602de439dc9b1ff 100644
|
| --- a/content/test/net/url_request_mock_http_job.cc
|
| +++ b/net/test/url_request/url_request_mock_http_job.cc
|
| @@ -2,16 +2,15 @@
|
| // Use of this source code is governed by a BSD-style license that can be
|
| // found in the LICENSE file.
|
|
|
| -#include "content/test/net/url_request_mock_http_job.h"
|
| +#include "net/test/url_request/url_request_mock_http_job.h"
|
|
|
| -#include "base/files/file_util.h"
|
| +#include "base/file_util.h"
|
| #include "base/message_loop/message_loop.h"
|
| #include "base/strings/string_util.h"
|
| #include "base/strings/utf_string_conversions.h"
|
| +#include "base/task_runner_util.h"
|
| #include "base/threading/sequenced_worker_pool.h"
|
| #include "base/threading/thread_restrictions.h"
|
| -#include "content/public/browser/browser_thread.h"
|
| -#include "content/public/common/url_constants.h"
|
| #include "net/base/filename_util.h"
|
| #include "net/http/http_response_headers.h"
|
| #include "net/url_request/url_request_filter.h"
|
| @@ -21,7 +20,7 @@ const char kMockHostname[] = "mock.http";
|
| const base::FilePath::CharType kMockHeaderFileSuffix[] =
|
| FILE_PATH_LITERAL(".mock-http-headers");
|
|
|
| -namespace content {
|
| +namespace net {
|
|
|
| namespace {
|
|
|
| @@ -31,18 +30,25 @@ class MockJobInterceptor : public net::URLRequestInterceptor {
|
| // contents of the file at |base_path|. When |map_all_requests_to_base_path|
|
| // is false, |base_path| is the file path leading to the root of the directory
|
| // to use as the root of the HTTP server.
|
| - MockJobInterceptor(const base::FilePath& base_path,
|
| - bool map_all_requests_to_base_path)
|
| + MockJobInterceptor(
|
| + const base::FilePath& base_path,
|
| + bool map_all_requests_to_base_path,
|
| + const scoped_refptr<base::SequencedWorkerPool>& worker_pool)
|
| : base_path_(base_path),
|
| - map_all_requests_to_base_path_(map_all_requests_to_base_path) {}
|
| + map_all_requests_to_base_path_(map_all_requests_to_base_path),
|
| + worker_pool_(worker_pool) {}
|
| virtual ~MockJobInterceptor() {}
|
|
|
| // net::URLRequestJobFactory::ProtocolHandler implementation
|
| virtual net::URLRequestJob* MaybeInterceptRequest(
|
| net::URLRequest* request,
|
| net::NetworkDelegate* network_delegate) const OVERRIDE {
|
| - return new URLRequestMockHTTPJob(request, network_delegate,
|
| - map_all_requests_to_base_path_ ? base_path_ : GetOnDiskPath(request));
|
| + return new URLRequestMockHTTPJob(
|
| + request,
|
| + network_delegate,
|
| + map_all_requests_to_base_path_ ? base_path_ : GetOnDiskPath(request),
|
| + worker_pool_->GetTaskRunnerWithShutdownBehavior(
|
| + base::SequencedWorkerPool::SKIP_ON_SHUTDOWN));
|
| }
|
|
|
| private:
|
| @@ -60,27 +66,45 @@ class MockJobInterceptor : public net::URLRequestInterceptor {
|
|
|
| const base::FilePath base_path_;
|
| const bool map_all_requests_to_base_path_;
|
| + const scoped_refptr<base::SequencedWorkerPool> worker_pool_;
|
|
|
| DISALLOW_COPY_AND_ASSIGN(MockJobInterceptor);
|
| };
|
|
|
| +std::string DoFileIO(const base::FilePath& file_path) {
|
| + base::FilePath header_file =
|
| + base::FilePath(file_path.value() + kMockHeaderFileSuffix);
|
| +
|
| + if (!base::PathExists(header_file)) {
|
| + // If there is no mock-http-headers file, fake a 200 OK.
|
| + return "HTTP/1.0 200 OK\n";
|
| + }
|
| +
|
| + std::string raw_headers;
|
| + base::ReadFileToString(header_file, &raw_headers);
|
| + return raw_headers;
|
| +}
|
| +
|
| } // namespace
|
|
|
| // static
|
| -void URLRequestMockHTTPJob::AddUrlHandler(const base::FilePath& base_path) {
|
| +void URLRequestMockHTTPJob::AddUrlHandler(
|
| + const base::FilePath& base_path,
|
| + const scoped_refptr<base::SequencedWorkerPool>& worker_pool) {
|
| // Add kMockHostname to net::URLRequestFilter.
|
| net::URLRequestFilter* filter = net::URLRequestFilter::GetInstance();
|
| filter->AddHostnameInterceptor(
|
| - "http", kMockHostname, CreateInterceptor(base_path));
|
| + "http", kMockHostname, CreateInterceptor(base_path, worker_pool));
|
| }
|
|
|
| // static
|
| void URLRequestMockHTTPJob::AddHostnameToFileHandler(
|
| const std::string& hostname,
|
| - const base::FilePath& file) {
|
| + const base::FilePath& file,
|
| + const scoped_refptr<base::SequencedWorkerPool>& worker_pool) {
|
| net::URLRequestFilter* filter = net::URLRequestFilter::GetInstance();
|
| filter->AddHostnameInterceptor(
|
| - "http", hostname, CreateInterceptorForSingleFile(file));
|
| + "http", hostname, CreateInterceptorForSingleFile(file, worker_pool));
|
| }
|
|
|
| // static
|
| @@ -95,38 +119,34 @@ GURL URLRequestMockHTTPJob::GetMockUrl(const base::FilePath& path) {
|
| }
|
|
|
| // static
|
| -GURL URLRequestMockHTTPJob::GetMockViewSourceUrl(const base::FilePath& path) {
|
| - std::string url = kViewSourceScheme;
|
| - url.append(":");
|
| - url.append(GetMockUrl(path).spec());
|
| - return GURL(url);
|
| -}
|
| -
|
| -// static
|
| -scoped_ptr<net::URLRequestInterceptor>
|
| -URLRequestMockHTTPJob::CreateInterceptor(const base::FilePath& base_path) {
|
| +scoped_ptr<net::URLRequestInterceptor> URLRequestMockHTTPJob::CreateInterceptor(
|
| + const base::FilePath& base_path,
|
| + const scoped_refptr<base::SequencedWorkerPool>& worker_pool) {
|
| return scoped_ptr<net::URLRequestInterceptor>(
|
| - new MockJobInterceptor(base_path, false));
|
| + new MockJobInterceptor(base_path, false, worker_pool));
|
| }
|
|
|
| // static
|
| scoped_ptr<net::URLRequestInterceptor>
|
| URLRequestMockHTTPJob::CreateInterceptorForSingleFile(
|
| - const base::FilePath& file) {
|
| + const base::FilePath& file,
|
| + const scoped_refptr<base::SequencedWorkerPool>& worker_pool) {
|
| return scoped_ptr<net::URLRequestInterceptor>(
|
| - new MockJobInterceptor(file, true));
|
| + new MockJobInterceptor(file, true, worker_pool));
|
| }
|
|
|
| URLRequestMockHTTPJob::URLRequestMockHTTPJob(
|
| - net::URLRequest* request, net::NetworkDelegate* network_delegate,
|
| - const base::FilePath& file_path)
|
| - : net::URLRequestFileJob(
|
| - request, network_delegate, file_path,
|
| - content::BrowserThread::GetBlockingPool()->
|
| - GetTaskRunnerWithShutdownBehavior(
|
| - base::SequencedWorkerPool::SKIP_ON_SHUTDOWN)) {}
|
| + net::URLRequest* request,
|
| + net::NetworkDelegate* network_delegate,
|
| + const base::FilePath& file_path,
|
| + const scoped_refptr<base::TaskRunner>& task_runner)
|
| + : net::URLRequestFileJob(request, network_delegate, file_path, task_runner),
|
| + task_runner_(task_runner),
|
| + weak_ptr_factory_(this) {
|
| +}
|
|
|
| -URLRequestMockHTTPJob::~URLRequestMockHTTPJob() { }
|
| +URLRequestMockHTTPJob::~URLRequestMockHTTPJob() {
|
| +}
|
|
|
| // Public virtual version.
|
| void URLRequestMockHTTPJob::GetResponseInfo(net::HttpResponseInfo* info) {
|
| @@ -141,29 +161,29 @@ bool URLRequestMockHTTPJob::IsRedirectResponse(GURL* location,
|
| return net::URLRequestJob::IsRedirectResponse(location, http_status_code);
|
| }
|
|
|
| -// Private const version.
|
| -void URLRequestMockHTTPJob::GetResponseInfoConst(
|
| - net::HttpResponseInfo* info) const {
|
| - // We have to load our headers from disk, but we only use this class
|
| - // from tests, so allow these IO operations to happen on any thread.
|
| - base::ThreadRestrictions::ScopedAllowIO allow_io;
|
| -
|
| - base::FilePath header_file =
|
| - base::FilePath(file_path_.value() + kMockHeaderFileSuffix);
|
| - std::string raw_headers;
|
| - if (!base::PathExists(header_file)) {
|
| - // If there is no mock-http-headers file, fake a 200 OK.
|
| - raw_headers = "HTTP/1.0 200 OK\n";
|
| - } else {
|
| - if (!base::ReadFileToString(header_file, &raw_headers))
|
| - return;
|
| - }
|
| +// Public virtual version.
|
| +void URLRequestMockHTTPJob::Start() {
|
| + base::PostTaskAndReplyWithResult(
|
| + task_runner_.get(),
|
| + FROM_HERE,
|
| + base::Bind(&DoFileIO, file_path_),
|
| + base::Bind(&URLRequestMockHTTPJob::GetRawHeaders,
|
| + weak_ptr_factory_.GetWeakPtr()));
|
| +}
|
|
|
| +void URLRequestMockHTTPJob::GetRawHeaders(std::string raw_headers) {
|
| // Handle CRLF line-endings.
|
| ReplaceSubstringsAfterOffset(&raw_headers, 0, "\r\n", "\n");
|
| // ParseRawHeaders expects \0 to end each header line.
|
| ReplaceSubstringsAfterOffset(&raw_headers, 0, "\n", std::string("\0", 1));
|
| - info->headers = new net::HttpResponseHeaders(raw_headers);
|
| + raw_headers_ = raw_headers;
|
| + URLRequestFileJob::Start();
|
| +}
|
| +
|
| +// Private const version.
|
| +void URLRequestMockHTTPJob::GetResponseInfoConst(
|
| + net::HttpResponseInfo* info) const {
|
| + info->headers = new net::HttpResponseHeaders(raw_headers_);
|
| }
|
|
|
| bool URLRequestMockHTTPJob::GetMimeType(std::string* mime_type) const {
|
| @@ -187,4 +207,4 @@ bool URLRequestMockHTTPJob::GetCharset(std::string* charset) {
|
| return info.headers.get() && info.headers->GetCharset(charset);
|
| }
|
|
|
| -} // namespace content
|
| +} // namespace net
|
|
|