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

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: Fixed a few compile errors 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 {
mmenke 2014/09/04 18:05:12 nit: Remove all the "net::", now that we're in ne
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(const base::FilePath& base_path,
35 bool map_all_requests_to_base_path) 33 bool map_all_requests_to_base_path,
34 base::SequencedWorkerPool* worker_pool)
36 : base_path_(base_path), 35 : base_path_(base_path),
37 map_all_requests_to_base_path_(map_all_requests_to_base_path) {} 36 map_all_requests_to_base_path_(map_all_requests_to_base_path),
37 worker_pool_(worker_pool) {}
38 virtual ~MockJobInterceptor() {} 38 virtual ~MockJobInterceptor() {}
39 39
40 // net::URLRequestJobFactory::ProtocolHandler implementation 40 // net::URLRequestJobFactory::ProtocolHandler implementation
41 virtual net::URLRequestJob* MaybeInterceptRequest( 41 virtual net::URLRequestJob* MaybeInterceptRequest(
42 net::URLRequest* request, 42 net::URLRequest* request,
43 net::NetworkDelegate* network_delegate) const OVERRIDE { 43 net::NetworkDelegate* network_delegate) const OVERRIDE {
44 return new URLRequestMockHTTPJob(request, network_delegate, 44 return new URLRequestMockHTTPJob(request, network_delegate,
45 map_all_requests_to_base_path_ ? base_path_ : GetOnDiskPath(request)); 45 map_all_requests_to_base_path_ ? base_path_ : GetOnDiskPath(request),
46 worker_pool_);
46 } 47 }
47 48
48 private: 49 private:
49 base::FilePath GetOnDiskPath(net::URLRequest* request) const { 50 base::FilePath GetOnDiskPath(net::URLRequest* request) const {
50 // Conceptually we just want to "return base_path_ + request->url().path()". 51 // 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). 52 // 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 53 // 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. 54 // path to that, and convert the final URL back to a FilePath.
54 GURL file_url(net::FilePathToFileURL(base_path_)); 55 GURL file_url(net::FilePathToFileURL(base_path_));
55 std::string url = file_url.spec() + request->url().path(); 56 std::string url = file_url.spec() + request->url().path();
56 base::FilePath file_path; 57 base::FilePath file_path;
57 net::FileURLToFilePath(GURL(url), &file_path); 58 net::FileURLToFilePath(GURL(url), &file_path);
58 return file_path; 59 return file_path;
59 } 60 }
60 61
61 const base::FilePath base_path_; 62 const base::FilePath base_path_;
62 const bool map_all_requests_to_base_path_; 63 const bool map_all_requests_to_base_path_;
64 base::SequencedWorkerPool* worker_pool_;
63 65
64 DISALLOW_COPY_AND_ASSIGN(MockJobInterceptor); 66 DISALLOW_COPY_AND_ASSIGN(MockJobInterceptor);
65 }; 67 };
66 68
67 } // namespace 69 } // namespace
68 70
69 // static 71 // static
70 void URLRequestMockHTTPJob::AddUrlHandler(const base::FilePath& base_path) { 72 void URLRequestMockHTTPJob::AddUrlHandler(
73 const base::FilePath& base_path,
74 base::SequencedWorkerPool* worker_pool) {
71 // Add kMockHostname to net::URLRequestFilter. 75 // Add kMockHostname to net::URLRequestFilter.
72 net::URLRequestFilter* filter = net::URLRequestFilter::GetInstance(); 76 net::URLRequestFilter* filter = net::URLRequestFilter::GetInstance();
73 filter->AddHostnameInterceptor( 77 filter->AddHostnameInterceptor(
74 "http", kMockHostname, CreateInterceptor(base_path)); 78 "http", kMockHostname, CreateInterceptor(base_path, worker_pool));
75 } 79 }
76 80
77 // static 81 // static
78 void URLRequestMockHTTPJob::AddHostnameToFileHandler( 82 void URLRequestMockHTTPJob::AddHostnameToFileHandler(
79 const std::string& hostname, 83 const std::string& hostname,
80 const base::FilePath& file) { 84 const base::FilePath& file,
85 base::SequencedWorkerPool* worker_pool) {
81 net::URLRequestFilter* filter = net::URLRequestFilter::GetInstance(); 86 net::URLRequestFilter* filter = net::URLRequestFilter::GetInstance();
82 filter->AddHostnameInterceptor( 87 filter->AddHostnameInterceptor(
83 "http", hostname, CreateInterceptorForSingleFile(file)); 88 "http", hostname,
89 CreateInterceptorForSingleFile(file, worker_pool));
84 } 90 }
85 91
86 // static 92 // static
87 GURL URLRequestMockHTTPJob::GetMockUrl(const base::FilePath& path) { 93 GURL URLRequestMockHTTPJob::GetMockUrl(const base::FilePath& path) {
88 std::string url = "http://"; 94 std::string url = "http://";
89 url.append(kMockHostname); 95 url.append(kMockHostname);
90 url.append("/"); 96 url.append("/");
91 std::string path_str = path.MaybeAsASCII(); 97 std::string path_str = path.MaybeAsASCII();
92 DCHECK(!path_str.empty()); // We only expect ASCII paths in tests. 98 DCHECK(!path_str.empty()); // We only expect ASCII paths in tests.
93 url.append(path_str); 99 url.append(path_str);
94 return GURL(url); 100 return GURL(url);
95 } 101 }
96 102
97 // static 103 // static
98 GURL URLRequestMockHTTPJob::GetMockViewSourceUrl(const base::FilePath& path) { 104 GURL URLRequestMockHTTPJob::GetMockViewSourceUrl(const base::FilePath& path) {
99 std::string url = kViewSourceScheme; 105 std::string url = "view-source";
mmenke 2014/09/04 18:05:12 This is a layering violation. net shouldn't know
xunjieli 2014/09/04 18:44:25 Done. Thanks! That was a bad thing to do. I was pl
100 url.append(":"); 106 url.append(":");
101 url.append(GetMockUrl(path).spec()); 107 url.append(GetMockUrl(path).spec());
102 return GURL(url); 108 return GURL(url);
103 } 109 }
104 110
105 // static 111 // static
106 scoped_ptr<net::URLRequestInterceptor> 112 scoped_ptr<net::URLRequestInterceptor>
107 URLRequestMockHTTPJob::CreateInterceptor(const base::FilePath& base_path) { 113 URLRequestMockHTTPJob::CreateInterceptor(
114 const base::FilePath& base_path,
115 base::SequencedWorkerPool* worker_pool) {
108 return scoped_ptr<net::URLRequestInterceptor>( 116 return scoped_ptr<net::URLRequestInterceptor>(
109 new MockJobInterceptor(base_path, false)); 117 new MockJobInterceptor(base_path, false, worker_pool));
110 } 118 }
111 119
112 // static 120 // static
113 scoped_ptr<net::URLRequestInterceptor> 121 scoped_ptr<net::URLRequestInterceptor>
114 URLRequestMockHTTPJob::CreateInterceptorForSingleFile( 122 URLRequestMockHTTPJob::CreateInterceptorForSingleFile(
115 const base::FilePath& file) { 123 const base::FilePath& file,
124 base::SequencedWorkerPool* worker_pool) {
116 return scoped_ptr<net::URLRequestInterceptor>( 125 return scoped_ptr<net::URLRequestInterceptor>(
117 new MockJobInterceptor(file, true)); 126 new MockJobInterceptor(file, true, worker_pool));
118 } 127 }
119 128
120 URLRequestMockHTTPJob::URLRequestMockHTTPJob( 129 URLRequestMockHTTPJob::URLRequestMockHTTPJob(
121 net::URLRequest* request, net::NetworkDelegate* network_delegate, 130 net::URLRequest* request, net::NetworkDelegate* network_delegate,
122 const base::FilePath& file_path) 131 const base::FilePath& file_path,
132 base::SequencedWorkerPool* worker_pool)
123 : net::URLRequestFileJob( 133 : net::URLRequestFileJob(
124 request, network_delegate, file_path, 134 request, network_delegate, file_path,
125 content::BrowserThread::GetBlockingPool()-> 135 worker_pool->GetTaskRunnerWithShutdownBehavior(
126 GetTaskRunnerWithShutdownBehavior( 136 base::SequencedWorkerPool::SKIP_ON_SHUTDOWN)) {}
127 base::SequencedWorkerPool::SKIP_ON_SHUTDOWN)) {}
128 137
129 URLRequestMockHTTPJob::~URLRequestMockHTTPJob() { } 138 URLRequestMockHTTPJob::~URLRequestMockHTTPJob() { }
130 139
131 // Public virtual version. 140 // Public virtual version.
132 void URLRequestMockHTTPJob::GetResponseInfo(net::HttpResponseInfo* info) { 141 void URLRequestMockHTTPJob::GetResponseInfo(net::HttpResponseInfo* info) {
133 // Forward to private const version. 142 // Forward to private const version.
134 GetResponseInfoConst(info); 143 GetResponseInfoConst(info);
135 } 144 }
136 145
137 bool URLRequestMockHTTPJob::IsRedirectResponse(GURL* location, 146 bool URLRequestMockHTTPJob::IsRedirectResponse(GURL* location,
138 int* http_status_code) { 147 int* http_status_code) {
139 // Override the net::URLRequestFileJob implementation to invoke the default 148 // Override the net::URLRequestFileJob implementation to invoke the default
140 // one based on HttpResponseInfo. 149 // one based on HttpResponseInfo.
141 return net::URLRequestJob::IsRedirectResponse(location, http_status_code); 150 return net::URLRequestJob::IsRedirectResponse(location, http_status_code);
142 } 151 }
143 152
144 // Private const version. 153 // Private const version.
145 void URLRequestMockHTTPJob::GetResponseInfoConst( 154 void URLRequestMockHTTPJob::GetResponseInfoConst(
146 net::HttpResponseInfo* info) const { 155 net::HttpResponseInfo* info) const {
147 // We have to load our headers from disk, but we only use this class 156 // 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. 157 // from tests, so allow these IO operations to happen on any thread.
149 base::ThreadRestrictions::ScopedAllowIO allow_io; 158 base::ThreadRestrictions::SetIOAllowed(true);
150 159
151 base::FilePath header_file = 160 base::FilePath header_file =
152 base::FilePath(file_path_.value() + kMockHeaderFileSuffix); 161 base::FilePath(file_path_.value() + kMockHeaderFileSuffix);
153 std::string raw_headers; 162 std::string raw_headers;
154 if (!base::PathExists(header_file)) { 163 if (!base::PathExists(header_file)) {
155 // If there is no mock-http-headers file, fake a 200 OK. 164 // If there is no mock-http-headers file, fake a 200 OK.
156 raw_headers = "HTTP/1.0 200 OK\n"; 165 raw_headers = "HTTP/1.0 200 OK\n";
157 } else { 166 } else {
158 if (!base::ReadFileToString(header_file, &raw_headers)) 167 if (!base::ReadFileToString(header_file, &raw_headers))
159 return; 168 return;
(...skipping 20 matching lines...) Expand all
180 return info.headers->response_code(); 189 return info.headers->response_code();
181 return net::URLRequestJob::GetResponseCode(); 190 return net::URLRequestJob::GetResponseCode();
182 } 191 }
183 192
184 bool URLRequestMockHTTPJob::GetCharset(std::string* charset) { 193 bool URLRequestMockHTTPJob::GetCharset(std::string* charset) {
185 net::HttpResponseInfo info; 194 net::HttpResponseInfo info;
186 GetResponseInfo(&info); 195 GetResponseInfo(&info);
187 return info.headers.get() && info.headers->GetCharset(charset); 196 return info.headers.get() && info.headers->GetCharset(charset);
188 } 197 }
189 198
190 } // namespace content 199 } // namespace net
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698