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 3// found in the LICENSE file. 4 5#include "chrome/browser/net/url_request_mock_util.h" 6 7#include <string> 8 9#include "base/path_service.h" 10#include "base/threading/thread_restrictions.h" 11#include "chrome/common/chrome_paths.h" 12#include "content/public/browser/browser_thread.h" 13#include "content/test/net/url_request_slow_download_job.h" 14#include "net/test/url_request/url_request_failed_job.h" 15#include "net/test/url_request/url_request_mock_http_job.h" 16#include "net/url_request/url_request_filter.h" 17 18using content::BrowserThread; 19 20namespace chrome_browser_net { 21 22void SetUrlRequestMocksEnabled(bool enabled) { 23 // Since this involves changing the net::URLRequest ProtocolFactory, we need 24 // to run on the IO thread. 25 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 26 27 if (enabled) { 28 // We have to look around for our helper files, but we only use 29 // this from tests, so allow these IO operations to happen 30 // anywhere. 31 base::ThreadRestrictions::ScopedAllowIO allow_io; 32 33 net::URLRequestFilter::GetInstance()->ClearHandlers(); 34 35 net::URLRequestFailedJob::AddUrlHandler(); 36 content::URLRequestSlowDownloadJob::AddUrlHandler(); 37 38 base::FilePath root_http; 39 PathService::Get(chrome::DIR_TEST_DATA, &root_http); 40 net::URLRequestMockHTTPJob::AddUrlHandler(root_http, 41 BrowserThread::GetBlockingPool()); 42 } else { 43 // Revert to the default handlers. 44 net::URLRequestFilter::GetInstance()->ClearHandlers(); 45 } 46} 47 48} // namespace chrome_browser_net 49