build/android/pylib/local/device/local_device_environment.py

   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