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 eff4e90815261d5592404a5be5223e0b3cc74938..cd788415236c5fecbce50d4223625bf8207122b7 100644 |
| --- a/build/android/pylib/remote/device/remote_device_environment.py |
| +++ b/build/android/pylib/remote/device/remote_device_environment.py |
| @@ -6,6 +6,7 @@ |
| import logging |
| import os |
| +import random |
| import sys |
| from pylib import constants |
| @@ -60,6 +61,8 @@ class RemoteDeviceEnvironment(environment.Environment): |
| self._runner_package = args.runner_package |
| self._runner_type = args.runner_type |
| self._device = '' |
| + self._verbose_count = args.verbose_count |
| + |
| if not args.trigger and not args.collect: |
| self._trigger = True |
| self._collect = True |
| @@ -96,8 +99,10 @@ class RemoteDeviceEnvironment(environment.Environment): |
| def _GetAccessToken(self): |
| """Generates access token for remote device service.""" |
| logging.info('Generating remote service access token') |
| + remote_device_helper.SetLogging(self._verbose_count, logging.WARNING) |
|
jbudorick
2015/01/05 16:52:29
Could we have remote_device_helper provide this fu
rnephew (Wrong account)
2015/01/05 19:31:53
Done, except I moved it to appurify_sanitize becau
|
| access_token_results = appurify_sanitized.api.access_token_generate( |
| self._api_key, self._api_secret) |
| + remote_device_helper.SetLogging(self._verbose_count, logging.NOTSET) |
| remote_device_helper.TestHttpResponse(access_token_results, |
| 'Unable to generate access token.') |
| self._access_token = access_token_results.json()['response']['access_token'] |
| @@ -105,22 +110,35 @@ class RemoteDeviceEnvironment(environment.Environment): |
| def _RevokeAccessToken(self): |
| """Destroys access token for remote device service.""" |
| logging.info('Revoking remote service access token') |
| + remote_device_helper.SetLogging(self._verbose_count, logging.WARNING) |
| revoke_token_results = appurify_sanitized.api.access_token_revoke( |
| self._access_token) |
| + remote_device_helper.SetLogging(self._verbose_count, logging.NOTSET) |
| remote_device_helper.TestHttpResponse(revoke_token_results, |
| 'Unable to revoke access token.') |
| def _SelectDevice(self): |
| """Select which device to use.""" |
| - logging.info('Finding %s with %s to run tests on.' % |
| - (self._remote_device, self._remote_device_os)) |
| + logging.info('Finding device to run tests on.') |
| + remote_device_helper.SetLogging(self._verbose_count, logging.WARNING) |
| dev_list_res = appurify_sanitized.api.devices_list(self._access_token) |
| + remote_device_helper.SetLogging(self._verbose_count, logging.NOTSET) |
| remote_device_helper.TestHttpResponse(dev_list_res, |
| 'Unable to generate access token.') |
| device_list = dev_list_res.json()['response'] |
| + random.shuffle(device_list) |
| for device in device_list: |
| - if (device['name'] == self._remote_device |
| - and device['os_version'] == self._remote_device_os): |
| + if device['os_name'] != 'Android': |
| + continue |
| + if self._remote_device: |
| + if device['name'] != self._remote_device: |
|
jbudorick
2015/01/05 16:52:29
Can't this just be
if self._remote_device and d
rnephew (Wrong account)
2015/01/05 19:31:53
Done.
|
| + continue |
| + if self._remote_device_os: |
| + if device['os_version'] != self._remote_device_os: |
|
jbudorick
2015/01/05 16:52:29
(similar)
rnephew (Wrong account)
2015/01/05 19:31:53
Done.
|
| + continue |
| + if device['available_devices_count'] > 0: |
| + logging.info('Found device: %s %s' % |
| + (device['name'], device['os_version'])) |
| return device['device_type_id'] |
| self._NoDeviceFound(device_list) |
| @@ -139,8 +157,7 @@ class RemoteDeviceEnvironment(environment.Environment): |
| def _NoDeviceFound(self, device_list): |
| self._PrintAvailableDevices(device_list) |
| - raise remote_device_helper.RemoteDeviceError('No device found: %s %s' % |
| - (self._remote_device, self._remote_device_os)) |
| + raise remote_device_helper.RemoteDeviceError('No device found.') |
| @property |
| def device(self): |
| @@ -169,3 +186,7 @@ class RemoteDeviceEnvironment(environment.Environment): |
| @property |
| def collect(self): |
| return self._collect |
| + |
| + @property |
| + def verbose_count(self): |
| + return self._verbose_count |