| 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.gtest import gtest_test_instance | 5 from pylib.gtest import gtest_test_instance |
| 6 from pylib.gtest import local_device_gtest_run | 6 from pylib.gtest import local_device_gtest_run |
| 7 from pylib.instrumentation import instrumentation_test_instance |
| 7 from pylib.local.device import local_device_environment | 8 from pylib.local.device import local_device_environment |
| 9 from pylib.local.device import local_device_instrumentation_test_run |
| 8 from pylib.remote.device import remote_device_environment | 10 from pylib.remote.device import remote_device_environment |
| 9 from pylib.remote.device import remote_device_gtest_run | 11 from pylib.remote.device import remote_device_gtest_run |
| 10 from pylib.remote.device import remote_device_uirobot_run | 12 from pylib.remote.device import remote_device_uirobot_run |
| 11 from pylib.uirobot import uirobot_test_instance | 13 from pylib.uirobot import uirobot_test_instance |
| 12 | 14 |
| 13 | 15 |
| 14 def CreateTestRun(_args, env, test_instance, error_func): | 16 def CreateTestRun(_args, env, test_instance, error_func): |
| 15 if isinstance(env, local_device_environment.LocalDeviceEnvironment): | 17 if isinstance(env, local_device_environment.LocalDeviceEnvironment): |
| 16 if isinstance(test_instance, gtest_test_instance.GtestTestInstance): | 18 if isinstance(test_instance, gtest_test_instance.GtestTestInstance): |
| 17 return local_device_gtest_run.LocalDeviceGtestRun(env, test_instance) | 19 return local_device_gtest_run.LocalDeviceGtestRun(env, test_instance) |
| 20 if isinstance(test_instance, |
| 21 instrumentation_test_instance.InstrumentationTestInstance): |
| 22 return (local_device_instrumentation_test_run |
| 23 .LocalDeviceInstrumentationTestRun(env, test_instance)) |
| 18 | 24 |
| 19 if isinstance(env, remote_device_environment.RemoteDeviceEnvironment): | 25 if isinstance(env, remote_device_environment.RemoteDeviceEnvironment): |
| 20 if isinstance(test_instance, gtest_test_instance.GtestTestInstance): | 26 if isinstance(test_instance, gtest_test_instance.GtestTestInstance): |
| 21 return remote_device_gtest_run.RemoteDeviceGtestRun(env, test_instance) | 27 return remote_device_gtest_run.RemoteDeviceGtestRun(env, test_instance) |
| 28 # TODO(rnephew): Add remote_device instrumentation test runs. |
| 22 if isinstance(test_instance, uirobot_test_instance.UirobotTestInstance): | 29 if isinstance(test_instance, uirobot_test_instance.UirobotTestInstance): |
| 23 return remote_device_uirobot_run.RemoteDeviceUirobotRun( | 30 return remote_device_uirobot_run.RemoteDeviceUirobotRun( |
| 24 env, test_instance) | 31 env, test_instance) |
| 25 | 32 |
| 26 # TODO(jbudorick): Add local instrumentation test runs. | |
| 27 # TODO(rnephew): Add remote_device instrumentation test runs. | |
| 28 | 33 |
| 29 error_func('Unable to create test run for %s tests in %s environment' | 34 error_func('Unable to create test run for %s tests in %s environment' |
| 30 % (str(test_instance), str(env))) | 35 % (str(test_instance), str(env))) |
| 31 | 36 |
| OLD | NEW |