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 |