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