Allow webdriver users to choose between sending the key events when...

chrome/test/webdriver/webdriver_remote_tests.py

 #!/usr/bin/python
 # Copyright (c) 2010 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.
 
 try:
   import json
 except ImportError:  # < 2.6
   import simplejson as json
 
 import os
 import optparse
 import platform
 import string
 import subprocess
 import sys
 import time
 import types
 import unittest
 import urllib2
-from selenium.common.exceptions import ErrorInResponseException
-from selenium.common.exceptions import NoSuchElementException
-from selenium.remote.webdriver import WebDriver
-from selenium.remote.webdriver.webelement import WebElement
+from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
+from selenium.webdriver.common.exceptions import ErrorInResponseException
+from selenium.webdriver.common.exceptions import NoSuchElementException
+from selenium.webdriver.remote.webdriver import WebDriver
+from selenium.webdriver.remote.webelement import WebElement
 from urlparse import urlparse
 
 
 if sys.version_info < (2,6):
   # Subprocess.Popen.kill is not available prior to 2.6.
   if platform.system() == 'Windows':
     import win32api
   else:
     import signal
 
@@ skipping 12 matching lines @@
   SEARCH = "http://www.google.com/webhp?hl=en"
   NEWS = "http://www.google.com/news?hl=en"
 
   """Verifies that navigation to a specific page is correct by asserting on the
   the reported URL.  The function will not work with pages that redirect."""
   def navigate(self, url):
     self.driver.get(url)
     self.assertURL(url)
 
   def assertURL(self, url):
-    u = self.driver.get_current_url()
+    u = self.driver.current_url
     self.assertEqual(u, url)
 
   """A new instance of chrome driver is started for every test case"""
   def setUp(self):
     global WEBDRIVER_SERVER_URL
     global WEBDRIVER_PROCESS
     WEBDRIVER_PROCESS = subprocess.Popen([WEBDRIVER_EXE,
                                         '--port=%d' % WEBDRIVER_PORT])
     if WEBDRIVER_PROCESS == None:
       print "Chromium executable not found.  The path used was: "
       print WEBDRIVER_EXE
       sys.exit(-1)
 
     time.sleep(5)
-    self.driver = WebDriver.WebDriver(WEBDRIVER_SERVER_URL, "chrome", "ANY")
+    self.driver = WebDriver(WEBDRIVER_SERVER_URL, {})
     self.assertTrue(self.driver)
 
   def tearDown(self):
     global WEBDRIVER_PROCESS
     self.driver.quit()
     if WEBDRIVER_PROCESS:
       if sys.version_info < (2,6):
         # From http://stackoverflow.com/questions/1064335
         if platform.system() == 'Windows':
           PROCESS_TERMINATE = 1
           handle = win32api.OpenProcess(PROCESS_TERMINATE, False,
                                         WEBDRIVER_PROCESS.pid)
           win32api.TerminateProcess(handle, -1)
           win32api.CloseHandle(handle)
         else:
           os.kill(WEBDRIVER_PROCESS.pid, signal.SIGKILL)
       else:
         WEBDRIVER_PROCESS.kill()
         WEBDRIVER_PROCESS = None
 
   """Preforms a string search ignoring case"""
   def assertFind(self, text, search):
     text = string.lower(text)
     search = string.lower(search)
     self.assertNotEqual(-1, string.find(text, search))
 
+class RemoteWebdriverNativeEventsTest(RemoteWebDriverTest):
+  def setUp(self):
+    global WEBDRIVER_SERVER_URL
+    global WEBDRIVER_PROCESS
+    WEBDRIVER_PROCESS = subprocess.Popen([WEBDRIVER_EXE,
+                                         '--port=%d' % WEBDRIVER_PORT])
+    if WEBDRIVER_PROCESS == None:
+      print "Chromium executable not found.  The path used was: "
+      print WEBDRIVER_EXE
+      sys.exit(-1)
+
+    time.sleep(5)
+    custom_cap = DesiredCapabilities.CHROME
+    custom_cap["chrome"] = {"nativeEvents" : True}
+    self.driver = WebDriver(WEBDRIVER_SERVER_URL, custom_cap)
+    self.assertTrue(self.driver)
+
+  def testCanStartWithNativeEventsCapabilities(self):
+    self.assertTrue(self.driver.capabilities["chrome"]["nativeEvents"])
+
+  def testSendKeysNative(self):
+    self.navigate(SEARCH)
+    # Find the Google search Button.
+    q = self.driver.find_element_by_name("q")
+    # Send some keys.
+    q.send_keys("toukyou")
+    self.assertEqual(q.get_text(), "toukyou")
+
+  def testSendKeysNativeIME(self):
+    self.navigate(SEARCH)
+    q = self.driver.find_element_by_name("q")
+    #TODO(timothe): Send the key combination that switches an JPN IME on/OFF.
+    #q.send_keys(Keys.F7)
+    q.send_keys("toukyou")
+    self.assertEqual(q.get_text(), "\xe6\x9d\xb1\xe4\xba\xac")
+
 
 class TestFindElement(RemoteWebDriverTest):
   def testFindByName(self):
     self.navigate(SEARCH)
     # Find the Google search button.
     q = self.driver.find_element_by_name("q")
     self.assertTrue(isinstance(q, WebElement))
     # Trying looking for an element not on the page.
     self.assertRaises(NoSuchElementException,
                       self.driver.find_elment_by_name, "q2")
     # Try to find the Google search button using the multiple find method.
     q = self.driver.find_elements_by_name("q")
     self.assertTrue(isinstance(q, list))
     self.assertTrue(len(q), 1)
     self.assertTrue(isinstance(q[0], WebElement))
     # Try finding something not on page, with multiple find an empty array
     # should return and no exception thrown.
     q = self.driver.find_elements_by_name("q2")
     assertTrue(q == [])
     # Find a hidden element on the page
     q = self.driver.find_element_by_name("oq")
     self.assertTrue(isinstance(q, WebElement))
 
- def testFindElementById(self):
+  def testFindElementById(self):
     self.navigate(SEARCH)
     # Find the padding for the logo near the search bar.
     elem = self.driver.find_element_by_id("logocont")
     self.assertTrue(isinstance(elem, WebElement))
     # Look for an ID not there.
     self.assertRaises(NoSuchElementException,
                       self.driver.find_element_by_id, "logocont")
 
   def testFindElementById0WithTimeout(self):
     self.set_implicit_wait(0)
@@ skipping 23 matching lines @@
     self.assertTrue(len(q), 1)
     self.assertTrue(isinstance(q[0], WebElement))
     # Try finding something not on page, with multiple find an empty array
     # should return and no exception thrown.
     q = self.driver.find_elements_by_name("q2")
     assertTrue(q == [])
     # Find a hidden element on the page
     q = self.driver.find_element_by_name("oq")
     self.assertTrue(isinstance(q, WebElement))
 
- def testFindElementById(self):
+  def testFindElementById(self):
     self.navigate(SEARCH)
     # Find the padding for the logo near the search bar.
     elem = self.driver.find_element_by_id("logocont")
     self.assertTrue(isinstance(elem, WebElement))
     # Look for an ID not there.
     self.assertRaises(NoSuchElementException,
                       self.driver.find_element_by_id, "logocont")
 
 
 class TestJavaScriptExecution(RemoteWebDriverTest):
@@ skipping 92 matching lines @@
                     help=('Specifies the URL of a remote WebDriver server to '
                           'test against. If not specified, a server will be '
                           'started on localhost according to the --exe and '
                           '--port flags'))
   parser.add_option('-e', '--exe', dest='exe', action='store',
                     type='string', default=None,
                     help=('Path to the WebDriver server executable that should '
                           'be started for testing; This flag is ignored if '
                           '--url is provided for a remote server.'))
   parser.add_option('-p', '--port', dest='port', action='store',
-                    type='int', default=8080,
+                    type='int', default=WEBDRIVER_PORT,
                     help=('The port to start the WebDriver server executable '
                           'on; This flag is ignored if --url is provided for a '
                           'remote server.'))
 
   (options, args) = parser.parse_args()
   # Strip out our flags so unittest.main() correct parses the remaining.
   sys.argv = sys.argv[:1]
   sys.argv.extend(args)
 
   if options.url:
     WEBDRIVER_SERVER_URL = options.url
   else:
     if options.port:
       WEBDRIVER_PORT = options.port
     if options.exe:
       WEBDRIVER_EXE = options.exe
     if not os.path.exists(WEBDRIVER_EXE):
       parser.error('WebDriver server executable not found:\n\t%s\n'
                    'Please specify a valid path with the --exe flag.'
                    % WEBDRIVER_EXE)
 
   unittest.main()