Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(81)

Side by Side Diff: net/test/url_request/url_request_mock_http_job.cc

Issue 541743002: Move url_request_mock_http_job to net/test/url_request/ (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Addressed comments Created 6 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2012 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 "content/test/net/url_request_mock_http_job.h" 5 #include "net/test/url_request/url_request_mock_http_job.h"
6 6
7 #include "base/file_util.h" 7 #include "base/file_util.h"
8 #include "base/message_loop/message_loop.h" 8 #include "base/message_loop/message_loop.h"
9 #include "base/strings/string_util.h" 9 #include "base/strings/string_util.h"
10 #include "base/strings/utf_string_conversions.h" 10 #include "base/strings/utf_string_conversions.h"
11 #include "base/threading/sequenced_worker_pool.h" 11 #include "base/threading/sequenced_worker_pool.h"
12 #include "base/threading/thread_restrictions.h" 12 #include "base/threading/thread_restrictions.h"
13 #include "content/public/browser/browser_thread.h"
14 #include "content/public/common/url_constants.h"
15 #include "net/base/filename_util.h" 13 #include "net/base/filename_util.h"
16 #include "net/http/http_response_headers.h" 14 #include "net/http/http_response_headers.h"
17 #include "net/url_request/url_request_filter.h" 15 #include "net/url_request/url_request_filter.h"
18 #include "net/url_request/url_request_interceptor.h" 16 #include "net/url_request/url_request_interceptor.h"
19 17
20 const char kMockHostname[] = "mock.http"; 18 const char kMockHostname[] = "mock.http";
21 const base::FilePath::CharType kMockHeaderFileSuffix[] = 19 const base::FilePath::CharType kMockHeaderFileSuffix[] =
22 FILE_PATH_LITERAL(".mock-http-headers"); 20 FILE_PATH_LITERAL(".mock-http-headers");
23 21
24 namespace content { 22 namespace net {
25 23
26 namespace { 24 namespace {
27 25
28 class MockJobInterceptor : public net::URLRequestInterceptor { 26 class MockJobInterceptor : public net::URLRequestInterceptor {
29 public: 27 public:
30 // When |map_all_requests_to_base_path| is true, all request should return the 28 // When |map_all_requests_to_base_path| is true, all request should return the
31 // contents of the file at |base_path|. When |map_all_requests_to_base_path| 29 // contents of the file at |base_path|. When |map_all_requests_to_base_path|
32 // is false, |base_path| is the file path leading to the root of the directory 30 // is false, |base_path| is the file path leading to the root of the directory
33 // to use as the root of the HTTP server. 31 // to use as the root of the HTTP server.
34 MockJobInterceptor(const base::FilePath& base_path, 32 MockJobInterceptor(
35 bool map_all_requests_to_base_path) 33 const base::FilePath& base_path,
34 bool map_all_requests_to_base_path,
35 const scoped_refptr<base::SequencedWorkerPool>& worker_pool)
36 : base_path_(base_path), 36 : base_path_(base_path),
37 map_all_requests_to_base_path_(map_all_requests_to_base_path) {} 37 map_all_requests_to_base_path_(map_all_requests_to_base_path),
38 worker_pool_(worker_pool) {}
38 virtual ~MockJobInterceptor() {} 39 virtual ~MockJobInterceptor() {}
39 40
40 // net::URLRequestJobFactory::ProtocolHandler implementation 41 // net::URLRequestJobFactory::ProtocolHandler implementation
41 virtual net::URLRequestJob* MaybeInterceptRequest( 42 virtual net::URLRequestJob* MaybeInterceptRequest(
42 net::URLRequest* request, 43 net::URLRequest* request,
43 net::NetworkDelegate* network_delegate) const OVERRIDE { 44 net::NetworkDelegate* network_delegate) const OVERRIDE {
44 return new URLRequestMockHTTPJob(request, network_delegate, 45 return new URLRequestMockHTTPJob(request, network_delegate,
45 map_all_requests_to_base_path_ ? base_path_ : GetOnDiskPath(request)); 46 map_all_requests_to_base_path_ ? base_path_ : GetOnDiskPath(request),
47 worker_pool_);
46 } 48 }
47 49
48 private: 50 private:
49 base::FilePath GetOnDiskPath(net::URLRequest* request) const { 51 base::FilePath GetOnDiskPath(net::URLRequest* request) const {
50 // Conceptually we just want to "return base_path_ + request->url().path()". 52 // Conceptually we just want to "return base_path_ + request->url().path()".
51 // But path in the request URL is in URL space (i.e. %-encoded spaces). 53 // But path in the request URL is in URL space (i.e. %-encoded spaces).
52 // So first we convert base FilePath to a URL, then append the URL 54 // So first we convert base FilePath to a URL, then append the URL
53 // path to that, and convert the final URL back to a FilePath. 55 // path to that, and convert the final URL back to a FilePath.
54 GURL file_url(net::FilePathToFileURL(base_path_)); 56 GURL file_url(net::FilePathToFileURL(base_path_));
55 std::string url = file_url.spec() + request->url().path(); 57 std::string url = file_url.spec() + request->url().path();
56 base::FilePath file_path; 58 base::FilePath file_path;
57 net::FileURLToFilePath(GURL(url), &file_path); 59 net::FileURLToFilePath(GURL(url), &file_path);
58 return file_path; 60 return file_path;
59 } 61 }
60 62
61 const base::FilePath base_path_; 63 const base::FilePath base_path_;
62 const bool map_all_requests_to_base_path_; 64 const bool map_all_requests_to_base_path_;
65 const scoped_refptr<base::SequencedWorkerPool> worker_pool_;
63 66
64 DISALLOW_COPY_AND_ASSIGN(MockJobInterceptor); 67 DISALLOW_COPY_AND_ASSIGN(MockJobInterceptor);
65 }; 68 };
66 69
67 } // namespace 70 } // namespace
68 71
69 // static 72 // static
70 void URLRequestMockHTTPJob::AddUrlHandler(const base::FilePath& base_path) { 73 void URLRequestMockHTTPJob::AddUrlHandler(
74 const base::FilePath& base_path,
75 const scoped_refptr<base::SequencedWorkerPool>& worker_pool) {
71 // Add kMockHostname to net::URLRequestFilter. 76 // Add kMockHostname to net::URLRequestFilter.
72 net::URLRequestFilter* filter = net::URLRequestFilter::GetInstance(); 77 net::URLRequestFilter* filter = net::URLRequestFilter::GetInstance();
73 filter->AddHostnameInterceptor( 78 filter->AddHostnameInterceptor(
74 "http", kMockHostname, CreateInterceptor(base_path)); 79 "http", kMockHostname, CreateInterceptor(base_path, worker_pool));
75 } 80 }
76 81
77 // static 82 // static
78 void URLRequestMockHTTPJob::AddHostnameToFileHandler( 83 void URLRequestMockHTTPJob::AddHostnameToFileHandler(
79 const std::string& hostname, 84 const std::string& hostname,
80 const base::FilePath& file) { 85 const base::FilePath& file,
86 const scoped_refptr<base::SequencedWorkerPool>& worker_pool) {
81 net::URLRequestFilter* filter = net::URLRequestFilter::GetInstance(); 87 net::URLRequestFilter* filter = net::URLRequestFilter::GetInstance();
82 filter->AddHostnameInterceptor( 88 filter->AddHostnameInterceptor(
83 "http", hostname, CreateInterceptorForSingleFile(file)); 89 "http", hostname,
90 CreateInterceptorForSingleFile(file, worker_pool));
84 } 91 }
85 92
86 // static 93 // static
87 GURL URLRequestMockHTTPJob::GetMockUrl(const base::FilePath& path) { 94 GURL URLRequestMockHTTPJob::GetMockUrl(const base::FilePath& path) {
88 std::string url = "http://"; 95 std::string url = "http://";
89 url.append(kMockHostname); 96 url.append(kMockHostname);
90 url.append("/"); 97 url.append("/");
91 std::string path_str = path.MaybeAsASCII(); 98 std::string path_str = path.MaybeAsASCII();
92 DCHECK(!path_str.empty()); // We only expect ASCII paths in tests. 99 DCHECK(!path_str.empty()); // We only expect ASCII paths in tests.
93 url.append(path_str); 100 url.append(path_str);
94 return GURL(url); 101 return GURL(url);
95 } 102 }
96 103
97 // static 104 // static
98 GURL URLRequestMockHTTPJob::GetMockViewSourceUrl(const base::FilePath& path) {
99 std::string url = kViewSourceScheme;
100 url.append(":");
101 url.append(GetMockUrl(path).spec());
102 return GURL(url);
103 }
104
105 // static
106 scoped_ptr<net::URLRequestInterceptor> 105 scoped_ptr<net::URLRequestInterceptor>
107 URLRequestMockHTTPJob::CreateInterceptor(const base::FilePath& base_path) { 106 URLRequestMockHTTPJob::CreateInterceptor(
107 const base::FilePath& base_path,
108 const scoped_refptr<base::SequencedWorkerPool>& worker_pool) {
108 return scoped_ptr<net::URLRequestInterceptor>( 109 return scoped_ptr<net::URLRequestInterceptor>(
109 new MockJobInterceptor(base_path, false)); 110 new MockJobInterceptor(base_path, false, worker_pool));
110 } 111 }
111 112
112 // static 113 // static
113 scoped_ptr<net::URLRequestInterceptor> 114 scoped_ptr<net::URLRequestInterceptor>
114 URLRequestMockHTTPJob::CreateInterceptorForSingleFile( 115 URLRequestMockHTTPJob::CreateInterceptorForSingleFile(
115 const base::FilePath& file) { 116 const base::FilePath& file,
117 const scoped_refptr<base::SequencedWorkerPool>& worker_pool) {
116 return scoped_ptr<net::URLRequestInterceptor>( 118 return scoped_ptr<net::URLRequestInterceptor>(
117 new MockJobInterceptor(file, true)); 119 new MockJobInterceptor(file, true, worker_pool));
118 } 120 }
119 121
120 URLRequestMockHTTPJob::URLRequestMockHTTPJob( 122 URLRequestMockHTTPJob::URLRequestMockHTTPJob(
121 net::URLRequest* request, net::NetworkDelegate* network_delegate, 123 net::URLRequest* request, net::NetworkDelegate* network_delegate,
122 const base::FilePath& file_path) 124 const base::FilePath& file_path,
125 const scoped_refptr<base::SequencedWorkerPool>& worker_pool)
123 : net::URLRequestFileJob( 126 : net::URLRequestFileJob(
124 request, network_delegate, file_path, 127 request, network_delegate, file_path,
125 content::BrowserThread::GetBlockingPool()-> 128 worker_pool->GetTaskRunnerWithShutdownBehavior(
126 GetTaskRunnerWithShutdownBehavior( 129 base::SequencedWorkerPool::SKIP_ON_SHUTDOWN)) {}
127 base::SequencedWorkerPool::SKIP_ON_SHUTDOWN)) {}
128 130
129 URLRequestMockHTTPJob::~URLRequestMockHTTPJob() { } 131 URLRequestMockHTTPJob::~URLRequestMockHTTPJob() { }
130 132
131 // Public virtual version. 133 // Public virtual version.
132 void URLRequestMockHTTPJob::GetResponseInfo(net::HttpResponseInfo* info) { 134 void URLRequestMockHTTPJob::GetResponseInfo(net::HttpResponseInfo* info) {
133 // Forward to private const version. 135 // Forward to private const version.
134 GetResponseInfoConst(info); 136 GetResponseInfoConst(info);
135 } 137 }
136 138
137 bool URLRequestMockHTTPJob::IsRedirectResponse(GURL* location, 139 bool URLRequestMockHTTPJob::IsRedirectResponse(GURL* location,
138 int* http_status_code) { 140 int* http_status_code) {
139 // Override the net::URLRequestFileJob implementation to invoke the default 141 // Override the net::URLRequestFileJob implementation to invoke the default
140 // one based on HttpResponseInfo. 142 // one based on HttpResponseInfo.
141 return net::URLRequestJob::IsRedirectResponse(location, http_status_code); 143 return net::URLRequestJob::IsRedirectResponse(location, http_status_code);
142 } 144 }
143 145
144 // Private const version. 146 // Private const version.
145 void URLRequestMockHTTPJob::GetResponseInfoConst( 147 void URLRequestMockHTTPJob::GetResponseInfoConst(
146 net::HttpResponseInfo* info) const { 148 net::HttpResponseInfo* info) const {
147 // We have to load our headers from disk, but we only use this class 149 // We have to load our headers from disk, but we only use this class
148 // from tests, so allow these IO operations to happen on any thread. 150 // from tests, so allow these IO operations to happen on any thread.
149 base::ThreadRestrictions::ScopedAllowIO allow_io; 151 base::ThreadRestrictions::SetIOAllowed(true);
mmenke 2014/09/05 15:54:09 This has me rather concerned - this basically allo
xunjieli 2014/09/05 19:26:15 Done. If it's a problem, we should definitely fix
150 152
151 base::FilePath header_file = 153 base::FilePath header_file =
152 base::FilePath(file_path_.value() + kMockHeaderFileSuffix); 154 base::FilePath(file_path_.value() + kMockHeaderFileSuffix);
153 std::string raw_headers; 155 std::string raw_headers;
154 if (!base::PathExists(header_file)) { 156 if (!base::PathExists(header_file)) {
155 // If there is no mock-http-headers file, fake a 200 OK. 157 // If there is no mock-http-headers file, fake a 200 OK.
156 raw_headers = "HTTP/1.0 200 OK\n"; 158 raw_headers = "HTTP/1.0 200 OK\n";
157 } else { 159 } else {
158 if (!base::ReadFileToString(header_file, &raw_headers)) 160 if (!base::ReadFileToString(header_file, &raw_headers))
159 return; 161 return;
(...skipping 20 matching lines...) Expand all
180 return info.headers->response_code(); 182 return info.headers->response_code();
181 return net::URLRequestJob::GetResponseCode(); 183 return net::URLRequestJob::GetResponseCode();
182 } 184 }
183 185
184 bool URLRequestMockHTTPJob::GetCharset(std::string* charset) { 186 bool URLRequestMockHTTPJob::GetCharset(std::string* charset) {
185 net::HttpResponseInfo info; 187 net::HttpResponseInfo info;
186 GetResponseInfo(&info); 188 GetResponseInfo(&info);
187 return info.headers.get() && info.headers->GetCharset(charset); 189 return info.headers.get() && info.headers->GetCharset(charset);
188 } 190 }
189 191
190 } // namespace content 192 } // namespace net
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698