[chromedriver] Implement command: switchToWindow.

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 XPathLookupError(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,
     19: XPathLookupError,
+    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):
   """Starts and controls a single Chrome instance on this machine."""
@@ skipping 73 matching lines @@
     return response
 
   def ExecuteSessionCommand(self, name, params={}):
     params = self._WrapValue(params)
     return self._UnwrapValue(
         self._ExecuteCommand(name, params, self._session_id)['value'])
 
   def GetWindowHandles(self):
     return self.ExecuteSessionCommand('getWindowHandles')
 
+  def SwitchToWindow(self, handle_or_name):
+    self.ExecuteSessionCommand('switchToWindow', {'name': handle_or_name})
+
   def GetCurrentWindowHandle(self):
     return self.ExecuteSessionCommand('getCurrentWindowHandle')
 
   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})
@@ skipping 52 matching lines @@
 
   def MouseButtonUp(self, button=0):
     self.ExecuteSessionCommand('mouseButtonUp', {'button': button})
 
   def MouseDoubleClick(self, button=0):
     self.ExecuteSessionCommand('mouseDoubleClick', {'button': button})
 
   def Quit(self):
     """Quits the browser and ends the session."""
     self.ExecuteSessionCommand('quit')