[chromedriver] Implement command: switchToWindow.

chrome/test/chromedriver/run_py_tests.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.
 
 """End to end tests for ChromeDriver."""
 
 import ctypes
 import optparse
 import os
@@ skipping 49 matching lines @@
     self._driver.Load(self.GetHttpUrlForFile('/chromedriver/page_test.html'))
     window_count = len(self._driver.GetWindowHandles())
     self._driver.FindElement('id', 'link').Click()
     timeout = time.time() + 20
     while time.time() < timeout:
       if (len(self._driver.GetWindowHandles()) > window_count):
         return
       time.sleep(0.01)
     self.assertTrue(False)
 
+  def testSwitchToWindow(self):
+    self._driver.Load(self.GetHttpUrlForFile('/chromedriver/page_test.html'))
+    self.assertEquals(
+        1, self._driver.ExecuteScript('window.name = "oldWindow"; return 1;'))
+    window1_handle = self._driver.GetCurrentWindowHandle()
+    window_count = len(self._driver.GetWindowHandles())
+    self._driver.FindElement('id', 'link').Click()
+    timeout = time.time() + 20
+    while time.time() < timeout:
+      all_handles = self._driver.GetWindowHandles()
+      if (len(all_handles) > window_count):
+        break
+      time.sleep(0.01)
+    self.assertTrue(window1_handle in all_handles)
+    all_handles.remove(window1_handle)
+    self._driver.SwitchToWindow(all_handles[0])
+    self.assertEquals(all_handles[0], self._driver.GetCurrentWindowHandle())
+    self.assertRaises(chromedriver.NoSuchElement,
+                      self._driver.FindElement, 'id', 'link')
+    self._driver.SwitchToWindow('oldWindow')
+    self.assertEquals(window1_handle, self._driver.GetCurrentWindowHandle())
+
   def testEvaluateScript(self):
     self.assertEquals(1, self._driver.ExecuteScript('return 1'))
     self.assertEquals(None, self._driver.ExecuteScript(''))
 
   def testEvaluateScriptWithArgs(self):
     script = ('document.body.innerHTML = "<div>b</div><div>c</div>";' +
               'return {stuff: document.querySelectorAll("div")};')
     stuff = self._driver.ExecuteScript(script)['stuff']
     script = 'return arguments[0].innerHTML + arguments[1].innerHTML'
     self.assertEquals(
@@ skipping 226 matching lines @@
   global _ANDROID_PACKAGE
   _ANDROID_PACKAGE = options.android_package
 
   all_tests_suite = unittest.defaultTestLoader.loadTestsFromModule(
       sys.modules[__name__])
   tests = unittest_util.FilterTestSuite(all_tests_suite, options.filter)
   ChromeDriverTest.GlobalSetUp()
   result = unittest.TextTestRunner().run(tests)
   ChromeDriverTest.GlobalTearDown()
   sys.exit(len(result.failures) + len(result.errors))