[chromedriver] Add page loading strategy to capabilities

chrome/test/chromedriver/client/chromedriver.py

 # Copyright 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.
 
 import sys
 import platform
 
 import command_executor
 from command_executor import Command
 from webelement import WebElement
@@ skipping 54 matching lines @@
 
 class ChromeDriver(object):
   """Starts and controls a single Chrome instance on this machine."""
 
   def __init__(self, server_url, chrome_binary=None, android_package=None,
                android_activity=None, android_process=None,
                android_use_running_app=None, chrome_switches=None,
                chrome_extensions=None, chrome_log_path=None,
                debugger_address=None, logging_prefs=None,
                mobile_emulation=None, experimental_options=None,
-               download_dir=None, network_connection=None):
+               download_dir=None, page_load_strategy=None,
+               network_connection=None):
     self._executor = command_executor.CommandExecutor(server_url)
 
     options = {}
 
     if experimental_options:
       assert isinstance(experimental_options, dict)
       options = experimental_options.copy()
 
     if android_package:
       options['androidPackage'] = android_package
@@ skipping 51 matching lines @@
         options['prefs']['download'] = {}
       options['prefs']['download']['default_directory'] = download_dir
 
     params = {
       'desiredCapabilities': {
         'chromeOptions': options,
         'loggingPrefs': logging_prefs
       }
     }
 
+    if page_load_strategy:
+      assert type(page_load_strategy) is str
+      params['desiredCapabilities']['pageLoadingStrategy'] = page_load_strategy
+
     if network_connection:
       params['desiredCapabilities']['networkConnectionEnabled'] = (
           network_connection)
 
     response = self._ExecuteCommand(Command.NEW_SESSION, params)
     self._session_id = response['sessionId']
     self.capabilities = self._UnwrapValue(response['value'])
 
   def _WrapValue(self, value):
     """Wrap value from client side for chromedriver side."""
@@ skipping 256 matching lines @@
 
   def GetNetworkConnection(self):
     return self.ExecuteCommand(Command.GET_NETWORK_CONNECTION)
 
   def DeleteNetworkConditions(self):
     self.ExecuteCommand(Command.DELETE_NETWORK_CONDITIONS)
 
   def SetNetworkConnection(self, connection_type):
     params = {'parameters': {'type': connection_type}}
     self.ExecuteCommand(Command.SET_NETWORK_CONNECTION, params)