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/url_request/test_url_fetcher_factory.h" | 5 #include "net/url_request/test_url_fetcher_factory.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/compiler_specific.h" | 10 #include "base/compiler_specific.h" |
(...skipping 265 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
276 } | 276 } |
277 | 277 |
278 void TestURLFetcherFactory::SetDelegateForTests( | 278 void TestURLFetcherFactory::SetDelegateForTests( |
279 TestURLFetcherDelegateForTests* delegate_for_tests) { | 279 TestURLFetcherDelegateForTests* delegate_for_tests) { |
280 delegate_for_tests_ = delegate_for_tests; | 280 delegate_for_tests_ = delegate_for_tests; |
281 } | 281 } |
282 | 282 |
283 FakeURLFetcher::FakeURLFetcher(const GURL& url, | 283 FakeURLFetcher::FakeURLFetcher(const GURL& url, |
284 URLFetcherDelegate* d, | 284 URLFetcherDelegate* d, |
285 const std::string& response_data, | 285 const std::string& response_data, |
286 HttpStatusCode response_code) | 286 URLRequestStatus request_status) |
287 : TestURLFetcher(0, url, d), | 287 : TestURLFetcher(0, url, d), |
288 weak_factory_(this) { | 288 weak_factory_(this) { |
289 set_status(URLRequestStatus( | 289 set_status(request_status); |
290 // Status is FAILED for HTTP/5xx server errors, and SUCCESS otherwise. | 290 set_response_code(HttpStatusCode(request_status.error())); |
291 response_code >= HTTP_INTERNAL_SERVER_ERROR ? URLRequestStatus::FAILED : | |
292 URLRequestStatus::SUCCESS, | |
293 0)); | |
294 set_response_code(response_code); | |
295 SetResponseString(response_data); | 291 SetResponseString(response_data); |
296 } | 292 } |
297 | 293 |
298 FakeURLFetcher::~FakeURLFetcher() {} | 294 FakeURLFetcher::~FakeURLFetcher() {} |
299 | 295 |
300 void FakeURLFetcher::Start() { | 296 void FakeURLFetcher::Start() { |
301 base::MessageLoop::current()->PostTask( | 297 base::MessageLoop::current()->PostTask( |
302 FROM_HERE, | 298 FROM_HERE, |
303 base::Bind(&FakeURLFetcher::RunDelegate, weak_factory_.GetWeakPtr())); | 299 base::Bind(&FakeURLFetcher::RunDelegate, weak_factory_.GetWeakPtr())); |
304 } | 300 } |
(...skipping 18 matching lines...) Expand all Loading... | |
323 const FakeURLFetcherCreator& creator) | 319 const FakeURLFetcherCreator& creator) |
324 : ScopedURLFetcherFactory(this), | 320 : ScopedURLFetcherFactory(this), |
325 creator_(creator), | 321 creator_(creator), |
326 default_factory_(default_factory) { | 322 default_factory_(default_factory) { |
327 } | 323 } |
328 | 324 |
329 scoped_ptr<FakeURLFetcher> FakeURLFetcherFactory::DefaultFakeURLFetcherCreator( | 325 scoped_ptr<FakeURLFetcher> FakeURLFetcherFactory::DefaultFakeURLFetcherCreator( |
330 const GURL& url, | 326 const GURL& url, |
331 URLFetcherDelegate* delegate, | 327 URLFetcherDelegate* delegate, |
332 const std::string& response_data, | 328 const std::string& response_data, |
333 HttpStatusCode response_code) { | 329 URLRequestStatus request_status) { |
334 return scoped_ptr<FakeURLFetcher>(new FakeURLFetcher(url, | 330 return scoped_ptr<FakeURLFetcher>(new FakeURLFetcher(url, |
335 delegate, | 331 delegate, |
336 response_data, | 332 response_data, |
337 response_code)); | 333 request_status)); |
338 } | 334 } |
339 | 335 |
340 FakeURLFetcherFactory::~FakeURLFetcherFactory() {} | 336 FakeURLFetcherFactory::~FakeURLFetcherFactory() {} |
341 | 337 |
342 URLFetcher* FakeURLFetcherFactory::CreateURLFetcher( | 338 URLFetcher* FakeURLFetcherFactory::CreateURLFetcher( |
343 int id, | 339 int id, |
344 const GURL& url, | 340 const GURL& url, |
345 URLFetcher::RequestType request_type, | 341 URLFetcher::RequestType request_type, |
346 URLFetcherDelegate* d) { | 342 URLFetcherDelegate* d) { |
347 FakeResponseMap::const_iterator it = fake_responses_.find(url); | 343 FakeResponseMap::const_iterator it = fake_responses_.find(url); |
348 if (it == fake_responses_.end()) { | 344 if (it == fake_responses_.end()) { |
349 if (default_factory_ == NULL) { | 345 if (default_factory_ == NULL) { |
350 // If we don't have a baked response for that URL we return NULL. | 346 // If we don't have a baked response for that URL we return NULL. |
351 DLOG(ERROR) << "No baked response for URL: " << url.spec(); | 347 DLOG(ERROR) << "No baked response for URL: " << url.spec(); |
352 return NULL; | 348 return NULL; |
353 } else { | 349 } else { |
354 return default_factory_->CreateURLFetcher(id, url, request_type, d); | 350 return default_factory_->CreateURLFetcher(id, url, request_type, d); |
355 } | 351 } |
356 } | 352 } |
357 | 353 |
358 scoped_ptr<FakeURLFetcher> fake_fetcher = | 354 scoped_ptr<FakeURLFetcher> fake_fetcher = |
359 creator_.Run(url, d, it->second.first, it->second.second); | 355 creator_.Run(url, d, it->second.first, it->second.second); |
360 // TODO: Make URLFetcherFactory::CreateURLFetcher return a scoped_ptr | 356 // TODO: Make URLFetcherFactory::CreateURLFetcher return a scoped_ptr |
361 return fake_fetcher.release(); | 357 return fake_fetcher.release(); |
362 } | 358 } |
363 | 359 |
364 void FakeURLFetcherFactory::SetFakeResponse( | 360 void FakeURLFetcherFactory::SetFakeResponse( |
365 const GURL& url, | 361 const GURL& url, |
366 const std::string& response_data, | 362 const std::string& response_data, |
367 HttpStatusCode response_code) { | 363 HttpStatusCode response_code, |
364 URLRequestStatus::Status status) { | |
368 // Overwrite existing URL if it already exists. | 365 // Overwrite existing URL if it already exists. |
369 fake_responses_[url] = std::make_pair(response_data, response_code); | 366 fake_responses_[url] = std::make_pair( |
367 response_data, URLRequestStatus(status, response_code)); | |
akalin
2013/11/05 22:03:09
is the 'error' field in URLRequestStatus supposed
Raghu Simha
2013/11/05 22:09:07
In that case, it probably makes sense to store the
| |
370 } | 368 } |
371 | 369 |
372 void FakeURLFetcherFactory::ClearFakeResponses() { | 370 void FakeURLFetcherFactory::ClearFakeResponses() { |
373 fake_responses_.clear(); | 371 fake_responses_.clear(); |
374 } | 372 } |
375 | 373 |
376 URLFetcherImplFactory::URLFetcherImplFactory() {} | 374 URLFetcherImplFactory::URLFetcherImplFactory() {} |
377 | 375 |
378 URLFetcherImplFactory::~URLFetcherImplFactory() {} | 376 URLFetcherImplFactory::~URLFetcherImplFactory() {} |
379 | 377 |
380 URLFetcher* URLFetcherImplFactory::CreateURLFetcher( | 378 URLFetcher* URLFetcherImplFactory::CreateURLFetcher( |
381 int id, | 379 int id, |
382 const GURL& url, | 380 const GURL& url, |
383 URLFetcher::RequestType request_type, | 381 URLFetcher::RequestType request_type, |
384 URLFetcherDelegate* d) { | 382 URLFetcherDelegate* d) { |
385 return new URLFetcherImpl(url, request_type, d); | 383 return new URLFetcherImpl(url, request_type, d); |
386 } | 384 } |
387 | 385 |
388 } // namespace net | 386 } // namespace net |
OLD | NEW |