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 devil.android import device_blacklist
   6 from devil.android import device_errors
   7 from devil.android import device_utils
   8 from devil.utils import parallelizer
   9 from pylib.base import environment
  10
  11
  12 class LocalDeviceEnvironment(environment.Environment):
  13
  14   def __init__(self, args, _error_func):
  15     super(LocalDeviceEnvironment, self).__init__()
  16     self._blacklist = device_blacklist.Blacklist(
  17         args.blacklist_file or device_blacklist.BLACKLIST_JSON)
  18     self._device_serial = args.test_device
  19     self._devices = []
  20     self._max_tries = 1 + args.num_retries
  21     self._tool_name = args.tool
  22
  23   #override
  24   def SetUp(self):
  25     available_devices = device_utils.DeviceUtils.HealthyDevices(
  26         self._blacklist)
  27     if not available_devices:
  28       raise device_errors.NoDevicesError
  29     if self._device_serial:
  30       self._devices = [d for d in available_devices
  31                        if d.adb.GetDeviceSerial() == self._device_serial]
  32       if not self._devices:
  33         raise device_errors.DeviceUnreachableError(
  34             'Could not find device %r' % self._device_serial)
  35     else:
  36       self._devices = available_devices
  37
  38   @property
  39   def devices(self):
  40     return self._devices
  41
  42   @property
  43   def parallel_devices(self):
  44     return parallelizer.SyncParallelizer(self._devices)
  45
  46   @property
  47   def max_tries(self):
  48     return self._max_tries
  49
  50   @property
  51   def tool(self):
  52     return self._tool_name
  53
  54   #override
  55   def TearDown(self):
  56     pass
  57