# Lint as: python2, python3 # Copyright 2018 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 contextlib import json import logging from lxml import etree import os import six import time from autotest_lib.client.common_lib import error, utils from autotest_lib.server.cros.tradefed import tradefed_chromelogin as login class ChartFixture: """Sets up chart tablet to display dummy scene image.""" DISPLAY_SCRIPT = '/usr/local/autotest/bin/display_chart.py' OUTPUT_LOG = '/tmp/chart_service.log' def __init__(self, chart_host, scene_uri): self.host = chart_host self.scene_uri = scene_uri self.display_pid = None self.host.run(['rm', '-f', self.OUTPUT_LOG]) def initialize(self): """Prepare scene file and display it on chart host.""" logging.info('Prepare scene file') tmpdir = self.host.get_tmp_dir() scene_path = os.path.join( tmpdir, self.scene_uri[self.scene_uri.rfind('/') + 1:]) self.host.run('wget', args=('-O', scene_path, self.scene_uri)) logging.info('Display scene file') self.display_pid = self.host.run_background( 'python2 {script} {scene} >{log} 2>&1'.format( script=self.DISPLAY_SCRIPT, scene=scene_path, log=self.OUTPUT_LOG)) logging.info( 'Poll for "is ready" message for ensuring chart is ready.') timeout = 60 poll_time_step = 0.1 while timeout > 0: if self.host.run( 'grep', args=('-q', 'Chart is ready.', self.OUTPUT_LOG), ignore_status=True).exit_status == 0: break time.sleep(poll_time_step) timeout -= poll_time_step else: raise error.TestError('Timeout waiting for chart ready') def cleanup(self): """Cleanup display script.""" if self.display_pid is not None: self.host.run( 'kill', args=('-2', str(self.display_pid)), ignore_status=True) self.host.get_file(self.OUTPUT_LOG, '.') def get_chart_address(host_address, args): """Get address of chart tablet from commandline args or mapping logic in test lab. @param host_address: a list of hostname strings. @param args: a dict parse from commandline args. @return: A list of strings for chart tablet addresses. """ address = utils.args_to_dict(args).get('chart') if address is not None: return address.split(',') elif utils.is_in_container(): return [utils.get_lab_chart_address(host) for host in host_address] else: return None class DUTFixture: """Sets up camera filter for target camera facing on DUT.""" TEST_CONFIG_PATH = '/var/cache/camera/test_config.json' CAMERA_PROFILE_PATH = ('/mnt/stateful_partition/encrypted/var/cache/camera' '/media_profiles.xml') CAMERA_SCENE_LOG = '/tmp/scene.jpg' def __init__(self, test, host, facing): self.test = test self.host = host self.facing = facing @contextlib.contextmanager def _set_selinux_permissive(self): selinux_mode = self.host.run_output('getenforce') self.host.run('setenforce 0') yield self.host.run('setenforce', args=(selinux_mode, )) def _filter_camera_profile(self, content, facing): """Filter camera profile of target facing from content of camera profile. @return: New camera profile with only target facing, camera ids are renumbered from 0. """ tree = etree.parse( six.StringIO(content), parser=etree.XMLParser(compact=False)) root = tree.getroot() profiles = root.findall('CamcorderProfiles') logging.debug('%d number of camera(s) found in camera profile', len(profiles)) assert 1 <= len(profiles) <= 2 if len(profiles) == 2: cam_id = 0 if facing == 'back' else 1 for p in profiles: if cam_id == int(p.attrib['cameraId']): p.attrib['cameraId'] = '0' else: root.remove(p) else: with login.login_chrome( hosts=[self.host], board=self.test._get_board_name(), ), self._set_selinux_permissive(): has_front_camera = ( 'feature:android.hardware.camera.front' in self.host. run_output('android-sh -c "pm list features"')) logging.debug('has_front_camera=%s', has_front_camera) if (facing == 'front') != has_front_camera: root.remove(profiles[0]) return etree.tostring( tree, xml_declaration=True, encoding=tree.docinfo.encoding) def _read_file(self, filepath): """Read content of filepath from host.""" tmp_path = os.path.join(self.test.tmpdir, os.path.basename(filepath)) self.host.get_file(filepath, tmp_path, delete_dest=True) with open(tmp_path) as f: return f.read() def _write_file(self, filepath, content, permission=None, owner=None): """Write content to filepath on remote host. @param permission: set permission to 0xxx octal number of remote file. @param owner: set owner of remote file. """ tmp_path = os.path.join(self.test.tmpdir, os.path.basename(filepath)) with open(tmp_path, 'w') as f: f.write(content) if permission is not None: os.chmod(tmp_path, permission) self.host.send_file(tmp_path, filepath, delete_dest=True) if owner is not None: self.host.run('chown', args=(owner, filepath)) def initialize(self): """Filter out camera other than target facing on DUT.""" logging.info('Restart camera service with filter option') self._write_file( self.TEST_CONFIG_PATH, json.dumps({ 'enable_back_camera': self.facing == 'back', 'enable_front_camera': self.facing == 'front', 'enable_external_camera': False }), owner='arc-camera') self.host.upstart_restart('cros-camera') logging.info('Replace camera profile in ARC++ container') profile = self._read_file(self.CAMERA_PROFILE_PATH) new_profile = self._filter_camera_profile(profile, self.facing) self._write_file(self.CAMERA_PROFILE_PATH, new_profile) self.host.run('restart ui') @contextlib.contextmanager def _stop_camera_service(self): self.host.upstart_stop('cros-camera') yield self.host.upstart_restart('cros-camera') def log_camera_scene(self): """Capture an image from camera as the log for debugging scene related problem.""" gtest_filter = ( 'Camera3StillCaptureTest/' 'Camera3DumpSimpleStillCaptureTest.DumpCaptureResult/0') with self._stop_camera_service(): self.host.run( 'sudo', args=('--user=arc-camera', 'cros_camera_test', '--gtest_filter=' + gtest_filter, '--camera_facing=' + self.facing, '--dump_still_capture_path=' + self.CAMERA_SCENE_LOG)) self.host.get_file(self.CAMERA_SCENE_LOG, '.') def cleanup(self): """Cleanup camera filter.""" logging.info('Remove filter option and restore camera service') self.host.run('rm', args=('-f', self.TEST_CONFIG_PATH)) self.host.upstart_restart('cros-camera') logging.info('Restore camera profile in ARC++ container') self.host.run('restart ui')