Chromium Code Reviews| Index: build/android/pylib/remote/device/remote_device_environment.py |
| diff --git a/build/android/pylib/remote/device/remote_device_environment.py b/build/android/pylib/remote/device/remote_device_environment.py |
| index a701d9541e6bae2b07497e1e6937339266d2ba81..571302a11ea0fb4c8faeec3146d56a3037085fe1 100644 |
| --- a/build/android/pylib/remote/device/remote_device_environment.py |
| +++ b/build/android/pylib/remote/device/remote_device_environment.py |
| @@ -4,6 +4,8 @@ |
| """Environment setup and teardown for remote devices.""" |
| +from distutils import version |
|
jbudorick
2015/01/27 20:01:50
changed my mind, import distutils.version
I don't
rnephew (Wrong account)
2015/01/27 21:07:23
Done.
|
| +import json |
| import logging |
| import os |
| import random |
| @@ -32,35 +34,24 @@ class RemoteDeviceEnvironment(environment.Environment): |
| if args.api_key_file: |
| with open(args.api_key_file) as api_key_file: |
| self._api_key = api_key_file.read().strip() |
| - elif args.api_key: |
| - self._api_key = args.api_key |
| else: |
| - error_func('Must set api key with --api-key or --api-key-file') |
| + self._api_key = args.api_key |
| if args.api_secret_file: |
| with open(args.api_secret_file) as api_secret_file: |
| self._api_secret = api_secret_file.read().strip() |
| - elif args.api_secret: |
| - self._api_secret = args.api_secret |
| else: |
| - error_func('Must set api secret with --api-secret or --api-secret-file') |
| + self._api_secret = args.api_secret |
| - if not args.api_protocol: |
| - error_func('Must set api protocol with --api-protocol. Example: http') |
| self._api_protocol = args.api_protocol |
| - |
| - if not args.api_address: |
| - error_func('Must set api address with --api-address') |
| self._api_address = args.api_address |
| - |
| - if not args.api_port: |
| - error_func('Must set api port with --api-port.') |
| self._api_port = args.api_port |
| self._access_token = '' |
| self._results_path = args.results_path |
| self._remote_device = args.remote_device |
| self._remote_device_os = args.remote_device_os |
| + self._remote_device_minimum_os = args.remote_device_minimum_os |
| self._runner_package = args.runner_package |
| self._runner_type = args.runner_type |
| self._device = '' |
| @@ -73,6 +64,46 @@ class RemoteDeviceEnvironment(environment.Environment): |
| 'unknown': 60 * 5 |
| } |
| + if args.remote_device_file: |
| + with open(args.remote_device_file) as device_file: |
| + device_json = json.load(device_file) |
| + self._remote_device = device_json.get( |
| + 'remote_device', self._remote_device) |
| + self._remote_device_os = device_json.get( |
| + 'remote_device_os', self._remote_device_os) |
| + self._remote_device_minimum_os = device_json.get( |
| + 'remote_device_minimum_os', self._remote_device_minimum_os) |
| + self._api_address = device_json.get('api_address', self._api_address) |
| + self._api_secret = device_json.get('api_secret', self._api_secret) |
| + self._api_key = device_json.get('api_key', self._api_key) |
| + self._api_protocol = device_json.get('api_protocol', self._api_protocol) |
| + self._api_port = device_json.get('api_port', self._api_port) |
| + self._timeouts = device_json.get('timeouts', self._timeouts) |
| + |
| + if not self._api_address: |
| + error_func('Must set api address with --api-address' |
| + 'or in --remote-device-file.') |
| + if not self._api_protocol: |
| + error_func('Must set api protocol with --api-protocol' |
| + ' or in --remote-device-file. Example: http') |
| + if not self._api_key: |
| + error_func('Must set api key with --api-key, --api-key-file' |
| + 'or in --remote-device-file') |
| + if not self._api_secret: |
| + error_func('Must set api secret with --api-secret, --api-secret-file' |
| + 'or in --remote-device-file') |
| + if not self._api_port: |
| + error_func('Must set api port with --api-port' |
| + 'or in --remote-device-file') |
| + if not isinstance(self._timeouts, dict): |
| + error_func('Timeouts must be a dictionary.') |
| + if ('queueing' not in self._timeouts |
| + or 'installing' not in self._timeouts |
| + or 'in-progress' not in self._timeouts |
| + or 'unknown' not in self._timeouts) : |
| + error_func('Timeouts must contain queueing,' |
| + ' installing, in-progress, and unknown.') |
| + |
|
jbudorick
2015/01/27 20:01:50
Perhaps we should log the attributes here? (Minus
rnephew (Wrong account)
2015/01/27 21:07:23
Done.
|
| if not args.trigger and not args.collect: |
| self._trigger = True |
| self._collect = True |
| @@ -150,10 +181,14 @@ class RemoteDeviceEnvironment(environment.Environment): |
| for device in device_list: |
|
jbudorick
2015/01/27 20:01:50
It may be helpful to be able to select on OEM (I t
rnephew (Wrong account)
2015/01/27 21:07:23
Done.
|
| if device['os_name'] != self._device_type: |
| continue |
| - if self._remote_device and device['name'] != self._remote_device: |
| + if self._remote_device and device['name'] not in self._remote_device: |
| continue |
| if (self._remote_device_os |
| - and device['os_version'] != self._remote_device_os): |
| + and device['os_version'] not in self._remote_device_os): |
| + continue |
| + if (self._remote_device_minimum_os |
| + and version.LooseVersion(device['os_version']) |
| + < version.LooseVersion(self._remote_device_minimum_os)): |
| continue |
| if ((self._remote_device and self._remote_device_os) |
| or device['available_devices_count']): |