OLD | NEW |
| (Empty) |
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 | |
3 # found in the LICENSE file. | |
4 | |
5 """Annotations for host-driven tests.""" | |
6 # pylint: disable=W0212 | |
7 | |
8 import os | |
9 | |
10 | |
11 class AnnotatedFunctions(object): | |
12 """A container for annotated methods.""" | |
13 _ANNOTATED = {} | |
14 | |
15 @staticmethod | |
16 def _AddFunction(annotation, function): | |
17 """Adds an annotated function to our container. | |
18 | |
19 Args: | |
20 annotation: the annotation string. | |
21 function: the function. | |
22 Returns: | |
23 The function passed in. | |
24 """ | |
25 module_name = os.path.splitext(os.path.basename( | |
26 function.__globals__['__file__']))[0] | |
27 qualified_function_name = '.'.join([module_name, function.func_name]) | |
28 function_list = AnnotatedFunctions._ANNOTATED.get(annotation, []) | |
29 function_list.append(qualified_function_name) | |
30 AnnotatedFunctions._ANNOTATED[annotation] = function_list | |
31 return function | |
32 | |
33 @staticmethod | |
34 def IsAnnotated(annotation, qualified_function_name): | |
35 """True if function name (module.function) contains the annotation. | |
36 | |
37 Args: | |
38 annotation: the annotation string. | |
39 qualified_function_name: the qualified function name. | |
40 Returns: | |
41 True if module.function contains the annotation. | |
42 """ | |
43 return qualified_function_name in AnnotatedFunctions._ANNOTATED.get( | |
44 annotation, []) | |
45 | |
46 @staticmethod | |
47 def GetTestAnnotations(qualified_function_name): | |
48 """Returns a list containing all annotations for the given function. | |
49 | |
50 Args: | |
51 qualified_function_name: the qualified function name. | |
52 Returns: | |
53 List of all annotations for this function. | |
54 """ | |
55 return [annotation | |
56 for annotation, tests in AnnotatedFunctions._ANNOTATED.iteritems() | |
57 if qualified_function_name in tests] | |
58 | |
59 | |
60 # The following functions are annotations used for the host-driven tests. | |
61 def Smoke(function): | |
62 return AnnotatedFunctions._AddFunction('Smoke', function) | |
63 | |
64 | |
65 def SmallTest(function): | |
66 return AnnotatedFunctions._AddFunction('SmallTest', function) | |
67 | |
68 | |
69 def MediumTest(function): | |
70 return AnnotatedFunctions._AddFunction('MediumTest', function) | |
71 | |
72 | |
73 def LargeTest(function): | |
74 return AnnotatedFunctions._AddFunction('LargeTest', function) | |
75 | |
76 | |
77 def EnormousTest(function): | |
78 return AnnotatedFunctions._AddFunction('EnormousTest', function) | |
79 | |
80 | |
81 def FlakyTest(function): | |
82 return AnnotatedFunctions._AddFunction('FlakyTest', function) | |
83 | |
84 | |
85 def DisabledTest(function): | |
86 return AnnotatedFunctions._AddFunction('DisabledTest', function) | |
87 | |
88 | |
89 def Feature(feature_list): | |
90 def _AddFeatures(function): | |
91 for feature in feature_list: | |
92 AnnotatedFunctions._AddFunction('Feature:%s' % feature, function) | |
93 return AnnotatedFunctions._AddFunction('Feature', function) | |
94 return _AddFeatures | |
OLD | NEW |