add shift key (Shift_L) LED to factory_Leds test

client/site_tests/factory_Leds/factory_Leds.py

 # -*- coding: utf-8 -*-
 #
 # 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 :
 #
 # This is a factory test to test the LEDs (wifi, battery, etc).
@@ skipping 25 matching lines @@
 _ORANGE = gtk.gdk.color_parse('orange')
 
 _LED_NAMES = 'sleepled battchgled wlan battfulled powerled 3g'.split()
 _ORANGE_LED_NAMES = _LED_NAMES[0:3]
 _BLUE_LED_NAMES = _LED_NAMES[3:]
 
 _LED_COUNT = 3
 _LED_Y_OFFSET = 10
 _LED_X_OFFSETS = [10, 80, 150]
 _LED_RADIUS = 18
+_SHFL_XYR = {'x':166,'y':20, 'r':9}
 
 
 def run_led_ctl(led_ctl_path, args):
     rc = subprocess.call([led_ctl_path] + args.split())
     if rc != 1:
         factory.log('%s(%s) failed with rc %s' % (led_ctl_path, args, rc))
 
 
 def pattern_all_off(led_ctl_fn):
     for n in _LED_NAMES:
         led_ctl_fn('-%s -off' % n)
     return [ful.BLACK, ful.BLACK, ful.BLACK]
 
 def pattern_blue_on(led_ctl_fn):
     pattern_all_off(led_ctl_fn)
     for n in _BLUE_LED_NAMES:
         led_ctl_fn('-%s -on' % n)
     return [ful.BLUE, ful.BLUE, ful.BLUE]
 
 def pattern_orange_on(led_ctl_fn):
     pattern_all_off(led_ctl_fn)
     for n in _ORANGE_LED_NAMES:
         led_ctl_fn('-%s -on' % n)
     return [_ORANGE, _ORANGE, _ORANGE]
 
 _PATTERN_LIST = [
     ('all off', pattern_all_off),
     ('blue on', pattern_blue_on),
-    ('orange on', pattern_orange_on)]
+    ('orange on', pattern_orange_on),
+    ('shift led', False)]
 
 
 class factory_Leds(test.test):
     version = 1
     preserve_srcdir = True
 
     def goto_next_pattern(self):
         if not self._pattern_queue:
             gtk.main_quit()
             return
         self._current_pattern, cb_fn = self._pattern_queue.pop()
         self._status_map[self._current_pattern] = ful.ACTIVE
-        led_ctl_fn = lambda args: run_led_ctl(self._led_ctl_path, args)
-        self._led_colors = cb_fn(led_ctl_fn)
+        if cb_fn:
+            led_ctl_fn = lambda args: run_led_ctl(self._led_ctl_path, args)
+            self._led_colors = cb_fn(led_ctl_fn)
+        else:
+            self._pattern_da.connect('expose_event', self.shift_led_expose)
         self._pattern_da.queue_draw()
 
+    def shift_led_expose(self, widget, event):
+        context = widget.window.cairo_create()
+        context.set_source_surface(self._shf_image, 0, 0)
+        context.paint()
+
+        if self._shift_cnt is 2:
+            if self._shift_color is ful.BLACK:
+                self._shift_color = ful.BLUE
+            else:
+                self._shift_color = ful.BLACK
+            self._shift_cnt = 0
+        context.set_source_color(self._shift_color)
+        context.arc(_SHFL_XYR['x'], _SHFL_XYR['y'], _SHFL_XYR['r'],
+                    0.0, 2.0 * pi)
+        context.fill()
+
     def pattern_expose(self, widget, event):
         context = widget.window.cairo_create()
         context.set_source_surface(self._leds_image, 0, 0)
         context.paint()
         for led_index in range(_LED_COUNT):
             color = self._led_colors[led_index]
             x_offset = _LED_X_OFFSETS[led_index] + _LED_RADIUS + 3
             y_offset = _LED_Y_OFFSET + _LED_RADIUS + 3
             context.set_source_color(color)
             context.arc(x_offset, y_offset, _LED_RADIUS, 0.0, 2.0 * pi)
@@ skipping 33 matching lines @@
         eb.add(hbox)
         return eb
 
     def key_release_callback(self, widget, event):
         if event.keyval == gtk.keysyms.Tab:
             self._status_map[self._current_pattern] = ful.FAILED
             self.goto_next_pattern()
         elif event.keyval == gtk.keysyms.Return:
             self._status_map[self._current_pattern] = ful.PASSED
             self.goto_next_pattern()
+        elif self._current_pattern.startswith("shift") and \
+                event.keyval == gtk.keysyms.Shift_L:
+            self._shift_cnt += 1
+            self._pattern_da.queue_draw()
         else:
             self._ft_state.exit_on_trigger(event)
         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,
                  test_widget_size=None,
                  trigger_set=None,
                  led_ctl_path=None):
 
         factory.log('%s run_once' % self.__class__)
 
         self._ft_state = ful.State(trigger_set)
 
         self._led_ctl_path = led_ctl_path
 
+        self._shift_color = ful.BLACK
+        self._shift_cnt = 0
+
         os.chdir(self.srcdir)
         image = cairo.ImageSurface.create_from_png('leds.png')
         image_size = (image.get_width(), image.get_height())
         self._leds_image = image
 
+        image = cairo.ImageSurface.create_from_png('shf.png')
+        self._shf_image = image
+
         self._pattern_queue = [x for x in reversed(_PATTERN_LIST)]
         self._status_map = dict((n, ful.UNTESTED) for n, f in _PATTERN_LIST)
 
         pattern_da = gtk.DrawingArea()
         pattern_da.set_size_request(*image_size)
         pattern_da.connect('expose_event', self.pattern_expose)
         self._pattern_da = pattern_da
 
         pattern_label = ful.make_label(_PATTERN_LABEL_STR)
 
@@ skipping 27 matching lines @@
             window_registration_callback=self.register_callbacks,
             cleanup_callback=self.quit)
 
         failed_set = set(name for name, status in self._status_map.items()
                          if status is not ful.PASSED)
         if failed_set:
             raise error.TestFail('some patterns failed (%s)' %
                                  ', '.join(failed_set))
 
         factory.log('%s run_once finished' % self.__class__)