test_url_fetcher_factory.cc revision 868fa2fe829687343ffae624259930155e16dbd8
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/test_url_fetcher_factory.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/compiler_specific.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/weak_ptr.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/message_loop.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/host_port_pair.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/http/http_response_headers.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_fetcher_delegate.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_fetcher_impl.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_status.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ScopedURLFetcherFactory::ScopedURLFetcherFactory( 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLFetcherFactory* factory) { 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(!URLFetcherImpl::factory()); 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLFetcherImpl::set_factory(factory); 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ScopedURLFetcherFactory::~ScopedURLFetcherFactory() { 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(URLFetcherImpl::factory()); 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLFetcherImpl::set_factory(NULL); 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TestURLFetcher::TestURLFetcher(int id, 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const GURL& url, 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLFetcherDelegate* d) 352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) : owner_(NULL), 362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) id_(id), 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) original_url_(url), 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate_(d), 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate_for_tests_(NULL), 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) did_receive_last_chunk_(false), 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_load_flags_(0), 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_response_code_(-1), 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_response_destination_(STRING), 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_was_fetched_via_proxy_(false), 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_max_retries_(0) { 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TestURLFetcher::~TestURLFetcher() { 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (delegate_for_tests_) 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate_for_tests_->OnRequestEnd(id_); 512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (owner_) 522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) owner_->RemoveFetcherFromMap(id_); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::SetUploadData(const std::string& upload_content_type, 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& upload_content) { 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) upload_data_ = upload_content; 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void TestURLFetcher::SetUploadFilePath( 612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const std::string& upload_content_type, 622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& file_path, 63c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) uint64 range_offset, 64c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) uint64 range_length, 652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_refptr<base::TaskRunner> file_task_runner) { 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) upload_file_path_ = file_path; 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::SetChunkedUpload(const std::string& upload_content_type) { 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::AppendChunkToUpload(const std::string& data, 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool is_last_chunk) { 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(!did_receive_last_chunk_); 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) did_receive_last_chunk_ = is_last_chunk; 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) chunks_.push_back(data); 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (delegate_for_tests_) 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate_for_tests_->OnChunkUpload(id_); 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::SetLoadFlags(int load_flags) { 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_load_flags_= load_flags; 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int TestURLFetcher::GetLoadFlags() const { 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return fake_load_flags_; 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::SetReferrer(const std::string& referrer) { 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::SetExtraRequestHeaders( 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& extra_request_headers) { 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_extra_request_headers_.Clear(); 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_extra_request_headers_.AddHeadersFromString(extra_request_headers); 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::AddExtraRequestHeader(const std::string& header_line) { 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_extra_request_headers_.AddHeaderFromString(header_line); 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::GetExtraRequestHeaders( 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HttpRequestHeaders* headers) const { 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *headers = fake_extra_request_headers_; 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::SetRequestContext( 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLRequestContextGetter* request_context_getter) { 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::SetFirstPartyForCookies( 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const GURL& first_party_for_cookies) { 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::SetURLRequestUserData( 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const void* key, 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CreateDataCallback& create_data_callback) { 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::SetStopOnRedirect(bool stop_on_redirect) { 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::SetAutomaticallyRetryOn5xx(bool retry) { 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void TestURLFetcher::SetMaxRetriesOn5xx(int max_retries) { 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_max_retries_ = max_retries; 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)int TestURLFetcher::GetMaxRetriesOn5xx() const { 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return fake_max_retries_; 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::TimeDelta TestURLFetcher::GetBackoffDelay() const { 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return fake_backoff_delay_; 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void TestURLFetcher::SetAutomaticallyRetryOnNetworkChanges(int max_retries) { 1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::SaveResponseToFileAtPath( 1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& file_path, 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<base::TaskRunner> file_task_runner) { 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::SaveResponseToTemporaryFile( 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<base::TaskRunner> file_task_runner) { 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HttpResponseHeaders* TestURLFetcher::GetResponseHeaders() const { 151868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) return fake_response_headers_.get(); 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HostPortPair TestURLFetcher::GetSocketAddress() const { 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTIMPLEMENTED(); 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HostPortPair(); 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool TestURLFetcher::WasFetchedViaProxy() const { 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return fake_was_fetched_via_proxy_; 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::Start() { 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Overriden to do nothing. It is assumed the caller will notify the delegate. 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (delegate_for_tests_) 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate_for_tests_->OnRequestStart(id_); 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const GURL& TestURLFetcher::GetOriginalURL() const { 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return original_url_; 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const GURL& TestURLFetcher::GetURL() const { 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return fake_url_; 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const URLRequestStatus& TestURLFetcher::GetStatus() const { 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return fake_status_; 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int TestURLFetcher::GetResponseCode() const { 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return fake_response_code_; 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const ResponseCookies& TestURLFetcher::GetCookies() const { 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return fake_cookies_; 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool TestURLFetcher::FileErrorOccurred(int* out_error_code) const { 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTIMPLEMENTED(); 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::ReceivedContentWasMalformed() { 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool TestURLFetcher::GetResponseAsString( 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string* out_response_string) const { 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (fake_response_destination_ != STRING) 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *out_response_string = fake_response_string_; 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool TestURLFetcher::GetResponseAsFilePath( 2072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool take_ownership, base::FilePath* out_response_path) const { 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (fake_response_destination_ != TEMP_FILE) 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *out_response_path = fake_response_file_path_; 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::set_status(const URLRequestStatus& status) { 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_status_ = status; 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::set_was_fetched_via_proxy(bool flag) { 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_was_fetched_via_proxy_ = flag; 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::set_response_headers( 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<HttpResponseHeaders> headers) { 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_response_headers_ = headers; 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::set_backoff_delay(base::TimeDelta backoff_delay) { 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_backoff_delay_ = backoff_delay; 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::SetDelegateForTests(DelegateForTests* delegate_for_tests) { 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate_for_tests_ = delegate_for_tests; 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcher::SetResponseString(const std::string& response) { 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_response_destination_ = STRING; 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_response_string_ = response; 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void TestURLFetcher::SetResponseFilePath(const base::FilePath& path) { 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_response_destination_ = TEMP_FILE; 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_response_file_path_ = path; 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TestURLFetcherFactory::TestURLFetcherFactory() 247c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) : ScopedURLFetcherFactory(this), 2482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) delegate_for_tests_(NULL), 2492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) remove_fetcher_on_delete_(false) { 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TestURLFetcherFactory::~TestURLFetcherFactory() {} 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcher* TestURLFetcherFactory::CreateURLFetcher( 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int id, 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const GURL& url, 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLFetcher::RequestType request_type, 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLFetcherDelegate* d) { 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestURLFetcher* fetcher = new TestURLFetcher(id, url, d); 2602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (remove_fetcher_on_delete_) 2612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) fetcher->set_owner(this); 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fetcher->SetDelegateForTests(delegate_for_tests_); 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fetchers_[id] = fetcher; 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return fetcher; 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TestURLFetcher* TestURLFetcherFactory::GetFetcherByID(int id) const { 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Fetchers::const_iterator i = fetchers_.find(id); 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return i == fetchers_.end() ? NULL : i->second; 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcherFactory::RemoveFetcherFromMap(int id) { 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Fetchers::iterator i = fetchers_.find(id); 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(i != fetchers_.end()); 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fetchers_.erase(i); 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestURLFetcherFactory::SetDelegateForTests( 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestURLFetcherDelegateForTests* delegate_for_tests) { 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate_for_tests_ = delegate_for_tests; 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)FakeURLFetcher::FakeURLFetcher(const GURL& url, 2842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) URLFetcherDelegate* d, 2852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const std::string& response_data, 2862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool success) 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : TestURLFetcher(0, url, d), 288c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) weak_factory_(this) { 2892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) set_status(URLRequestStatus( 2902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) success ? URLRequestStatus::SUCCESS : URLRequestStatus::FAILED, 2912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 0)); 2922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) set_response_code(success ? 200 : 500); 2932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SetResponseString(response_data); 2942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)FakeURLFetcher::~FakeURLFetcher() {} 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void FakeURLFetcher::Start() { 29990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoop::current()->PostTask( 3002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) FROM_HERE, 3012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&FakeURLFetcher::RunDelegate, weak_factory_.GetWeakPtr())); 3022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void FakeURLFetcher::RunDelegate() { 3052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) delegate()->OnURLFetchComplete(this); 3062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const GURL& FakeURLFetcher::GetURL() const { 3092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return TestURLFetcher::GetOriginalURL(); 3102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)FakeURLFetcherFactory::FakeURLFetcherFactory( 3132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) URLFetcherFactory* default_factory) 314c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) : ScopedURLFetcherFactory(this), 3152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) creator_(base::Bind(&DefaultFakeURLFetcherCreator)), 3162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) default_factory_(default_factory) { 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)FakeURLFetcherFactory::FakeURLFetcherFactory( 3202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) URLFetcherFactory* default_factory, 3212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const FakeURLFetcherCreator& creator) 322c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) : ScopedURLFetcherFactory(this), 3232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) creator_(creator), 3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) default_factory_(default_factory) { 3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)scoped_ptr<FakeURLFetcher> FakeURLFetcherFactory::DefaultFakeURLFetcherCreator( 3282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const GURL& url, 3292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) URLFetcherDelegate* delegate, 3302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const std::string& response, 3312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool success) { 3322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return scoped_ptr<FakeURLFetcher>(new FakeURLFetcher(url, delegate, 3332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) response, success)); 3342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)FakeURLFetcherFactory::~FakeURLFetcherFactory() {} 3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcher* FakeURLFetcherFactory::CreateURLFetcher( 3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int id, 3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const GURL& url, 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLFetcher::RequestType request_type, 3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLFetcherDelegate* d) { 3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FakeResponseMap::const_iterator it = fake_responses_.find(url); 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it == fake_responses_.end()) { 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (default_factory_ == NULL) { 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If we don't have a baked response for that URL we return NULL. 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DLOG(ERROR) << "No baked response for URL: " << url.spec(); 3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return default_factory_->CreateURLFetcher(id, url, request_type, d); 3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<FakeURLFetcher> fake_fetcher = 3552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) creator_.Run(url, d, it->second.first, it->second.second); 3562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // TODO: Make URLFetcherFactory::CreateURLFetcher return a scoped_ptr 3572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return fake_fetcher.release(); 3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void FakeURLFetcherFactory::SetFakeResponse(const std::string& url, 3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& response_data, 3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool success) { 3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Overwrite existing URL if it already exists. 3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_responses_[GURL(url)] = std::make_pair(response_data, success); 3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void FakeURLFetcherFactory::ClearFakeResponses() { 3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fake_responses_.clear(); 3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcherImplFactory::URLFetcherImplFactory() {} 3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcherImplFactory::~URLFetcherImplFactory() {} 3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcher* URLFetcherImplFactory::CreateURLFetcher( 3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int id, 3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const GURL& url, 3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLFetcher::RequestType request_type, 3795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLFetcherDelegate* d) { 3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return new URLFetcherImpl(url, request_type, d); 3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 384