[chromedriver] Add page loading strategy to capabilities

chrome/test/chromedriver/chrome/chrome_impl.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 "chrome/test/chromedriver/chrome/chrome_impl.h"
 
 #include <stddef.h>
 #include <utility>
 
 #include "chrome/test/chromedriver/chrome/devtools_client.h"
 #include "chrome/test/chromedriver/chrome/devtools_event_listener.h"
 #include "chrome/test/chromedriver/chrome/devtools_http_client.h"
+#include "chrome/test/chromedriver/chrome/page_load_strategy.h"
 #include "chrome/test/chromedriver/chrome/status.h"
 #include "chrome/test/chromedriver/chrome/web_view_impl.h"
 #include "chrome/test/chromedriver/net/port_server.h"
 
 ChromeImpl::~ChromeImpl() {
   if (!quit_)
     port_reservation_->Leak();
 }
 
 Status ChromeImpl::GetAsDesktop(ChromeDesktopImpl** desktop) {
@@ skipping 48 matching lines @@
         std::unique_ptr<DevToolsClient> client(
             devtools_http_client_->CreateClient(view.id));
         for (ScopedVector<DevToolsEventListener>::const_iterator listener =
                  devtools_event_listeners_.begin();
              listener != devtools_event_listeners_.end(); ++listener) {
           client->AddListener(*listener);
           // OnConnected will fire when DevToolsClient connects later.
         }
         web_views_.push_back(make_linked_ptr(new WebViewImpl(
             view.id, devtools_http_client_->browser_info(), std::move(client),
-            devtools_http_client_->device_metrics())));
+            devtools_http_client_->device_metrics(), page_load_strategy_)));
       }
     }
   }
 
   std::list<std::string> web_view_ids_tmp;
   for (WebViewList::const_iterator web_view_iter = web_views_.begin();
        web_view_iter != web_views_.end(); ++web_view_iter) {
     web_view_ids_tmp.push_back((*web_view_iter)->GetId());
   }
   web_view_ids->swap(web_view_ids_tmp);
@@ skipping 30 matching lines @@
 }
 
 bool ChromeImpl::IsMobileEmulationEnabled() const {
   return false;
 }
 
 bool ChromeImpl::HasTouchScreen() const {
   return false;
 }
 
+std::string ChromeImpl::page_load_strategy() const {
+  return page_load_strategy_;
+}
+
+void ChromeImpl::set_page_load_strategy(std::string strategy) {
+  // Support for page load strategy already checked when capability is parsed.
+  page_load_strategy_ = strategy;
+}
+
 Status ChromeImpl::Quit() {
   Status status = QuitImpl();
   if (status.IsOk())
     quit_ = true;
   return status;
 }
 
 ChromeImpl::ChromeImpl(
     std::unique_ptr<DevToolsHttpClient> http_client,
     std::unique_ptr<DevToolsClient> websocket_client,
     ScopedVector<DevToolsEventListener>& devtools_event_listeners,
     std::unique_ptr<PortReservation> port_reservation)
     : quit_(false),
       devtools_http_client_(std::move(http_client)),
       devtools_websocket_client_(std::move(websocket_client)),
+      page_load_strategy_(PageLoadStrategy::kNormal),
       port_reservation_(std::move(port_reservation)) {
   devtools_event_listeners_.swap(devtools_event_listeners);
 }