Use devtools to set user agent in chromedriver

chrome/test/chromedriver/chrome/devtools_http_client.cc

 // Copyright (c) 2013 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/chrome/devtools_http_client.h"
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
@@ skipping 53 matching lines @@
   }
   return NULL;
 }
 
 DevToolsHttpClient::DevToolsHttpClient(
     const NetAddress& address,
     scoped_refptr<URLRequestContextGetter> context_getter,
     const SyncWebSocketFactory& socket_factory,
     std::unique_ptr<DeviceMetrics> device_metrics,
     std::unique_ptr<std::set<WebViewInfo::Type>> window_types,
-    std::string page_load_strategy)
+    std::string page_load_strategy,
+    std::string user_agent)
     : context_getter_(context_getter),
       socket_factory_(socket_factory),
       server_url_("http://" + address.ToString()),
       web_socket_url_prefix_(base::StringPrintf("ws://%s/devtools/page/",
                                                 address.ToString().c_str())),
       device_metrics_(std::move(device_metrics)),
       window_types_(std::move(window_types)),
-      page_load_strategy_(page_load_strategy) {
+      page_load_strategy_(page_load_strategy),
+      user_agent_(user_agent) {
   window_types_->insert(WebViewInfo::kPage);
   window_types_->insert(WebViewInfo::kApp);
 }
 
 DevToolsHttpClient::~DevToolsHttpClient() {}
 
 Status DevToolsHttpClient::Init(const base::TimeDelta& timeout) {
   base::TimeTicks deadline = base::TimeTicks::Now() + timeout;
   std::string version_url = server_url_ + "/json/version";
   std::string data;
@@ skipping 57 matching lines @@
 }
 
 const BrowserInfo* DevToolsHttpClient::browser_info() {
   return &browser_info_;
 }
 
 const DeviceMetrics* DevToolsHttpClient::device_metrics() {
   return device_metrics_.get();
 }
 
+const std::string& DevToolsHttpClient::user_agent() const {
+  return user_agent_;
+}
+
 bool DevToolsHttpClient::IsBrowserWindow(const WebViewInfo& view) const {
   return base::ContainsKey(*window_types_, view.type) ||
          (view.type == WebViewInfo::kOther &&
           (view.url == "chrome://print/" ||
            view.url == "chrome://media-router/"));
 }
 
 Status DevToolsHttpClient::CloseFrontends(const std::string& for_client_id) {
   WebViewsInfo views_info;
   Status status = GetWebViewsInfo(&views_info);
@@ skipping 27 matching lines @@
     if (status.IsError())
       return status;
   }
 
   for (std::list<std::string>::const_iterator it = docked_frontend_ids.begin();
        it != docked_frontend_ids.end(); ++it) {
     std::unique_ptr<DevToolsClient> client(new DevToolsClientImpl(
         socket_factory_, web_socket_url_prefix_ + *it, *it));
     std::unique_ptr<WebViewImpl> web_view(
         new WebViewImpl(*it, false, &browser_info_, std::move(client), NULL,
-                        page_load_strategy_));
+                        page_load_strategy_, user_agent_));
 
     status = web_view->ConnectIfNecessary();
     // Ignore disconnected error, because the debugger might have closed when
     // its container page was closed above.
     if (status.IsError() && status.code() != kDisconnected)
       return status;
 
     std::unique_ptr<base::Value> result;
     status = CloseWebView(*it);
     // Ignore disconnected error, because it may be closed already.
@@ skipping 92 matching lines @@
     Status status = ParseType(type_as_string, &type);
     if (status.IsError())
       return status;
     temp_views_info.push_back(WebViewInfo(id, debugger_url, url, type));
   }
   *views_info = WebViewsInfo(temp_views_info);
   return Status(kOk);
 }
 
 }  // namespace internal