Added IME tests for web forms

client/site_tests/desktopui_ImeTest/desktopui_ImeTest.py

 # Copyright (c) 2010 The Chromium OS 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 os, string, time, gtk
 from autotest_lib.client.bin import site_ui_test, test
-from autotest_lib.client.common_lib import error, site_ui, utils
+from autotest_lib.client.common_lib import error, site_ui, utils, site_httpd
 
 
 class desktopui_ImeTest(site_ui_test.UITest):
     version = 1
     preserve_srcdir = True
 
     def setup(self):
         # TODO: We shouldn't use ibusclient, we should talk to Chrome directly
         self.job.setup_dep(['ibusclient'])
 
+    def initialize(self, creds='$default'):
+        self._test_url = 'http://localhost:8000/interaction_form.html'
+        self._test_server = site_httpd.HTTPListener(8000, docroot=self.bindir)
+        self._test_server.run()
+
+        site_ui_test.UITest.initialize(self, creds)
+
+
+    def cleanup(self):
+        self._test_server.stop()
+        site_ui_test.UITest.cleanup(self)
+
 
     def log_error(self, test_name, message):
         self.job.record('ERROR', None, test_name, message)
         self._failed.append(test_name)
 
 
     # TODO(zork) We should share this with platform_ProcessPrivleges.
     # See: crosbug.com/7453
     def check_process(self, process, user=None):
         """Check if the process is running as the specified user / root.
@@ skipping 123 matching lines @@
         ax.send_hotkey('Ctrl+t')
         time.sleep(1)
         ax.send_hotkey('Ctrl+l')
         time.sleep(1)
         ax.send_text('chrome://settings/language#lang=%s,focus=%s\n' %
                      (language, engine))
         time.sleep(3)
 
         # Toggle the checkbox.
         ax.send_text(' ')
-        time.sleep(1)
+        # The toggling can take longer than 1 sec.
+        time.sleep(2)
 
         # Close the window.
         ax.send_hotkey('Ctrl+w')
         time.sleep(1)
 
 
     def stop_ime_language(self, language):
         """
         Remove a language from Chrome's preferred list and disable all its IMEs.
 
@@ skipping 72 matching lines @@
         expected_initial_engine = 'xkb:us::eng'
         expected_other_engine = 'xkb:us:altgr-intl:eng'
 
         current_engine = self.get_active_engine()
         if current_engine != expected_initial_engine:
             self.log_error('test_keyboard_shortcut',
                            'Initial engine is %s, expected %s' %
                            (current_engine, expected_initial_engine))
         ax = self.get_autox()
         ax.send_hotkey('Ctrl-l')
+        # If we don't sleep here sometimes the following keys are not received
+        time.sleep(1)
         ax.send_hotkey('Ctrl-space')
         start_time = time.time()
         while time.time() - start_time < 10:
             current_engine = self.get_active_engine()
             if current_engine == expected_other_engine:
                 ax.send_hotkey('Ctrl-space')
                 return
             time.sleep(1)
         self.log_error('test_keyboard_shortcut',
                        'Current engine is %s, expected %s' %
                        (current_engine, expected_other_engine))
 
 
     def test_engine(self, language, engine_name, input_string, expected_string):
         self.start_ime_engine(language, engine_name)
         self.activate_engine(engine_name)
+        self.test_engine_omnibox(language, engine_name, input_string,
+                                 expected_string)
+        self.test_engine_form(language, engine_name, input_string,
+                              expected_string)
+        self.activate_engine('xkb:us::eng')
+        self.stop_ime_language(language)
 
+
+    def test_engine_omnibox(self, language, engine_name, input_string,
+                            expected_string):
         ax = self.get_autox()
 
         # Focus on the omnibox so that we can enter text.
         ax.send_hotkey('Ctrl-l')
 
         # Sometimes there is a slight delay before input can be received in the
         # omnibox.
         time.sleep(1)
 
         ax.send_text(input_string)
 
         text = self.get_current_text()
         if text != expected_string:
             self.log_error(
-                'test_engine %s' % engine_name,
-                'Engine %s failed: Got %s, expected %s' % (engine_name, text,
-                                                           expected_string))
+                'test_engine %s in omnibox' % engine_name,
+                'Engine %s failed : Got %s, expected %s' % (
+                    engine_name, text, expected_string))
+        # Clear the omnibox for future tests.
+        ax.send_hotkey('BackSpace')
+
+
+    def test_engine_form(self, language, engine_name, input_string,
+                         expected_string):
+        ax = self.get_autox()
+        # Go to the page containing the form.
         self.activate_engine('xkb:us::eng')
-        self.stop_ime_language(language)
+        ax.send_hotkey("Ctrl+l")
+        time.sleep(1)
+        ax.send_text("%s \n" % self._test_url)
+        time.sleep(1)
+        self.activate_engine(engine_name)
+
+        ax.send_text(input_string)
+        text = self.get_current_text()
+        if text != expected_string:
+            self.log_error(
+                'test_engine %s in form' % engine_name,
+                'Engine %s failed : Got %s, expected %s' % (
+                    engine_name, text, expected_string))
 
 
     def run_once(self):
         self._failed = []
         dep = 'ibusclient'
         dep_dir = os.path.join(self.autodir, 'deps', dep)
         self.job.install_pkg(dep, 'dep', dep_dir)
 
         self.exefile = os.path.join(self.autodir,
                                     'deps/ibusclient/ibusclient')
 
         # Before we try to activate the options menu, we need to wait for
         # previous actions to complete.  Most notable is that keystrokes
         # immediately after login get lost.
         time.sleep(5)
 
         self.test_ibus_start_process()
 
         self.check_process('candidate_window', user='chronos')
         self.check_process('ibus-daemon', user='chronos')
         self.check_process('ibus-memconf', user='chronos')
 
         self.test_keyboard_shortcut()
         self.test_engine('ja', 'mozc', 'nihongo \n',
                          '\xE6\x97\xA5\xE6\x9C\xAC\xE8\xAA\x9E')
         self.test_engine('zh-TW', 'chewing', 'hol \n', '\xE6\x93\x8D')
         self.test_engine('ko', 'hangul', 'wl ', '\xEC\xA7\x80 ')
         self.test_engine('zh-CN', 'pinyin', 'nihao ',
                          '\xE4\xBD\xA0\xE5\xA5\xBD')
-        self.test_engine('zh-TW', 'm17n:zh:quick', 'aa', '\xE9\x96\x93')
+        self.test_engine('zh-TW', 'm17n:zh:quick', 'aa ', '\xE9\x96\x93')
 
         self.test_ibus_stop_process()
+
         if len(self._failed) != 0:
             raise error.TestFail(
                 'Failed: %s' % ','.join(self._failed))