Simplify objects passed to the result stage

expect_tests/type_definitions.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
 import os
 import re
 
 from collections import namedtuple
 
@@ skipping 133 matching lines @@
     assert f.fully_bound
     return f.func(*f.args, **f.kwargs)
 
   def __repr__(self):
     return 'FuncCall(%r, *%r, **%r)' % (self.func, self.args, self.kwargs)
 
 
 _Test = namedtuple(
     'Test', 'name func_call expect_dir expect_base ext covers breakpoints')
 
+TestInfo = namedtuple(
+  'TestInfo', 'name  expect_dir expect_base ext')
+
+
 class Test(_Test):
   TEST_COVERS_MATCH = re.compile('.*/test/([^/]*)_test\.py$')
 
   def __new__(cls, name, func_call, expect_dir=None, expect_base=None,
               ext='json', covers=None, breakpoints=None, break_funcs=()):
     """Create a new test.
 
     @param name: The name of the test. Will be used as the default expect_base
 
     @param func_call: A FuncCall object
@@ skipping 74 matching lines @@
     """
     yield self, func(self.bind(context=None))
 
   def bind(self, *args, **kwargs):
     return self._replace(func_call=self.func_call.bind(*args, **kwargs))
 
   def restrict(self, tests):
     assert tests[0] is self
     return self
 
+  def get_info(self):
+    """Strips test instance of hard-to-pickle stuff
+
+    Returns a TestInfo instance.
+    """
+    return TestInfo(self.name, self.expect_dir, self.expect_base, self.ext)
+
 
 _MultiTest = namedtuple(
     'MultiTest', 'name make_ctx_call destroy_ctx_call tests atomic')
 
+MultiTestInfo = namedtuple('MultiTestInfo', 'name tests atomic')
+
+
 class MultiTest(_MultiTest):
   """A wrapper around one or more Test instances.
 
   Allows the entire group to have common pre- and post- actions and an optional
   shared context between the Test methods (represented by Bind(name='context')).
 
   Args:
     name - The name of the MultiTest. Each Test's name should be prefixed with
         this name, though this is not enforced.
     make_ctx_call - A FuncCall which will be called once before any test in this
@@ skipping 41 matching lines @@
     try:
       for test in self.tests:
         yield test, func(test.bind(context=ctx_object))
     finally:
       self.destroy_ctx_call.bind(context=ctx_object)()
 
   @staticmethod
   def expect_path(_ext=None):
     return None
 
+  def get_info(self):
+    """Strips MultiTest instance of hard-to-pickle stuff
+
+    Returns a MultiTestInfo instance.
+    """
+    all_tests = [test.get_restricted() for test in self.tests]
+    test = MultiTestInfo(name=self.name,
+                         tests=all_tests,
+                         atomic=self.atomic
+                         )
+    return test
+
 
 class Handler(object):
   """Handler object.
 
   Defines 3 handler methods for each stage of the test pipeline. The pipeline
   looks like:
 
                          ->           ->
                          ->    jobs   ->                   (main)
   GenStage -> test_queue ->      *    -> result_queue -> ResultStage
@@ skipping 102 matching lines @@
         print 'UNHANDLED:', obj
       return Failure()
 
     def finalize(self, aborted):
       """Called after __call__() has been called for all results.
 
       @param aborted: True if the user aborted the run.
       @type aborted: bool
       """
       pass