OLD | NEW |
1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 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 import os | 4 import os |
5 | 5 |
6 from telemetry import discover | 6 from telemetry.test import discover |
7 from telemetry import page_action | 7 from telemetry.page import page_action |
8 | 8 |
9 _page_action_classes = discover.Discover(os.path.dirname(__file__), | 9 _page_action_classes = discover.Discover( |
10 'action', | 10 os.path.dirname(__file__), |
11 page_action.PageAction, | 11 os.path.join(os.path.dirname(__file__), '..', '..'), |
12 import_error_should_raise=True) | 12 'action', |
| 13 page_action.PageAction, |
| 14 import_error_should_raise=True) |
13 | 15 |
14 def GetAllClasses(): | 16 def GetAllClasses(): |
15 return list(_page_action_classes.values()) | 17 return list(_page_action_classes.values()) |
16 | 18 |
17 def FindClassWithName(name): | 19 def FindClassWithName(name): |
18 return _page_action_classes.get(name) | 20 return _page_action_classes.get(name) |
19 | 21 |
20 def RegisterClassForTest(name, clazz): | 22 def RegisterClassForTest(name, clazz): |
21 _page_action_classes[name] = clazz | 23 _page_action_classes[name] = clazz |
OLD | NEW |