url_request_mock_util.cc revision 21d179b334e59e9a3bfcaed4c4430bef1bc5759d 
1// Copyright (c) 2009 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/thread_restrictions.h"
11#include "chrome/browser/browser_thread.h"
12#include "chrome/browser/net/url_request_failed_dns_job.h"
13#include "chrome/browser/net/url_request_mock_http_job.h"
14#include "chrome/browser/net/url_request_mock_link_doctor_job.h"
15#include "chrome/browser/net/url_request_slow_download_job.h"
16#include "chrome/browser/net/url_request_slow_http_job.h"
17#include "chrome/common/chrome_paths.h"
18#include "net/url_request/url_request_filter.h"
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    URLRequestFilter::GetInstance()->ClearHandlers();
34
35    URLRequestFailedDnsJob::AddUrlHandler();
36    URLRequestMockLinkDoctorJob::AddUrlHandler();
37    URLRequestSlowDownloadJob::AddUrlHandler();
38
39    FilePath root_http;
40    PathService::Get(chrome::DIR_TEST_DATA, &root_http);
41    URLRequestMockHTTPJob::AddUrlHandler(root_http);
42    URLRequestSlowHTTPJob::AddUrlHandler(root_http);
43  } else {
44    // Revert to the default handlers.
45    URLRequestFilter::GetInstance()->ClearHandlers();
46  }
47}
48
49}  // namespace chrome_browser_net
50