| OLD | NEW |
| (Empty) |
| 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 | |
| 3 # found in the LICENSE file. | |
| 4 | |
| 5 import inspect | |
| 6 | |
| 7 EXPECT_TESTS_COVER_FUNCTION = 'EXPECT_TESTS_COVER_FUNCTION' | |
| 8 | |
| 9 def covers(coverage_path_function): | |
| 10 """Allows annotation of a Test generator function with a function that will | |
| 11 return a list of coverage patterns which should be enabled during the use of | |
| 12 the Test generator function. | |
| 13 """ | |
| 14 def _decorator(func): | |
| 15 setattr(func, EXPECT_TESTS_COVER_FUNCTION, coverage_path_function) | |
| 16 return func | |
| 17 return _decorator | |
| 18 | |
| 19 | |
| 20 def get_cover_list(test_gen_function): | |
| 21 """Given a Test generator, return the list of coverage globs that should | |
| 22 be included while executing the Test generator.""" | |
| 23 return getattr(test_gen_function, EXPECT_TESTS_COVER_FUNCTION, | |
| 24 lambda: [inspect.getabsfile(test_gen_function)])() | |
| OLD | NEW |