Updates the test runner script exit codes

build/android/pylib/base/base_test_result.py

 # Copyright (c) 2013 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.
 
 """Module containing base test results classes."""
 
+from pylib import constants
+
 
 class ResultType(object):
   """Class enumerating test types."""
   PASS = 'PASS'
   FAIL = 'FAIL'
   CRASH = 'CRASH'
   TIMEOUT = 'TIMEOUT'
   UNKNOWN = 'UNKNOWN'
 
   @staticmethod
   def GetTypes():
     """Get a list of all test types."""
     return [ResultType.PASS, ResultType.FAIL, ResultType.CRASH,
             ResultType.TIMEOUT, ResultType.UNKNOWN]
 
 
 class BaseTestResult(object):
   """Base class for a single test result."""
+
   def __init__(self, name, test_type, log=''):
     """Construct a BaseTestResult.
 
     Args:
       name: Name of the test which defines uniqueness.
       test_type: Type of the test result as defined in ResultType.
       log: An optional string listing any errors.
     """
     assert name
     assert test_type in ResultType.GetTypes()
@@ skipping 22 matching lines @@
     """Get the test result type."""
     return self._test_type
 
   def GetLog(self):
     """Get the test log."""
     return self._log
 
 
 class TestRunResults(object):
   """Set of results for a test run."""
+
   def __init__(self):
     self._results = set()
+    self.exit_code = 0
 
   def GetLogs(self):
     """Get the string representation of all test logs."""
     s = []
     for test_type in ResultType.GetTypes():
       if test_type != ResultType.PASS:
         for t in sorted(self._GetType(test_type)):
           log = t.GetLog()
           if log:
             s.append('[%s] %s:' % (test_type, t))
@@ skipping 24 matching lines @@
     return self.GetLongForm()
 
   def AddResult(self, result):
     """Add |result| to the set.
 
     Args:
       result: An instance of BaseTestResult.
     """
     assert isinstance(result, BaseTestResult)
     self._results.add(result)
+    if result.GetType() not in [ResultType.PASS, ResultType.UNKNOWN]:
+      self.exit_code = constants.ERROR_EXIT_CODE
 
   def AddResults(self, results):
     """Add |results| to the set.
 
     Args:
       results: An iterable of BaseTestResult objects.
     """
     for t in results:
       self.AddResult(t)
 
   def AddTestRunResults(self, results):
     """Add the set of test results from |results|.
 
     Args:
       results: An instance of TestRunResults.
     """
     assert isinstance(results, TestRunResults)
     # pylint: disable=W0212
     self._results.update(results._results)
+    self.exit_code = self.exit_code or results.exit_code
 
   def GetAll(self):
     """Get the set of all test results."""
     return self._results.copy()
 
   def _GetType(self, test_type):
     """Get the set of test results with the given test type."""
     return set(t for t in self._results if t.GetType() == test_type)
 
   def GetPass(self):
@@ skipping 16 matching lines @@
     """Get the set of all unknown test results."""
     return self._GetType(ResultType.UNKNOWN)
 
   def GetNotPass(self):
     """Get the set of all non-passed test results."""
     return self.GetAll() - self.GetPass()
 
   def DidRunPass(self):
     """Return whether the test run was successful."""
     return not self.GetNotPass()