| # Copyright 2014 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| from pylib import constants |
| from pylib.local.device import local_device_environment |
| try: |
| from pylib.remote.device import remote_device_environment |
| except ImportError: |
| remote_device_environment = None |
| |
| def CreateEnvironment(args, error_func): |
| |
| if args.environment == 'local': |
| if args.command not in constants.LOCAL_MACHINE_TESTS: |
| return local_device_environment.LocalDeviceEnvironment(args, error_func) |
| # TODO(jbudorick) Add local machine environment. |
| if args.environment == 'remote_device' and remote_device_environment: |
| return remote_device_environment.RemoteDeviceEnvironment(args, |
| error_func) |
| error_func('Unable to create %s environment.' % args.environment) |
