| # Copyright (c) 2012 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. |
| |
| """Some utility classes and functions.""" |
| |
| import os |
| import re |
| import sys |
| import time |
| |
| import common_util |
| import test_conf as conf |
| |
| |
| def get_display_name(): |
| """Return the display name.""" |
| return ':0' |
| |
| |
| def get_current_time_str(): |
| """Get the string of current time.""" |
| time_format = '%Y%m%d_%H%M%S' |
| return time.strftime(time_format, time.gmtime()) |
| |
| |
| def get_board(): |
| """Get board of the Chromebook machine.""" |
| with open('/etc/lsb-release') as lsb: |
| context = lsb.read() |
| board = 'unknown_board' |
| if context is not None: |
| for line in context.splitlines(): |
| if line.startswith('CHROMEOS_RELEASE_BOARD'): |
| board_str = line.split('=')[1] |
| if '-' in board_str: |
| board = board_str.split('-')[1] |
| elif '_' in board_str: |
| board = board_str.split('_')[1] |
| else: |
| board = board_str |
| # Some boards, e.g. alex, may have board name as alex32 |
| board = re.search('(\D+)\d*', board, re.I).group(1) |
| break |
| return board |
| |
| |
| def create_log_dir(firmware_version, mode): |
| """Create a directory to save the report and device event files.""" |
| dir_basename = conf.filename.sep.join([get_current_time_str(), |
| 'fw_' + firmware_version, |
| mode]) |
| log_root_dir = conf.log_root_dir |
| log_dir = os.path.join(log_root_dir, dir_basename) |
| latest_symlink = os.path.join(log_root_dir, 'latest') |
| |
| # Create the log directory. |
| try: |
| os.makedirs(log_dir) |
| except OSError, e: |
| print 'Error in create the directory (%s): %s' % (log_dir, e) |
| sys.exit(-1) |
| |
| # Set up the latest symbolic link to the newly created log directory. |
| try: |
| if os.path.islink(latest_symlink): |
| os.remove(latest_symlink) |
| os.symlink(log_dir, latest_symlink) |
| except OSError, e: |
| print 'Error in setup latest symlink (%s): %s' % (latest_symlink, e) |
| sys.exit(-1) |
| return log_dir |
| |
| |
| def stop_power_management(): |
| """Stop the power daemon management.""" |
| ret_d = common_util.simple_system('stop -q powerd') |
| if ret_d: |
| print 'Error in stopping powerd.' |
| print 'The screen may dim during the test.' |
| |
| |
| def start_power_management(): |
| """Start the power daemon management.""" |
| ret_d = common_util.simple_system('start -q powerd') |
| if ret_d: |
| print 'Error in starting powerd.' |
| print 'The screen may not go into suspend mode.' |
| print 'If this is a problem, you could reboot the machine.' |
| |
| |
| class GestureList: |
| """A class defines the gesture list.""" |
| |
| def __init__(self, gesture_names=None): |
| self.gesture_names = (gesture_names if gesture_names |
| else conf.gesture_names_complete) |
| |
| def get_gesture_list(self): |
| """Get the list of Gesture objects based on the gesture names.""" |
| gesture_dict = conf.get_gesture_dict() |
| gesture_list = [] |
| for name in self.gesture_names: |
| gesture = gesture_dict.get(name) |
| if gesture is None: |
| msg = 'Error: the gesture "%s" is not defined in the config.' |
| print msg % name |
| return [] |
| gesture_list.append(gesture) |
| return gesture_list |
| |
| |
| class Output: |
| """A class to handle outputs to the window and to the report.""" |
| def __init__(self, log_dir, report_name, win, report_html): |
| self.log_dir = log_dir |
| self.report_name = report_name |
| self.report = open(report_name, 'w') |
| self.win = win |
| self.prefix_space = ' ' * 4 |
| self.msg = None |
| self.report_html = report_html |
| |
| def __del__(self): |
| self.stop() |
| |
| def stop(self): |
| """Close the report file and print it on stdout.""" |
| self.report.close() |
| with open(self.report_name) as f: |
| for line in f.read().splitlines(): |
| print line |
| report_msg = '\n*** This test report is saved in the file: %s\n' |
| print report_msg % self.report_name |
| |
| def get_prefix_space(self): |
| """Get the prefix space when printing the report.""" |
| return self.prefix_space |
| |
| def print_report_line(self, msg): |
| """Print the line with proper indentation.""" |
| self.report.write(self.prefix_space + str(msg) + os.linesep) |
| |
| def print_window(self, msg): |
| """Print the message to the result window.""" |
| if type(msg) is list: |
| msg = os.linesep.join(msg) |
| self.win.set_result(msg) |
| print msg |
| |
| def _print_report(self, msg): |
| """Print the message to the report.""" |
| if type(msg) is list: |
| for line in msg: |
| self.print_report_line(line) |
| else: |
| self.print_report_line(msg) |
| |
| def buffer_report(self, msg): |
| """Buffer the message and print it later if not over-written. |
| |
| Usage of the method: the validator test result of a gesture may |
| be discarded because the user chooses to re-perform the gesture |
| again. So it should be able to over-write the message. |
| """ |
| self.msg = msg |
| |
| def flush_report(self): |
| """Print the buffered message if any.""" |
| if self.msg: |
| self._print_report(self.msg) |
| self.msg = None |
| |
| def print_report(self, msg): |
| """Print the message to the report.""" |
| # Print any buffered message first. |
| self.flush_report() |
| # Print this incoming message |
| self._print_report(msg) |
| |
| def print_all(self, msg): |
| """Print the message to both report and to the window.""" |
| self.print_window(msg) |
| self.buffer_report(msg) |
| |
| |
| class SimpleX: |
| """A simple class provides some simple X methods and properties.""" |
| |
| def __init__(self, win_name='aura'): |
| import Xlib |
| import Xlib.display |
| self.Xlib = Xlib |
| self.Xlib.display = Xlib.display |
| |
| self.disp = self._get_display() |
| self._get_screen() |
| self._get_window(win_name) |
| |
| def _get_display(self): |
| """Get the display object.""" |
| return self.Xlib.display.Display(get_display_name()) |
| |
| def _get_screen(self): |
| """Get the screen instance.""" |
| self.screen = self.disp.screen() |
| |
| def _get_window(self, win_name): |
| """Get the window with the specified name.""" |
| wins = self.screen.root.query_tree().children |
| for win in wins: |
| name = win.get_wm_name() |
| if name and win_name in name: |
| self.win = win |
| break |
| else: |
| self.win = None |
| print 'Error: No window is named as "%s".' % win_name |
| |
| def _get_focus_info(self): |
| """Get the input focus information.""" |
| return self.disp.get_input_focus() |
| |
| def set_input_focus(self): |
| """Set the input focus to the window id.""" |
| if self.win: |
| self.disp.set_input_focus(self.win.id, |
| self.Xlib.X.RevertToParent, |
| self.Xlib.X.CurrentTime) |
| self._get_focus_info() |
| |
| def get_screen_size(self): |
| """Get the screen size in pixels.""" |
| return (self.screen.width_in_pixels, self.screen.height_in_pixels) |
| |
| def get_screen_size_in_mms(self): |
| """Get the screen size in milli-meters.""" |
| return (self.screen.width_in_mms, self.screen.height_in_mss) |
| |
| def get_DPMM(self): |
| """Get Dots per Milli-meter.""" |
| return (1.0 * self.screen.width_in_pixels / self.screen.width_in_mms, |
| 1.0 * self.screen.height_in_pixels / self.screen.height_in_mms) |
| |
| def _recover_input_focus(self): |
| """Set the input focus back to the original settings.""" |
| self.disp.set_input_focus(self.Xlib.X.PointerRoot, |
| self.Xlib.X.RevertToParent, |
| self.Xlib.X.CurrentTime) |
| self._get_focus_info() |
| |
| def __del__(self): |
| self._recover_input_focus() |
| |
| |
| class ScreenShot: |
| """Handle screen shot.""" |
| |
| def __init__(self, geometry_str): |
| self.geometry_str = geometry_str |
| environment_str = 'DISPLAY=:0.0 XAUTHORITY=/home/chronos/.Xauthority ' |
| dump_util = '/usr/local/bin/import -quality 20' |
| self.dump_window_format = ' '.join([environment_str, dump_util, |
| '-window %s %s.png']) |
| self.dump_root_format = ' '.join([environment_str, dump_util, |
| '-window root -crop %s %s.png']) |
| self.get_id_cmd = 'DISPLAY=:0 xwininfo -root -tree' |
| |
| def dump_window(self, filename): |
| """Dump the screenshot of a window to the specified file name.""" |
| win_id = self._get_win_id() |
| if win_id: |
| dump_cmd = self.dump_window_format % (win_id, filename) |
| common_util.simple_system(dump_cmd) |
| else: |
| print 'Warning: cannot get the window id.' |
| |
| def dump_root(self, filename): |
| """Dump the screenshot of root to the specified file name.""" |
| dump_cmd = self.dump_root_format % (self.geometry_str, filename) |
| common_util.simple_system(dump_cmd) |
| |
| def _get_win_id(self): |
| """Get the window ID based on the characteristic string.""" |
| result = common_util.simple_system_output(self.get_id_cmd) |
| for line in result.splitlines(): |
| if self.geometry_str in line: |
| return line.split()[0].strip() |
| return None |
