| OLD | NEW |
| (Empty) |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 # Use of this source code is governed by a BSD-style license that can be | |
| 3 # found in the LICENSE file. | |
| 4 | |
| 5 from pylib.base import environment | |
| 6 from pylib.device import adb_wrapper | |
| 7 from pylib.device import device_errors | |
| 8 from pylib.device import device_utils | |
| 9 from pylib.utils import parallelizer | |
| 10 | |
| 11 | |
| 12 class LocalDeviceEnvironment(environment.Environment): | |
| 13 | |
| 14 def __init__(self, args, _error_func): | |
| 15 super(LocalDeviceEnvironment, self).__init__() | |
| 16 self._device_serial = args.test_device | |
| 17 self._devices = [] | |
| 18 self._max_tries = 1 + args.num_retries | |
| 19 self._tool_name = args.tool | |
| 20 | |
| 21 #override | |
| 22 def SetUp(self): | |
| 23 available_devices = device_utils.DeviceUtils.HealthyDevices() | |
| 24 if not available_devices: | |
| 25 raise device_errors.NoDevicesError | |
| 26 if self._device_serial: | |
| 27 self._devices = [d for d in available_devices | |
| 28 if d.adb.GetDeviceSerial() == self._device_serial] | |
| 29 if not self._devices: | |
| 30 raise device_errors.DeviceUnreachableError( | |
| 31 'Could not find device %r' % self._device_serial) | |
| 32 else: | |
| 33 self._devices = available_devices | |
| 34 | |
| 35 @property | |
| 36 def devices(self): | |
| 37 return self._devices | |
| 38 | |
| 39 @property | |
| 40 def parallel_devices(self): | |
| 41 return parallelizer.SyncParallelizer(self._devices) | |
| 42 | |
| 43 @property | |
| 44 def max_tries(self): | |
| 45 return self._max_tries | |
| 46 | |
| 47 @property | |
| 48 def tool(self): | |
| 49 return self._tool_name | |
| 50 | |
| 51 #override | |
| 52 def TearDown(self): | |
| 53 pass | |
| 54 | |
| OLD | NEW |