Chromium Code Reviews| OLD | NEW |
|---|---|
| 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 "net/test/url_request/url_request_mock_http_job.h" | 5 #include "net/test/url_request/url_request_mock_http_job.h" |
| 6 | 6 |
| 7 #include "base/files/file_util.h" | 7 #include "base/files/file_util.h" |
| 8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
| 9 #include "base/strings/string_number_conversions.h" | |
| 9 #include "base/strings/string_util.h" | 10 #include "base/strings/string_util.h" |
| 10 #include "base/strings/utf_string_conversions.h" | 11 #include "base/strings/utf_string_conversions.h" |
| 11 #include "base/task_runner_util.h" | 12 #include "base/task_runner_util.h" |
| 12 #include "base/threading/sequenced_worker_pool.h" | 13 #include "base/threading/sequenced_worker_pool.h" |
| 13 #include "base/threading/thread_restrictions.h" | 14 #include "base/threading/thread_restrictions.h" |
| 14 #include "net/base/filename_util.h" | 15 #include "net/base/filename_util.h" |
| 16 #include "net/base/net_errors.h" | |
| 17 #include "net/base/url_util.h" | |
| 15 #include "net/http/http_response_headers.h" | 18 #include "net/http/http_response_headers.h" |
| 16 #include "net/url_request/url_request_filter.h" | 19 #include "net/url_request/url_request_filter.h" |
| 17 #include "net/url_request/url_request_interceptor.h" | 20 #include "net/url_request/url_request_interceptor.h" |
| 18 | 21 |
| 19 const char kMockHostname[] = "mock.http"; | 22 const char kMockHostname[] = "mock.http"; |
| 20 const base::FilePath::CharType kMockHeaderFileSuffix[] = | 23 const base::FilePath::CharType kMockHeaderFileSuffix[] = |
| 21 FILE_PATH_LITERAL(".mock-http-headers"); | 24 FILE_PATH_LITERAL(".mock-http-headers"); |
| 22 | 25 |
| 26 const char kMockErrorStepRedirect[] = "redirect"; | |
| 27 const char kMockErrorStepHeaders[] = "headers"; | |
| 28 const char kMockErrorStepData[] = "data"; | |
| 29 | |
| 23 namespace net { | 30 namespace net { |
| 24 | 31 |
| 25 namespace { | 32 namespace { |
| 26 | 33 |
| 27 class MockJobInterceptor : public net::URLRequestInterceptor { | 34 class MockJobInterceptor : public net::URLRequestInterceptor { |
| 28 public: | 35 public: |
| 29 // When |map_all_requests_to_base_path| is true, all request should return the | 36 // When |map_all_requests_to_base_path| is true, all request should return the |
| 30 // contents of the file at |base_path|. When |map_all_requests_to_base_path| | 37 // contents of the file at |base_path|. When |map_all_requests_to_base_path| |
| 31 // is false, |base_path| is the file path leading to the root of the directory | 38 // is false, |base_path| is the file path leading to the root of the directory |
| 32 // to use as the root of the HTTP server. | 39 // to use as the root of the HTTP server. |
| (...skipping 116 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 149 } | 156 } |
| 150 | 157 |
| 151 // Public virtual version. | 158 // Public virtual version. |
| 152 void URLRequestMockHTTPJob::GetResponseInfo(net::HttpResponseInfo* info) { | 159 void URLRequestMockHTTPJob::GetResponseInfo(net::HttpResponseInfo* info) { |
| 153 // Forward to private const version. | 160 // Forward to private const version. |
| 154 GetResponseInfoConst(info); | 161 GetResponseInfoConst(info); |
| 155 } | 162 } |
| 156 | 163 |
| 157 bool URLRequestMockHTTPJob::IsRedirectResponse(GURL* location, | 164 bool URLRequestMockHTTPJob::IsRedirectResponse(GURL* location, |
| 158 int* http_status_code) { | 165 int* http_status_code) { |
| 166 if (ReportMockErrorFromQuery(kMockErrorStepRedirect)) | |
| 167 return false; | |
|
mmenke
2014/10/16 16:39:34
Hrm...I don't think this makes any sense. For err
mef
2014/10/16 21:36:50
Done.
| |
| 159 // Override the net::URLRequestFileJob implementation to invoke the default | 168 // Override the net::URLRequestFileJob implementation to invoke the default |
| 160 // one based on HttpResponseInfo. | 169 // one based on HttpResponseInfo. |
| 161 return net::URLRequestJob::IsRedirectResponse(location, http_status_code); | 170 return net::URLRequestJob::IsRedirectResponse(location, http_status_code); |
| 162 } | 171 } |
| 163 | 172 |
| 164 // Public virtual version. | 173 // Public virtual version. |
| 165 void URLRequestMockHTTPJob::Start() { | 174 void URLRequestMockHTTPJob::Start() { |
| 166 base::PostTaskAndReplyWithResult( | 175 base::PostTaskAndReplyWithResult( |
| 167 task_runner_.get(), | 176 task_runner_.get(), |
| 168 FROM_HERE, | 177 FROM_HERE, |
| 169 base::Bind(&DoFileIO, file_path_), | 178 base::Bind(&DoFileIO, file_path_), |
| 170 base::Bind(&URLRequestMockHTTPJob::GetRawHeaders, | 179 base::Bind(&URLRequestMockHTTPJob::GetRawHeaders, |
| 171 weak_ptr_factory_.GetWeakPtr())); | 180 weak_ptr_factory_.GetWeakPtr())); |
| 172 } | 181 } |
| 173 | 182 |
| 183 // Public virtual version. | |
| 184 bool URLRequestMockHTTPJob::ReadRawData(IOBuffer* buf, | |
| 185 int buf_size, | |
| 186 int* bytes_read) { | |
| 187 if (ReportMockErrorFromQuery(kMockErrorStepData)) | |
| 188 return false; | |
| 189 return URLRequestFileJob::ReadRawData(buf, buf_size, bytes_read); | |
| 190 } | |
| 191 | |
| 174 void URLRequestMockHTTPJob::GetRawHeaders(std::string raw_headers) { | 192 void URLRequestMockHTTPJob::GetRawHeaders(std::string raw_headers) { |
|
mmenke
2014/10/16 16:39:34
While you're here, think this should probably be r
mef
2014/10/16 21:36:50
Done.
| |
| 175 // Handle CRLF line-endings. | 193 // Handle CRLF line-endings. |
| 176 ReplaceSubstringsAfterOffset(&raw_headers, 0, "\r\n", "\n"); | 194 ReplaceSubstringsAfterOffset(&raw_headers, 0, "\r\n", "\n"); |
| 177 // ParseRawHeaders expects \0 to end each header line. | 195 // ParseRawHeaders expects \0 to end each header line. |
| 178 ReplaceSubstringsAfterOffset(&raw_headers, 0, "\n", std::string("\0", 1)); | 196 ReplaceSubstringsAfterOffset(&raw_headers, 0, "\n", std::string("\0", 1)); |
| 179 raw_headers_ = raw_headers; | 197 raw_headers_ = raw_headers; |
| 198 if (ReportMockErrorFromQuery(kMockErrorStepHeaders)) | |
| 199 return; | |
|
mmenke
2014/10/16 16:39:34
Hrm...wonder if we should set the headers or not i
mef
2014/10/16 21:36:50
Done.
| |
| 180 URLRequestFileJob::Start(); | 200 URLRequestFileJob::Start(); |
| 181 } | 201 } |
| 182 | 202 |
| 203 bool URLRequestMockHTTPJob::ReportMockErrorFromQuery(const std::string& step) { | |
| 204 std::string value; | |
| 205 if (!GetValueForKeyInQuery(request_->url(), step, &value)) | |
| 206 return false; | |
| 207 | |
| 208 int net_error; | |
| 209 if (!base::StringToInt(value, &net_error)) | |
| 210 return false; | |
| 211 | |
| 212 if (net_error == net::ERR_IO_PENDING) { | |
| 213 DLOG(ERROR) << "Report Mock ERR_IO_PENDING for step " << step; | |
| 214 SetStatus(net::URLRequestStatus(net::URLRequestStatus::IO_PENDING, 0)); | |
| 215 } else { | |
| 216 DLOG(ERROR) << "Report Mock Error " << net_error << " for step " << step; | |
| 217 NotifyDone(net::URLRequestStatus(net::URLRequestStatus::FAILED, net_error)); | |
| 218 } | |
| 219 return true; | |
| 220 } | |
| 221 | |
| 183 // Private const version. | 222 // Private const version. |
| 184 void URLRequestMockHTTPJob::GetResponseInfoConst( | 223 void URLRequestMockHTTPJob::GetResponseInfoConst( |
| 185 net::HttpResponseInfo* info) const { | 224 net::HttpResponseInfo* info) const { |
| 186 info->headers = new net::HttpResponseHeaders(raw_headers_); | 225 info->headers = new net::HttpResponseHeaders(raw_headers_); |
| 187 } | 226 } |
| 188 | 227 |
| 189 bool URLRequestMockHTTPJob::GetMimeType(std::string* mime_type) const { | 228 bool URLRequestMockHTTPJob::GetMimeType(std::string* mime_type) const { |
| 190 net::HttpResponseInfo info; | 229 net::HttpResponseInfo info; |
| 191 GetResponseInfoConst(&info); | 230 GetResponseInfoConst(&info); |
| 192 return info.headers.get() && info.headers->GetMimeType(mime_type); | 231 return info.headers.get() && info.headers->GetMimeType(mime_type); |
| 193 } | 232 } |
| 194 | 233 |
| 195 int URLRequestMockHTTPJob::GetResponseCode() const { | 234 int URLRequestMockHTTPJob::GetResponseCode() const { |
| 196 net::HttpResponseInfo info; | 235 net::HttpResponseInfo info; |
| 197 GetResponseInfoConst(&info); | 236 GetResponseInfoConst(&info); |
| 198 // If we have headers, get the response code from them. | 237 // If we have headers, get the response code from them. |
| 199 if (info.headers.get()) | 238 if (info.headers.get()) |
| 200 return info.headers->response_code(); | 239 return info.headers->response_code(); |
| 201 return net::URLRequestJob::GetResponseCode(); | 240 return net::URLRequestJob::GetResponseCode(); |
| 202 } | 241 } |
| 203 | 242 |
| 204 bool URLRequestMockHTTPJob::GetCharset(std::string* charset) { | 243 bool URLRequestMockHTTPJob::GetCharset(std::string* charset) { |
| 205 net::HttpResponseInfo info; | 244 net::HttpResponseInfo info; |
| 206 GetResponseInfo(&info); | 245 GetResponseInfo(&info); |
| 207 return info.headers.get() && info.headers->GetCharset(charset); | 246 return info.headers.get() && info.headers->GetCharset(charset); |
| 208 } | 247 } |
| 209 | 248 |
| 210 } // namespace net | 249 } // namespace net |
| OLD | NEW |