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.gtest import gtest_test_instance | 6 from pylib.gtest import gtest_test_instance |
| 7 from pylib.instrumentation import instrumentation_test_instance |
7 from pylib.utils import isolator | 8 from pylib.utils import isolator |
8 from pylib.uirobot import uirobot_test_instance | 9 from pylib.uirobot import uirobot_test_instance |
9 | 10 |
10 | 11 |
11 | 12 |
12 def CreateTestInstance(args, error_func): | 13 def CreateTestInstance(args, error_func): |
13 | 14 |
14 if args.command == 'gtest': | 15 if args.command == 'gtest': |
15 return gtest_test_instance.GtestTestInstance( | 16 return gtest_test_instance.GtestTestInstance( |
16 args, isolator.Isolator(constants.ISOLATE_DEPS_DIR), error_func) | 17 args, isolator.Isolator(constants.ISOLATE_DEPS_DIR), error_func) |
17 if args.command == 'uirobot': | 18 elif args.command == 'instrumentation': |
| 19 return instrumentation_test_instance.InstrumentationTestInstance( |
| 20 args, isolator.Isolator(constants.ISOLATE_DEPS_DIR), error_func) |
| 21 elif args.command == 'uirobot': |
18 return uirobot_test_instance.UirobotTestInstance(args) | 22 return uirobot_test_instance.UirobotTestInstance(args) |
19 # TODO(jbudorick) Add instrumentation test instance. | 23 |
20 | 24 |
21 error_func('Unable to create %s test instance.' % args.command) | 25 error_func('Unable to create %s test instance.' % args.command) |
OLD | NEW |