Allow use of Mojo/V8 ProxyResolvers with URLRequestContextBuilder.

net/url_request/url_request_context_builder_v8_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 "net/url_request/url_request_context_builder_v8.h"
+
+#include "base/run_loop.h"
+#include "base/strings/stringprintf.h"
+#include "net/base/host_port_pair.h"
+#include "net/proxy/proxy_config.h"
+#include "net/proxy/proxy_config_service_fixed.h"
+#include "net/test/embedded_test_server/embedded_test_server.h"
+#include "net/test/embedded_test_server/http_request.h"
+#include "net/test/embedded_test_server/http_response.h"
+#include "net/test/embedded_test_server/simple_connection_listener.h"
+#include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
+#include "net/url_request/url_request.h"
+#include "net/url_request/url_request_context.h"
+#include "net/url_request/url_request_test_util.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/platform_test.h"
+#include "url/gurl.h"
+
+#ifdef ENABLE_NET_MOJO
+#include "net/proxy/test_mojo_proxy_resolver_factory.h"
+#endif
+
+namespace net {
+
+namespace {
+
+const char kPacPath[] = "/super.pac";
+
+// When kPacPath is requested, returns a PAC script that uses the test server
+// itself as the proxy.
+std::unique_ptr<test_server::HttpResponse> HandlePacRequest(
+    const test_server::HttpRequest& request) {
+  if (request.relative_url != kPacPath)
+    return nullptr;
+  std::unique_ptr<test_server::BasicHttpResponse> response =
+      base::MakeUnique<test_server::BasicHttpResponse>();
+  response->set_content(base::StringPrintf(
+      "function FindProxyForURL(url, host) { return 'PROXY %s;'; }",
+      HostPortPair::FromURL(request.base_url).ToString().c_str()));
+  response->set_content_type("text/html");
+  return std::move(response);
+}
+
+class URLRequestContextBuilderV8Test : public PlatformTest {
+ protected:
+  URLRequestContextBuilderV8Test() {
+    test_server_.RegisterRequestHandler(base::Bind(&HandlePacRequest));
+    test_server_.AddDefaultHandlers(
+        base::FilePath(FILE_PATH_LITERAL("net/data/url_request_unittest")));
+  }
+
+  EmbeddedTestServer test_server_;
+  URLRequestContextBuilderV8 builder_;
+};
+
+TEST_F(URLRequestContextBuilderV8Test, V8InProcess) {
+  EXPECT_TRUE(test_server_.Start());
+
+  builder_.set_proxy_config_service(base::MakeUnique<ProxyConfigServiceFixed>(
+      ProxyConfig::CreateFromCustomPacURL(test_server_.GetURL(kPacPath))));
+  std::unique_ptr<URLRequestContext> context(builder_.Build());
+
+  TestDelegate delegate;
+  std::unique_ptr<URLRequest> request(context->CreateRequest(
+      GURL("http://hats:12345/echoheader?Foo"), DEFAULT_PRIORITY, &delegate,
+      TRAFFIC_ANNOTATION_FOR_TESTS));
+  request->SetExtraRequestHeaderByName("Foo", "Bar", false);
+  request->Start();
+  base::RunLoop().Run();
+  EXPECT_EQ("Bar", delegate.data_received());
+}
+
+// Makes sure that pending PAC requests are correctly shutdown during teardown.
+TEST_F(URLRequestContextBuilderV8Test, V8InProcessShutdownWithHungRequest) {
+  test_server::SimpleConnectionListener connection_listener(
+      1, test_server::SimpleConnectionListener::FAIL_ON_ADDITIONAL_CONNECTIONS);
+  test_server_.SetConnectionListener(&connection_listener);
+  EXPECT_TRUE(test_server_.Start());
+
+  builder_.set_proxy_config_service(base::MakeUnique<ProxyConfigServiceFixed>(
+      ProxyConfig::CreateFromCustomPacURL(test_server_.GetURL("/hung"))));
+
+  std::unique_ptr<URLRequestContext> context(builder_.Build());
+  TestDelegate delegate;
+  std::unique_ptr<URLRequest> request(context->CreateRequest(
+      GURL("http://hats:12345/echoheader?Foo"), DEFAULT_PRIORITY, &delegate,
+      TRAFFIC_ANNOTATION_FOR_TESTS));
+  request->Start();
+  connection_listener.WaitForConnections();
+
+  // Tearing down the URLRequestContext should not cause an AssertNoURLRequests
+  // failure.
+}
+
+#ifdef ENABLE_NET_MOJO
+TEST_F(URLRequestContextBuilderV8Test, MojoProxyResolver) {
+  EXPECT_TRUE(test_server_.Start());
+  TestMojoProxyResolverFactory::GetInstance()->set_resolver_created(false);
+
+  builder_.set_proxy_config_service(base::MakeUnique<ProxyConfigServiceFixed>(
+      ProxyConfig::CreateFromCustomPacURL(test_server_.GetURL(kPacPath))));
+  builder_.set_mojo_proxy_resolver_factory(
+      TestMojoProxyResolverFactory::GetInstance());
+
+  std::unique_ptr<URLRequestContext> context(builder_.Build());
+  TestDelegate delegate;
+  std::unique_ptr<URLRequest> request(context->CreateRequest(
+      GURL("http://hats:12345/echoheader?Foo"), DEFAULT_PRIORITY, &delegate,
+      TRAFFIC_ANNOTATION_FOR_TESTS));
+  request->SetExtraRequestHeaderByName("Foo", "Bar", false);
+  request->Start();
+  base::RunLoop().Run();
+  EXPECT_EQ("Bar", delegate.data_received());
+
+  // Make sure that the Mojo factory was used.
+  EXPECT_TRUE(TestMojoProxyResolverFactory::GetInstance()->resolver_created());
+}
+#endif  // ENABLE_NET_MOJO
+
+}  // namespace
+
+}  // namespace net