Added a new automation hook ExecuteJavascriptInOOBEWebUI() to execute javascript prior to login on…

chrome/test/pyautolib/pyauto.py

 #!/usr/bin/env python
 # 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.
 
 """PyAuto: Python Interface to Chromium's Automation Proxy.
 
 PyAuto uses swig to expose Automation Proxy interfaces to Python.
 For complete documentation on the functionality available,
 run pydoc on this file.
@@ skipping 3195 matching lines @@
       'javascript' : js,
       'view' : view,
       'frame_xpath' : frame_xpath,
     }
     result = self._GetResultFromJSONRequest(cmd_dict, windex=None)['result']
     # Wrap result in an array before deserializing because valid JSON has an
     # array or an object as the root.
     json_string = '[' + result + ']'
     return json.loads(json_string)[0]
 
+  def ExecuteJavascriptInOOBEWebUI(self, js, frame_xpath=''):
+    """Executes a script in the specified frame of the OOBE WebUI.
+
+    By default, execute the script in the top frame of the OOBE window. This
+    also works for all OOBE pages, including the enterprise enrollment
+    screen and login page. The invoked javascript function must send a result
+    back via the domAutomationController.send function, or this function will
+    never return.
+
+    Args:
+      js: Script to be executed.
+      frame_xpath: XPath of the frame to execute the script. Default is no
+          frame. Example: '//frames[1]'
+
+    Returns:
+      A value that was sent back via the domAutomationController.send method.
+
+    Raises:
+      pyauto_errors.JSONInterfaceError if the automation call returns an error.
+    """
+    cmd_dict = {
+      'command': 'ExecuteJavascriptInOOBEWebUI',
+
+      'javascript': js,
+      'frame_xpath': frame_xpath,
+    }
+    result = self._GetResultFromJSONRequest(cmd_dict, windex=None)['result']
+    # Wrap result in an array before deserializing because valid JSON has an
+    # array or an object as the root.
+    return json.loads('[' + result + ']')[0]
+
+
   def GetDOMValue(self, expr, tab_index=0, windex=0, frame_xpath=''):
     """Executes a Javascript expression and returns the value.
 
     This is a wrapper for ExecuteJavascript, eliminating the need to
     explicitly call domAutomationController.send function.
 
     Args:
       expr: expression value to be returned.
       tab_index: index of the tab.
       windex: index of the window.
@@ skipping 2182 matching lines @@
     successful = result.wasSuccessful()
     if not successful:
       pyauto_tests_file = os.path.join(self.TestsDir(), self._tests_filename)
       print >>sys.stderr, 'Tests can be disabled by editing %s. ' \
                           'Ref: %s' % (pyauto_tests_file, _PYAUTO_DOC_URL)
     sys.exit(not successful)
 
 
 if __name__ == '__main__':
   Main()