[chromedriver] Implement command: executeAsyncScript and setScriptTimeout

chrome/test/chromedriver/chromedriver.py

 # 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.
 
 import ctypes
 import json
 
 from webelement import WebElement
 
 class ChromeDriverException(Exception):
   pass
 class NoSuchElement(ChromeDriverException):
   pass
 class UnknownCommand(ChromeDriverException):
   pass
 class StaleElementReference(ChromeDriverException):
   pass
 class UnknownError(ChromeDriverException):
   pass
+class JavaScriptError(ChromeDriverException):
+  pass
 class XPathLookupError(ChromeDriverException):
   pass
+class Timeout(ChromeDriverException):
+  pass
 class NoSuchWindow(ChromeDriverException):
   pass
 class InvalidSelector(ChromeDriverException):
   pass
 class SessionNotCreatedException(ChromeDriverException):
   pass
 class NoSuchSession(ChromeDriverException):
   pass
 
 def _ExceptionForResponse(response):
   exception_class_map = {
     7: NoSuchElement,
     9: UnknownCommand,
     10: StaleElementReference,
     13: UnknownError,
+    17: JavaScriptError,
     19: XPathLookupError,
+    21: Timeout,
     23: NoSuchWindow,
     32: InvalidSelector,
     33: SessionNotCreatedException,
     100: NoSuchSession
   }
   status = response['status']
   msg = response['value']['message']
   return exception_class_map.get(status, ChromeDriverException)(msg)
 
 class ChromeDriver(object):
@@ skipping 97 matching lines @@
     self.ExecuteSessionCommand('close')
 
   def Load(self, url):
     self.ExecuteSessionCommand('get', {'url': url})
 
   def ExecuteScript(self, script, *args):
     converted_args = list(args)
     return self.ExecuteSessionCommand(
         'executeScript', {'script': script, 'args': converted_args})
 
+  def ExecuteAsyncScript(self, script, *args):
+    converted_args = list(args)
+    return self.ExecuteSessionCommand(
+        'executeAsyncScript', {'script': script, 'args': converted_args})
+
   def SwitchToFrame(self, id_or_name):
     self.ExecuteSessionCommand('switchToFrame', {'id': id_or_name})
 
   def SwitchToFrameByIndex(self, index):
     self.SwitchToFrame(index)
 
   def SwitchToMainFrame(self):
     self.SwitchToFrame(None)
 
   def GetTitle(self):
@@ skipping 59 matching lines @@
       self.ExecuteSessionCommand('setAlertValue', {'text': prompt_text})
     if accept:
       cmd = 'acceptAlert'
     else:
       cmd = 'dismissAlert'
     self.ExecuteSessionCommand(cmd)
 
   def Quit(self):
     """Quits the browser and ends the session."""
     self.ExecuteSessionCommand('quit')