Introduce net::HttpServerPropertiesManager to manage server-specific properties.

net/url_request/url_request_unittest.cc

 // Copyright (c) 2011 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 "build/build_config.h"
 
 #if defined(OS_WIN)
 #include <shlobj.h>
 #include <windows.h>
 #endif
@@ skipping 235 matching lines @@
 
  private:
   mutable URLRequestJob* main_intercept_job_;
 };
 
 // Inherit PlatformTest since we require the autorelease pool on Mac OS X.f
 class URLRequestTest : public PlatformTest {
  public:
   URLRequestTest() : default_context_(new TestURLRequestContext(true)) {
     default_context_->set_network_delegate(&default_network_delegate_);
+    default_context_->set_http_server_properties(NULL);
     default_context_->Init();
   }
 
   static void SetUpTestCase() {
     URLRequest::AllowFileAccess();
   }
 
   // Adds the TestJobInterceptor to the default context.
   TestJobInterceptor* AddTestInterceptor() {
     TestJobInterceptor* interceptor = new TestJobInterceptor();
@@ skipping 90 matching lines @@
 // issuing a CONNECT request with the magic host name "www.redirect.com".
 // The HTTPTestServer will return a 302 response, which we should not
 // follow.
 TEST_F(URLRequestTestHTTP, ProxyTunnelRedirectTest) {
   ASSERT_TRUE(test_server_.Start());
 
   TestNetworkDelegate network_delegate;  // must outlive URLRequest
   scoped_refptr<TestURLRequestContext> context(new TestURLRequestContext(true));
   context->SetProxyFromString(test_server_.host_port_pair().ToString());
   context->set_network_delegate(&network_delegate);
+  context->set_http_server_properties(NULL);
   context->Init();
 
   TestDelegate d;
   {
     URLRequest r(GURL("https://www.redirect.com/"), &d);
     r.set_context(context);
     r.Start();
     EXPECT_TRUE(r.is_pending());
 
     MessageLoop::current()->Run();
 
     EXPECT_EQ(URLRequestStatus::FAILED, r.status().status());
     EXPECT_EQ(ERR_TUNNEL_CONNECTION_FAILED, r.status().error());
     EXPECT_EQ(1, d.response_started_count());
     // We should not have followed the redirect.
     EXPECT_EQ(0, d.received_redirect_count());
   }
 }
 
 // This is the same as the previous test, but checks that the network delegate
 // registers the error.
 TEST_F(URLRequestTestHTTP, NetworkDelegateTunnelConnectionFailed) {
   ASSERT_TRUE(test_server_.Start());
 
   TestNetworkDelegate network_delegate;  // must outlive URLRequest
   scoped_refptr<TestURLRequestContext> context(new TestURLRequestContext(true));
   context->SetProxyFromString(test_server_.host_port_pair().ToString());
   context->set_network_delegate(&network_delegate);
+  context->set_http_server_properties(NULL);
   context->Init();
 
   TestDelegate d;
   {
     URLRequest r(GURL("https://www.redirect.com/"), &d);
     r.set_context(context);
     r.Start();
     EXPECT_TRUE(r.is_pending());
 
     MessageLoop::current()->Run();
@@ skipping 13 matching lines @@
 TEST_F(URLRequestTestHTTP, NetworkDelegateCancelRequest) {
   ASSERT_TRUE(test_server_.Start());
 
   TestDelegate d;
   BlockingNetworkDelegate network_delegate;
   network_delegate.set_callback_retval(ERR_EMPTY_RESPONSE);
 
   scoped_refptr<TestURLRequestContext> context(new TestURLRequestContext(true));
   context->SetProxyFromString(test_server_.host_port_pair().ToString());
   context->set_network_delegate(&network_delegate);
+  context->set_http_server_properties(NULL);
   context->Init();
 
   {
     TestURLRequest r(test_server_.GetURL(""), &d);
     r.set_context(context);
 
     r.Start();
     MessageLoop::current()->Run();
 
     EXPECT_EQ(URLRequestStatus::FAILED, r.status().status());
     EXPECT_EQ(ERR_EMPTY_RESPONSE, r.status().error());
     EXPECT_EQ(1, network_delegate.created_requests());
     EXPECT_EQ(0, network_delegate.destroyed_requests());
   }
   EXPECT_EQ(1, network_delegate.destroyed_requests());
 }
 
 // Tests that the network delegate can cancel a request synchronously.
 TEST_F(URLRequestTestHTTP, NetworkDelegateCancelRequestSynchronously) {
   ASSERT_TRUE(test_server_.Start());
 
   TestDelegate d;
   BlockingNetworkDelegate network_delegate;
   network_delegate.set_retval(ERR_EMPTY_RESPONSE);
 
   scoped_refptr<TestURLRequestContext> context(new TestURLRequestContext(true));
   context->SetProxyFromString(test_server_.host_port_pair().ToString());
   context->set_network_delegate(&network_delegate);
+  context->set_http_server_properties(NULL);
   context->Init();
 
   {
     TestURLRequest r(test_server_.GetURL(""), &d);
     r.set_context(context);
 
     r.Start();
     MessageLoop::current()->Run();
 
     EXPECT_EQ(URLRequestStatus::FAILED, r.status().status());
@@ skipping 10 matching lines @@
   ASSERT_TRUE(test_server_.Start());
 
   TestDelegate d;
   BlockingNetworkDelegate network_delegate;
   GURL redirect_url(test_server_.GetURL("simple.html"));
   network_delegate.set_redirect_url(redirect_url);
 
   scoped_refptr<TestURLRequestContext> context(new TestURLRequestContext(true));
   context->SetProxyFromString(test_server_.host_port_pair().ToString());
   context->set_network_delegate(&network_delegate);
+  context->set_http_server_properties(NULL);
   context->Init();
 
   {
     GURL original_url(test_server_.GetURL("empty.html"));
     TestURLRequest r(original_url, &d);
     r.set_context(context);
 
     r.Start();
     MessageLoop::current()->Run();
 
@@ skipping 14 matching lines @@
 
   const char kData[] = "hello world";
 
   TestDelegate d;
   BlockingNetworkDelegate network_delegate;
   GURL redirect_url(test_server_.GetURL("echo"));
   network_delegate.set_redirect_url(redirect_url);
 
   scoped_refptr<TestURLRequestContext> context(new TestURLRequestContext(true));
   context->set_network_delegate(&network_delegate);
+  context->set_http_server_properties(NULL);
   context->Init();
 
   {
     GURL original_url(test_server_.GetURL("empty.html"));
     TestURLRequest r(original_url, &d);
     r.set_context(context);
     r.set_method("POST");
     r.set_upload(CreateSimpleUploadData(kData).get());
     HttpRequestHeaders headers;
     headers.SetHeader(HttpRequestHeaders::kContentLength,
@@ skipping 231 matching lines @@
 // In this unit test, we're using the HTTPTestServer as a proxy server and
 // issuing a CONNECT request with the magic host name "www.server-auth.com".
 // The HTTPTestServer will return a 401 response, which we should balk at.
 TEST_F(URLRequestTestHTTP, UnexpectedServerAuthTest) {
   ASSERT_TRUE(test_server_.Start());
 
   TestNetworkDelegate network_delegate;  // must outlive URLRequest
   scoped_refptr<TestURLRequestContext> context(new TestURLRequestContext(true));
   context->SetProxyFromString(test_server_.host_port_pair().ToString());
   context->set_network_delegate(&network_delegate);
+  context->set_http_server_properties(NULL);
   context->Init();
 
   TestDelegate d;
   {
     URLRequest r(GURL("https://www.server-auth.com/"), &d);
     r.set_context(context);
 
     r.Start();
     EXPECT_TRUE(r.is_pending());
 
@@ skipping 73 matching lines @@
     TestDelegate d;
     {
       std::string test_file =
           base::StringPrintf("compressedfiles/BullRunSpeech.txt?%c",
                              test_parameters[i]);
 
       TestNetworkDelegate network_delegate;  // must outlive URLRequest
       scoped_refptr<TestURLRequestContext> context(
           new TestURLRequestContext(true));
       context->set_network_delegate(&network_delegate);
+      context->set_http_server_properties(NULL);
       context->Init();
 
       TestURLRequest r(test_server_.GetURL(test_file), &d);
       r.set_context(context);
       r.Start();
       EXPECT_TRUE(r.is_pending());
 
       MessageLoop::current()->Run();
 
       EXPECT_EQ(1, d.response_started_count());
@@ skipping 59 matching lines @@
   ASSERT_EQ(3U, req.url_chain().size());
   EXPECT_EQ(original_url, req.url_chain()[0]);
   EXPECT_EQ(middle_redirect_url, req.url_chain()[1]);
   EXPECT_EQ(destination_url, req.url_chain()[2]);
 }
 
 class HTTPSRequestTest : public testing::Test {
  public:
   HTTPSRequestTest() : default_context_(new TestURLRequestContext(true)) {
     default_context_->set_network_delegate(&default_network_delegate_);
+    default_context_->set_http_server_properties(NULL);
     default_context_->Init();
   }
   virtual ~HTTPSRequestTest() {}
 
  protected:
   TestNetworkDelegate default_network_delegate_;  // must outlive URLRequest
   scoped_refptr<TestURLRequestContext> default_context_;
 };
 
 TEST_F(HTTPSRequestTest, HTTPSGetTest) {
@@ skipping 1002 matching lines @@
   GURL url_requiring_auth =
       test_server_.GetURL("auth-basic?set-cookie-if-challenged");
 
   // Request a page that will give a 401 containing a Set-Cookie header.
   // Verify that when the transaction is restarted, it includes the new cookie.
   {
     TestNetworkDelegate network_delegate;  // must outlive URLRequest
     scoped_refptr<TestURLRequestContext> context(
         new TestURLRequestContext(true));
     context->set_network_delegate(&network_delegate);
+    context->set_http_server_properties(NULL);
     context->Init();
 
     TestDelegate d;
     d.set_username(kUser);
     d.set_password(kSecret);
 
     URLRequest r(url_requiring_auth, &d);
     r.set_context(context);
     r.Start();
 
     MessageLoop::current()->Run();
 
     EXPECT_TRUE(d.data_received().find("user/secret") != std::string::npos);
 
     // Make sure we sent the cookie in the restarted transaction.
     EXPECT_TRUE(d.data_received().find("Cookie: got_challenged=true")
         != std::string::npos);
   }
 
   // Same test as above, except this time the restart is initiated earlier
   // (without user intervention since identity is embedded in the URL).
   {
     TestNetworkDelegate network_delegate;  // must outlive URLRequest
     scoped_refptr<TestURLRequestContext> context(
         new TestURLRequestContext(true));
     context->set_network_delegate(&network_delegate);
+    context->set_http_server_properties(NULL);
     context->Init();
 
     TestDelegate d;
 
     GURL::Replacements replacements;
     std::string username("user2");
     std::string password("secret");
     replacements.SetUsernameStr(username);
     replacements.SetPasswordStr(password);
     GURL url_with_identity = url_requiring_auth.ReplaceComponents(replacements);
@@ skipping 992 matching lines @@
 
 // Check that a failure to connect to the proxy is reported to the network
 // delegate.
 TEST_F(URLRequestTest, NetworkDelegateProxyError) {
   MockHostResolver host_resolver;
   host_resolver.rules()->AddSimulatedFailure("*");
 
   TestNetworkDelegate network_delegate;  // must outlive URLRequests
   scoped_refptr<TestURLRequestContext> context(new TestURLRequestContext(true));
   context->set_network_delegate(&network_delegate);
+  context->set_http_server_properties(NULL);
   context->SetProxyFromString("myproxy:70");
   context->set_host_resolver(&host_resolver);
   context->Init();
 
   TestDelegate d;
   TestURLRequest req(GURL("http://example.com"), &d);
   req.set_context(context);
   req.set_method("GET");
 
   req.Start();
@@ skipping 396 matching lines @@
   }
 }
 
 // Check that default A-L header is sent.
 TEST_F(URLRequestTestHTTP, DefaultAcceptLanguage) {
   ASSERT_TRUE(test_server_.Start());
 
   TestNetworkDelegate network_delegate;  // must outlive URLRequests
   scoped_refptr<TestURLRequestContext> context(new TestURLRequestContext(true));
   context->set_network_delegate(&network_delegate);
+  context->set_http_server_properties(NULL);
   context->set_accept_language("en");
   context->Init();
 
   TestDelegate d;
   TestURLRequest req(test_server_.GetURL("echoheader?Accept-Language"), &d);
   req.set_context(context);
   req.Start();
   MessageLoop::current()->Run();
   EXPECT_EQ("en", d.data_received());
 }
 
 // Check that an empty A-L header is not sent. http://crbug.com/77365.
 TEST_F(URLRequestTestHTTP, EmptyAcceptLanguage) {
   ASSERT_TRUE(test_server_.Start());
 
   TestNetworkDelegate network_delegate;  // must outlive URLRequests
   scoped_refptr<TestURLRequestContext> context(new TestURLRequestContext(true));
   context->set_network_delegate(&network_delegate);
+  context->set_http_server_properties(NULL);
   context->Init();
   // We override the language after initialization because empty entries
   // get overridden by Init().
   context->set_accept_language("");
 
   TestDelegate d;
   TestURLRequest req(test_server_.GetURL("echoheader?Accept-Language"), &d);
   req.set_context(context);
   req.Start();
   MessageLoop::current()->Run();
@@ skipping 47 matching lines @@
   EXPECT_TRUE(ContainsString(d.data_received(), "identity"));
 }
 
 // Check that default A-C header is sent.
 TEST_F(URLRequestTestHTTP, DefaultAcceptCharset) {
   ASSERT_TRUE(test_server_.Start());
 
   TestNetworkDelegate network_delegate;  // must outlive URLRequests
   scoped_refptr<TestURLRequestContext> context(new TestURLRequestContext(true));
   context->set_network_delegate(&network_delegate);
+  context->set_http_server_properties(NULL);
   context->set_accept_charset("en");
   context->Init();
 
   TestDelegate d;
   TestURLRequest req(test_server_.GetURL("echoheader?Accept-Charset"), &d);
   req.set_context(context);
   req.Start();
   MessageLoop::current()->Run();
   EXPECT_EQ("en", d.data_received());
 }
 
 // Check that an empty A-C header is not sent. http://crbug.com/77365.
 TEST_F(URLRequestTestHTTP, EmptyAcceptCharset) {
   ASSERT_TRUE(test_server_.Start());
 
   TestNetworkDelegate network_delegate;  // must outlive URLRequests
   scoped_refptr<TestURLRequestContext> context(new TestURLRequestContext(true));
   context->set_network_delegate(&network_delegate);
+  context->set_http_server_properties(NULL);
   context->Init();
   // We override the accepted charset after initialization because empty
   // entries get overridden otherwise.
   context->set_accept_charset("");
 
   TestDelegate d;
   TestURLRequest req(test_server_.GetURL("echoheader?Accept-Charset"), &d);
   req.set_context(context);
   req.Start();
   MessageLoop::current()->Run();
@@ skipping 41 matching lines @@
   req.SetExtraRequestHeaders(headers);
   req.Start();
   MessageLoop::current()->Run();
   // If the net tests are being run with ChromeFrame then we need to allow for
   // the 'chromeframe' suffix which is added to the user agent before the
   // closing parentheses.
   EXPECT_TRUE(StartsWithASCII(d.data_received(), "Lynx (textmode", true));
 }
 
 }  // namespace net