Add check if EC is supported for 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).
 
 
 import cairo
 import gtk
 import pango
 import os
 import subprocess
 import sys
+import re
 
 from cmath import pi
 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.bin import utils
 from autotest_lib.client.common_lib import error
 
 
 _LABEL_STATUS_SIZE = (140, 30)
 _LABEL_FONT = pango.FontDescription('courier new condensed 16')
 _LABEL_FG = gtk.gdk.color_parse('light green')
 _LABEL_UNTESTED_FG = gtk.gdk.color_parse('grey40')
 
 _PATTERN_LABEL_STR = 'pattern / 圖樣: '
 
@@ skipping 129 matching lines @@
         elif self._current_pattern.startswith("shift") and \
                 event.keyval == gtk.keysyms.Shift_L:
             self._shift_cnt += 1
             self._pattern_da.queue_draw()
         return True
 
     def register_callbacks(self, window):
         window.connect('key-release-event', self.key_release_callback)
         window.add_events(gdk.KEY_RELEASE_MASK)
 
+    def _get_embedded_controller_vendor(self):
+        # example output of superiotool:
+        #  Found Nuvoton WPCE775x (id=0x05, rev=0x02) at 0x2e
+        re_vendor = re.compile(r'Found (\w*)')
+        vendor = []
+        res = utils.system_output('superiotool', ignore_status=True).split('\n')
+        for line in res:
+            match = re_vendor.search(line)
+            if match:
+                vendor.append(match.group(1))
+        return vendor
+
     def run_once(self, led_ctl_path=None):
 
         factory.log('%s run_once' % self.__class__)
 
+        ec_vendor = self._get_embedded_controller_vendor()
+        if len(ec_vendor) == 0:
+            raise error.TestNAError('No embedded controller vendor found')
+        if 'Nuvoton' not in ec_vendor:
+            raise error.TestNAError('Currently not supported embedded controllers: %s' %
+                                    ', '.join(ec_vendor))
+
         self._led_ctl_path = led_ctl_path
         if not os.path.exists(self._led_ctl_path):
             raise error.TestNAError('Command %s does not exist' %
                                     self._led_ctl_path)
 
         self._shift_color = ful.BLACK
         self._shift_cnt = 0
 
         os.chdir(self.srcdir)
         try:
@@ skipping 46 matching lines @@
             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\n' \
                                  '以下圖樣測試未通過: %s' %
                                  ', '.join(failed_set))
 
         factory.log('%s run_once finished' % self.__class__)