| OLD | NEW |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 from pylib import constants | 5 from pylib import constants |
| 6 from pylib.local.device import local_device_environment | 6 from pylib.local.device import local_device_environment |
| 7 try: | |
| 8 from pylib.remote.device import remote_device_environment | |
| 9 except ImportError: | |
| 10 remote_device_environment = None | |
| 11 | 7 |
| 12 def CreateEnvironment(args, error_func): | 8 def CreateEnvironment(args, error_func): |
| 13 | 9 |
| 14 if args.environment == 'local': | 10 if args.environment == 'local': |
| 15 if args.command not in constants.LOCAL_MACHINE_TESTS: | 11 if args.command not in constants.LOCAL_MACHINE_TESTS: |
| 16 return local_device_environment.LocalDeviceEnvironment(args, error_func) | 12 return local_device_environment.LocalDeviceEnvironment(args, error_func) |
| 17 # TODO(jbudorick) Add local machine environment. | 13 # TODO(jbudorick) Add local machine environment. |
| 18 if args.environment == 'remote_device' and remote_device_environment: | 14 |
| 19 return remote_device_environment.RemoteDeviceEnvironment(args, | |
| 20 error_func) | |
| 21 error_func('Unable to create %s environment.' % args.environment) | 15 error_func('Unable to create %s environment.' % args.environment) |
| OLD | NEW |