Add module discovery and autoloading to expect_tests.

scripts/slave/unittests/expect_tests/util.py

 # Copyright 2014 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.
 
 import inspect
 
 EXPECT_TESTS_COVER_FUNCTION = 'EXPECT_TESTS_COVER_FUNCTION'

[Vadim Sh., 2014/06/30 17:15:46]

constant for this seems to be an overkill :) but as you prefer

+EXPECT_TESTS_GENERATOR = 'EXPECT_TESTS_GENERATOR'
 
 def covers(coverage_path_function):
   """Allows annotation of a Test generator function with a function that will
   return a list of coverage patterns which should be enabled during the use of
   the Test generator function.
   """
   def _decorator(func):
     setattr(func, EXPECT_TESTS_COVER_FUNCTION, coverage_path_function)
     return func
   return _decorator
 
 
 def get_cover_list(test_gen_function):
   """Given a Test generator, return the list of coverage globs that should
   be included while executing the Test generator."""
-  if hasattr(test_gen_function, EXPECT_TESTS_COVER_FUNCTION):
-    # decorated with expect_tests.covers
-    return test_gen_function._covers()  # pylint: disable=W0212
-  else:
-    return [inspect.getabsfile(test_gen_function)]
+  return getattr(test_gen_function, EXPECT_TESTS_COVER_FUNCTION,
+                 lambda: [inspect.getabsfile(test_gen_function)])()
+
+
+def test_generator(function):
+  """Marks function as an expect_tests Test generator"""
+  setattr(function, EXPECT_TESTS_GENERATOR, True)
+  return function
+
+
+def is_test_generator(function):
+  return getattr(function, EXPECT_TESTS_GENERATOR, False)