Move TestURLRequestContextGetter to url_request_test_util.{h,cc}

jingle/notifier/base/proxy_resolving_client_socket_unittest.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 "jingle/notifier/base/proxy_resolving_client_socket.h"
 
 #include "base/basictypes.h"
+#include "base/compiler_specific.h"
 #include "base/message_loop.h"
 #include "net/base/mock_host_resolver.h"
 #include "net/base/test_completion_callback.h"
 #include "net/socket/socket_test_util.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
-// TODO(sanjeevr): Move this to net_test_support.
-// Used to return a dummy context.
-class TestURLRequestContextGetter : public net::URLRequestContextGetter {
+
+class ProxyTestURLRequestContextGetter : public TestURLRequestContextGetter {
  public:
-  TestURLRequestContextGetter()
-      : message_loop_proxy_(base::MessageLoopProxy::current()) {
-  }
-  virtual ~TestURLRequestContextGetter() { }
+  ProxyTestURLRequestContextGetter()
+      : TestURLRequestContextGetter(base::MessageLoopProxy::current()),
+        set_context_members_(false) {}
 
-  // net::URLRequestContextGetter:
-  virtual net::URLRequestContext* GetURLRequestContext() {
-    if (!context_)
-      CreateURLRequestContext();
-    return context_.get();
-  }
-  virtual scoped_refptr<base::MessageLoopProxy> GetIOMessageLoopProxy() const {
-    return message_loop_proxy_;
+  // Override GetURLRequestContext to set the host resolver and proxy
+  // service (used by the unit tests).
+  virtual TestURLRequestContext* GetURLRequestContext() OVERRIDE {
+    TestURLRequestContext* context =
+        TestURLRequestContextGetter::GetURLRequestContext();
+    if (!set_context_members_) {
+      context->set_host_resolver(new net::MockHostResolver());
+      context->set_proxy_service(net::ProxyService::CreateFixedFromPacResult(
+          "PROXY bad:99; PROXY maybe:80; DIRECT"));
+      set_context_members_ = true;
+    }
+    return context;
   }
 
+ protected:
+  virtual ~ProxyTestURLRequestContextGetter() {}
+
  private:
-  void CreateURLRequestContext() {
-    context_ = new TestURLRequestContext();
-    context_->set_host_resolver(new net::MockHostResolver());
-    context_->set_proxy_service(net::ProxyService::CreateFixedFromPacResult(
-        "PROXY bad:99; PROXY maybe:80; DIRECT"));
-  }
+  bool set_context_members_;
+};
 
-  scoped_refptr<net::URLRequestContext> context_;
-  scoped_refptr<base::MessageLoopProxy> message_loop_proxy_;
-};
 }  // namespace
 
 namespace notifier {
 
 class ProxyResolvingClientSocketTest : public testing::Test {
  protected:
   ProxyResolvingClientSocketTest()
-      : url_request_context_getter_(new TestURLRequestContextGetter()) {}
+      : url_request_context_getter_(new ProxyTestURLRequestContextGetter()) {}
 
   virtual ~ProxyResolvingClientSocketTest() {}
 
   virtual void TearDown() {
     // Clear out any messages posted by ProxyResolvingClientSocket's
     // destructor.
     message_loop_.RunAllPending();
   }
 
-  // Needed by XmppConnection.
-  MessageLoopForIO message_loop_;
-  scoped_refptr<TestURLRequestContextGetter> url_request_context_getter_;
+  MessageLoop message_loop_;
+  scoped_refptr<ProxyTestURLRequestContextGetter> url_request_context_getter_;
 };
 
 // TODO(sanjeevr): Fix this test on Linux.
 TEST_F(ProxyResolvingClientSocketTest, DISABLED_ConnectError) {
   net::HostPortPair dest("0.0.0.0", 0);
   ProxyResolvingClientSocket proxy_resolving_socket(
       NULL,
       url_request_context_getter_,
       net::SSLConfig(),
       dest);
@@ skipping 46 matching lines @@
   const net::ProxyRetryInfoMap& retry_info =
       context->proxy_service()->proxy_retry_info();
 
   EXPECT_EQ(1u, retry_info.size());
   net::ProxyRetryInfoMap::const_iterator iter = retry_info.find("bad:99");
   EXPECT_TRUE(iter != retry_info.end());
 }
 
 // TODO(sanjeevr): Add more unit-tests.
 }  // namespace notifier