| # Copyright 2016 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 logging |
| import time |
| from contextlib import contextmanager |
| |
| from autotest_lib.client.common_lib import error |
| from autotest_lib.client.cros.chameleon import chameleon_port_finder |
| |
| |
| class _BaseChameleonMeasurer(object): |
| """Base class of performing measurement using Chameleon.""" |
| |
| _TIME_WAIT_FADE_OUT = 10 |
| _WAIT_TIME_LID_TRANSITION = 5 |
| |
| def __init__(self, cros_host, outputdir=None): |
| """Initializes the object.""" |
| raise NotImplementedError('_BaseChameleonMeasurer.__init__') |
| |
| |
| @contextmanager |
| def start_mirrored_mode_measurement(self): |
| """Starts the mirrored mode to measure. |
| |
| It iterates the connection ports between DUT and Chameleon and uses |
| the first port. Sets DUT into the mirrored mode. Then yields the |
| connected ports. |
| |
| It is used via a with statement, like the following: |
| |
| measurer = LocalChameleonMeasurer(cros_host, args, chrome) |
| with measurer.start_mirrored_mode_measurement() as chameleon_port: |
| # chameleon_port is automatically plugged before this line. |
| do_some_test_on(chameleon_port) |
| # chameleon_port is automatically unplugged after this line. |
| |
| @yields the first connected ChameleonVideoInput which is ensured plugged |
| before yielding. |
| |
| @raises TestFail if no connected video port. |
| """ |
| finder = chameleon_port_finder.ChameleonVideoInputFinder( |
| self.chameleon, self.display_facade) |
| with finder.use_first_port() as chameleon_port: |
| logging.info('Used Chameleon port: %s', |
| chameleon_port.get_connector_type()) |
| |
| logging.info('Setting to mirrored mode') |
| self.display_facade.set_mirrored(True) |
| |
| # Hide the typing cursor. |
| self.display_facade.hide_typing_cursor() |
| |
| # Sleep a while to wait the pop-up window faded-out. |
| time.sleep(self._TIME_WAIT_FADE_OUT) |
| |
| # Get the resolution to make sure Chameleon in a good state. |
| resolution = chameleon_port.get_resolution() |
| logging.info('Detected the resolution: %dx%d', *resolution) |
| |
| yield chameleon_port |
| |
| @contextmanager |
| def start_dock_mode_measurement(self): |
| """Starts the dock mode to measure. |
| |
| It iterates the connection ports between DUT and Chameleon and uses |
| the first port. Sets DUT into the dock mode. Then yields the |
| connected ports. |
| |
| It is used via a with statement, like the following: |
| |
| measurer = LocalChameleonMeasurer(cros_host, args, chrome) |
| with measurer.start_dock_mode_measurement() as chameleon_port: |
| # chameleon_port is automatically plugged before this line |
| # and lid is close to enter doc mode. |
| do_some_test_on(chameleon_port) |
| # chameleon_port is automatically unplugged after this line |
| # and lid is open again. |
| |
| @yields the first connected ChameleonVideoInput which is ensured plugged |
| before yielding. |
| |
| @raises TestFail if no connected video port or fail to enter dock mode. |
| """ |
| finder = chameleon_port_finder.ChameleonVideoInputFinder( |
| self.chameleon, self.display_facade) |
| try: |
| with finder.use_first_port() as chameleon_port: |
| logging.info('Used Chameleon port: %s', |
| chameleon_port.get_connector_type()) |
| |
| logging.info('Close lid to switch into dock mode...') |
| self.host.servo.lid_close() |
| time.sleep(self._WAIT_TIME_LID_TRANSITION) |
| |
| # Hide the typing cursor. |
| self.display_facade.hide_typing_cursor() |
| |
| # Sleep a while to wait the pop-up window faded-out. |
| time.sleep(self._TIME_WAIT_FADE_OUT) |
| |
| # Get the resolution to make sure Chameleon in a good state. |
| resolution = chameleon_port.get_resolution() |
| logging.info('Detected the resolution: %dx%d', *resolution) |
| |
| # Check if it is dock mode, no internal screen. |
| if self.display_facade.get_internal_resolution() is not None: |
| raise error.TestError('Failed to enter dock mode: ' |
| 'internal display still valid') |
| |
| yield chameleon_port |
| |
| finally: |
| logging.info('Open lid again...') |
| self.host.servo.lid_open() |