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 |