Keyboard localization

client/site_tests/factory_SelectKeyboard/factory_SelectKeyboard.py

Index: client/site_tests/factory_SelectKeyboard/factory_SelectKeyboard.py
diff --git a/client/site_tests/factory_SelectKeyboard/factory_SelectKeyboard.py b/client/site_tests/factory_SelectKeyboard/factory_SelectKeyboard.py
new file mode 100644
index 0000000000000000000000000000000000000000..7245a90b4c78a22723161cca0cd707b6583ec572
--- /dev/null
+++ b/client/site_tests/factory_SelectKeyboard/factory_SelectKeyboard.py
@@ -0,0 +1,71 @@
+# 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.
+
+
+# DESCRIPTION :
+#
+# Select the keyboard type, and write to VPD.
+
+
+import gtk
+import pango
+import sys
+import utils
+
+from gtk import gdk
+
+from autotest_lib.client.bin import factory
+from autotest_lib.client.bin import factory_ui_lib as ful
+from autotest_lib.client.bin import test
+from autotest_lib.client.common_lib import error
+
+# Mapping between menu choice and KB.
+kb_map = {
+  '1': 'en-US',
+  '2': 'en-GB',
+  'q': None,
+}
+
+# Message to display.
+msg = ('Choose a keyboard:\n' + 
+      "".join([ '%s) %s\n' % (i, kb_map[i]) for i in sorted(kb_map)]))
+
+class factory_SelectKeyboard(test.test):
+    version = 1
+
+    def write_kb(self, kb):
+        cmd = 'vpd -s "initial_locale"="%s"' % kb
+        utils.system_output(cmd)
+
+    def key_release_callback(self, widget, event):
+        char = event.keyval in range(32,127) and chr(event.keyval) or None
+        factory.log('key_release %s(%s)' % (event.keyval, char))
+        if char in kb_map:
+          kb = kb_map[char]
+          factory.log('Keyboard specified as %s, (pressed %s)' % (
+              kb, char))
+
+          if kb:
+            self.write_kb(kb)
+
+          gtk.main_quit()
+        return True
+
+    def register_callbacks(self, window):
+        window.connect('key-release-event', self.key_release_callback)
+        window.add_events(gdk.KEY_RELEASE_MASK)
+
+    def run_once(self):
+
+        factory.log('%s run_once' % self.__class__)
+        label = ful.make_label(msg)
+
+        test_widget = gtk.EventBox()
+        test_widget.modify_bg(gtk.STATE_NORMAL, ful.BLACK)
+        test_widget.add(label)
+
+        ful.run_test_widget(self.job, test_widget,
+            window_registration_callback=self.register_callbacks)
+
+        factory.log('%s run_once finished' % repr(self.__class__))