[net] Switch TestURLRequestContext to use MockCachingHostResolver.

net/url_request/url_request_test_util.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/url_request/url_request_test_util.h"
 
 #include "base/compiler_specific.h"
 #include "base/logging.h"
 #include "base/message_loop.h"
 #include "base/threading/thread.h"
 #include "base/threading/worker_pool.h"
 #include "net/base/cert_verifier.h"
 #include "net/base/default_server_bound_cert_store.h"
 #include "net/base/host_port_pair.h"
+#include "net/base/mock_host_resolver.h"
 #include "net/base/server_bound_cert_service.h"
 #include "net/http/http_network_session.h"
 #include "net/http/http_server_properties_impl.h"
 #include "net/url_request/url_request_job_factory.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 // These constants put the net::NetworkDelegate events of TestNetworkDelegate
 // into an order. They are used in conjunction with
 // |TestNetworkDelegate::next_states_| to check that we do not send
 // events in the wrong order.
 const int kStageBeforeURLRequest = 1 << 0;
 const int kStageBeforeSendHeaders = 1 << 1;
 const int kStageSendHeaders = 1 << 2;
 const int kStageHeadersReceived = 1 << 3;
 const int kStageAuthRequired = 1 << 4;
 const int kStageBeforeRedirect = 1 << 5;
 const int kStageResponseStarted = 1 << 6;
 const int kStageCompletedSuccess = 1 << 7;
 const int kStageCompletedError = 1 << 8;
 const int kStageURLRequestDestroyed = 1 << 9;
 const int kStageDestruction = 1 << 10;
 
 }  // namespace
 
 TestURLRequestContext::TestURLRequestContext()
     : initialized_(false),
       ALLOW_THIS_IN_INITIALIZER_LIST(context_storage_(this)) {
-  context_storage_.set_host_resolver(
-      net::CreateSystemHostResolver(net::HostResolver::kDefaultParallelism,
-                                    net::HostResolver::kDefaultRetryAttempts,
-                                    NULL));
-  SetProxyDirect();
   Init();
 }
 
 TestURLRequestContext::TestURLRequestContext(bool delay_initialization)
     : initialized_(false),
       ALLOW_THIS_IN_INITIALIZER_LIST(context_storage_(this)) {
-  context_storage_.set_host_resolver(
-      net::CreateSystemHostResolver(net::HostResolver::kDefaultParallelism,
-                                    net::HostResolver::kDefaultRetryAttempts,
-                                    NULL));
-  SetProxyDirect();
   if (!delay_initialization)
     Init();
 }
 
-TestURLRequestContext::TestURLRequestContext(const std::string& proxy)
-    : initialized_(false),
-      ALLOW_THIS_IN_INITIALIZER_LIST(context_storage_(this)) {
-  context_storage_.set_host_resolver(
-      net::CreateSystemHostResolver(net::HostResolver::kDefaultParallelism,
-                                    net::HostResolver::kDefaultRetryAttempts,
-                                    NULL));
-  SetProxyFromString(proxy);
-  Init();
-}
-
-TestURLRequestContext::TestURLRequestContext(const char* proxy)
-    : initialized_(false),
-      ALLOW_THIS_IN_INITIALIZER_LIST(context_storage_(this)) {
-  context_storage_.set_host_resolver(
-      net::CreateSystemHostResolver(net::HostResolver::kDefaultParallelism,
-                                    net::HostResolver::kDefaultRetryAttempts,
-                                    NULL));
-  SetProxyFromString(proxy);
-  Init();
-}
-
-TestURLRequestContext::TestURLRequestContext(const std::string& proxy,
-                                             net::HostResolver* host_resolver)
-    : initialized_(false),
-      ALLOW_THIS_IN_INITIALIZER_LIST(context_storage_(this)) {
-  context_storage_.set_host_resolver(host_resolver);
-  SetProxyFromString(proxy);
-  Init();
-}
-
-void TestURLRequestContext::SetProxyFromString(const std::string& proxy) {
-  DCHECK(!initialized_);
-  net::ProxyConfig proxy_config;
-  proxy_config.proxy_rules().ParseFromString(proxy);
-  context_storage_.set_proxy_service(
-      net::ProxyService::CreateFixed(proxy_config));
-}
-
-void TestURLRequestContext::SetProxyDirect() {
-  DCHECK(!initialized_);
-  context_storage_.set_proxy_service(net::ProxyService::CreateDirect());
-}
-
 TestURLRequestContext::~TestURLRequestContext() {
   DCHECK(initialized_);
 }
 
 void TestURLRequestContext::Init() {
   DCHECK(!initialized_);
   initialized_ = true;
+
+  if (!host_resolver())
+    context_storage_.set_host_resolver(new net::MockCachingHostResolver());
+  if (!proxy_service())
+    context_storage_.set_proxy_service(net::ProxyService::CreateDirect());
   if (!cert_verifier())
     context_storage_.set_cert_verifier(net::CertVerifier::CreateDefault());
   if (!ftp_transaction_factory()) {
     context_storage_.set_ftp_transaction_factory(
         new net::FtpNetworkLayer(host_resolver()));
   }
   if (!ssl_config_service())
     context_storage_.set_ssl_config_service(new net::SSLConfigServiceDefaults);
   if (!http_auth_handler_factory()) {
     context_storage_.set_http_auth_handler_factory(
@@ skipping 28 matching lines @@
             base::WorkerPool::GetTaskRunner(true)));
   }
   if (accept_language().empty())
     set_accept_language("en-us,fr");
   if (accept_charset().empty())
     set_accept_charset("iso-8859-1,*,utf-8");
   if (!job_factory())
     context_storage_.set_job_factory(new net::URLRequestJobFactory);
 }
 
-
 TestURLRequest::TestURLRequest(const GURL& url, Delegate* delegate)
     : net::URLRequest(url, delegate),
       context_(new TestURLRequestContext) {
   set_context(context_.get());
 }
 
 TestURLRequest::~TestURLRequest() {
   set_context(NULL);
 }
 
 TestURLRequestContextGetter::TestURLRequestContextGetter(
     const scoped_refptr<base::MessageLoopProxy>& io_message_loop_proxy)
     : io_message_loop_proxy_(io_message_loop_proxy) {
   DCHECK(io_message_loop_proxy.get());
 }
 
+TestURLRequestContextGetter::TestURLRequestContextGetter(
+    const scoped_refptr<base::MessageLoopProxy>& io_message_loop_proxy,
+    scoped_ptr<TestURLRequestContext> context)
+    : io_message_loop_proxy_(io_message_loop_proxy), context_(context.Pass()) {
+  DCHECK(io_message_loop_proxy.get());
+}
+
 TestURLRequestContextGetter::~TestURLRequestContextGetter() {}
 
 TestURLRequestContext* TestURLRequestContextGetter::GetURLRequestContext() {
   if (!context_.get())
     context_.reset(new TestURLRequestContext);
   return context_.get();
 }
 
 scoped_refptr<base::MessageLoopProxy>
 TestURLRequestContextGetter::GetIOMessageLoopProxy() const {
@@ skipping 357 matching lines @@
 
 ScopedCustomUrlRequestTestHttpHost::~ScopedCustomUrlRequestTestHttpHost() {
   DCHECK_EQ(value_, new_value_);
   value_ = old_value_;
 }
 
 // static
 const std::string& ScopedCustomUrlRequestTestHttpHost::value() {
   return value_;
 }