15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved.
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be
35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file.
45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "chrome/test/chromedriver/net/url_request_context_getter.h"
62a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string>
85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/proxy/proxy_config_service_fixed.h"
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_context.h"
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_context_builder.h"
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLRequestContextGetter::URLRequestContextGetter(
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    scoped_refptr<base::SingleThreadTaskRunner> network_task_runner)
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    : network_task_runner_(network_task_runner) {
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)net::URLRequestContext* URLRequestContextGetter::GetURLRequestContext() {
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  CHECK(network_task_runner_->BelongsToCurrentThread());
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!url_request_context_) {
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    net::URLRequestContextBuilder builder;
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // net::HttpServer fails to parse headers if user-agent header is blank.
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    builder.set_user_agent("chromedriver");
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    builder.DisableHttpCache();
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    builder.set_proxy_config_service(
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        new net::ProxyConfigServiceFixed(net::ProxyConfig::CreateDirect()));
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    url_request_context_.reset(builder.Build());
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return url_request_context_.get();
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)scoped_refptr<base::SingleThreadTaskRunner>
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    URLRequestContextGetter::GetNetworkTaskRunner() const {
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return network_task_runner_;
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLRequestContextGetter::~URLRequestContextGetter() {}
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)