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

chrome/test/webdriver/chromedriver_tests.py

Index: chrome/test/webdriver/chromedriver_tests.py
===================================================================
--- chrome/test/webdriver/chromedriver_tests.py	(revision 78866)
+++ chrome/test/webdriver/chromedriver_tests.py	(working copy)
@@ -27,6 +27,8 @@
 import simplejson as json
 
 from selenium.webdriver.remote.webdriver import WebDriver
+from selenium.webdriver.common.keys import Keys
+from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
 
 
 class Request(urllib2.Request):
@@ -111,17 +113,56 @@
   def testCanStartChromeDriverOnSpecificPort(self):
     launcher = ChromeDriverLauncher(port=9520)
     self.assertEquals(9520, launcher.GetPort())
-    driver = WebDriver(launcher.GetURL(), {})
+    driver = WebDriver(launcher.GetURL(), DesiredCapabilities.CHROME)
     driver.quit()
     launcher.Kill()
 
 
+class NativeInputTest(unittest.TestCase):
+  """Native input ChromeDriver tests."""
+
+  SEARCH = "http://www.google.com/webhp?hl=en"
+
+  def setUp(self):
+    self._launcher = ChromeDriverLauncher()
+    self.capabilities = DesiredCapabilities.CHROME
+    self.capabilities["chrome"] = { "nativeEvents" : True }
+
+  def tearDown(self):
+    self._launcher.Kill()
+
+  def testCanStartsWithNativeEvents(self):
+    driver = WebDriver(self._launcher.GetURL(), self.capabilities)
+    self.assertTrue(driver.capabilities["chrome"].has_key("nativeEvents"))
+    self.assertTrue(driver.capabilities["chrome"]["nativeEvents"])
+
+  def testSendKeysNative(self):
+    driver = WebDriver(self._launcher.GetURL(), self.capabilities)
+    driver.get(self.SEARCH)
+    # Find the Google search Button.
+    q = driver.find_element_by_name("q")
+    # Send some keys.
+    q.send_keys("tokyo")
+    self.assertEqual(q.value, "tokyo")
+
+  #@unittest.skip("Need to run this on a machine with an IME installed.")
+  #def testSendKeysNativeProcessedByIME(self):
+    #driver = WebDriver(self._launcher.GetURL(), self.capabilities)
+    #driver.get(self.SEARCH)
+    #q = driver.find_element_by_name("q")
+    ##Send key combination to turn IME on.
+    #q.send_keys(Keys.F7)
+    #q.send_keys("toukyou")
+    #self.assertEqual(q.value, "\xe6\x9d\xb1\xe4\xba\xac")
+
+
 class CookieTest(unittest.TestCase):
   """Cookie test for the json webdriver protocol"""
 
   def setUp(self):
     self._launcher = ChromeDriverLauncher()
-    self._driver = WebDriver(self._launcher.GetURL(), {})
+    self._driver = WebDriver(self._launcher.GetURL(),
+                             DesiredCapabilities.CHROME)
 
   def tearDown(self):
     self._driver.quit()
@@ -199,16 +240,16 @@
     driver.quit()
 
   def testSessionCreationDeletion(self):
-    driver = WebDriver(self._launcher.GetURL(), {})
+    driver = WebDriver(self._launcher.GetURL(), DesiredCapabilities.CHROME)
     driver.quit()
 
   def testMultipleSessionCreationDeletion(self):
     for i in range(10):
-      driver = WebDriver(self._launcher.GetURL(), {})
+      driver = WebDriver(self._launcher.GetURL(), DesiredCapabilities.CHROME)
       driver.quit()
 
   def testSessionCommandsAfterSessionDeletionReturn404(self):
-    driver = WebDriver(self._launcher.GetURL(), {})
+    driver = WebDriver(self._launcher.GetURL(), DesiredCapabilities.CHROME)
     session_id = driver.session_id
     driver.quit()
     try:
@@ -221,7 +262,8 @@
   def testMultipleConcurrentSessions(self):
     drivers = []
     for i in range(10):
-      drivers += [WebDriver(self._launcher.GetURL(), {})]
+      drivers += [WebDriver(self._launcher.GetURL(),
+                            DesiredCapabilities.CHROME)]
     for driver in drivers:
       driver.quit()
 
@@ -231,7 +273,8 @@
 
   def setUp(self):
     self._launcher = ChromeDriverLauncher(root_path=os.path.dirname(__file__))
-    self._driver = WebDriver(self._launcher.GetURL(), {})
+    self._driver = WebDriver(self._launcher.GetURL(),
+                             DesiredCapabilities.CHROME)
 
   def tearDown(self):
     self._driver.quit()