Index: chrome/test/chromedriver/net/url_request_context_getter.cc |
diff --git a/chrome/test/chromedriver/net/url_request_context_getter.cc b/chrome/test/chromedriver/net/url_request_context_getter.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..cdda0c2711b6d3ee42c1dd76a8d9d877b1eab9a1 |
--- /dev/null |
+++ b/chrome/test/chromedriver/net/url_request_context_getter.cc |
@@ -0,0 +1,54 @@ |
+// 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 "chrome/test/chromedriver/net/url_request_context_getter.h" |
+ |
+#include <string> |
+ |
+#include "net/proxy/proxy_config_service.h" |
+#include "net/url_request/url_request_context.h" |
+#include "net/url_request/url_request_context_builder.h" |
+ |
+namespace { |
+ |
+// Config getter that always returns direct settings. |
+class ProxyConfigServiceDirect : public net::ProxyConfigService { |
+ public: |
+ // Overridden from ProxyConfigService: |
+ virtual void AddObserver(Observer* observer) OVERRIDE {} |
+ virtual void RemoveObserver(Observer* observer) OVERRIDE {} |
+ virtual ConfigAvailability GetLatestProxyConfig( |
+ net::ProxyConfig* config) OVERRIDE { |
+ *config = net::ProxyConfig::CreateDirect(); |
+ return CONFIG_VALID; |
+ } |
+}; |
+ |
+} // namespace |
+ |
+URLRequestContextGetter::URLRequestContextGetter( |
+ scoped_refptr<base::SingleThreadTaskRunner> network_task_runner) |
+ : network_task_runner_(network_task_runner) { |
+} |
+ |
+net::URLRequestContext* URLRequestContextGetter::GetURLRequestContext() { |
+ if (!url_request_context_) { |
+ net::URLRequestContextBuilder builder; |
+ // net::HttpServer fails to parse headers if user-agent header is blank. |
+ builder.set_user_agent("chromedriver"); |
+ builder.DisableHttpCache(); |
+#if defined(OS_LINUX) || defined(OS_ANDROID) |
+ builder.set_proxy_config_service(new ProxyConfigServiceDirect()); |
+#endif |
+ url_request_context_.reset(builder.Build()); |
+ } |
+ return url_request_context_.get(); |
+} |
+ |
+scoped_refptr<base::SingleThreadTaskRunner> |
+ URLRequestContextGetter::GetNetworkTaskRunner() const { |
+ return network_task_runner_; |
+} |
+ |
+URLRequestContextGetter::~URLRequestContextGetter() {} |