Index: chrome/test/webdriver/test/py_unittest_util.py |
diff --git a/chrome/test/webdriver/test/py_unittest_util.py b/chrome/test/webdriver/test/py_unittest_util.py |
index 13d0e5b9073ec75954ae93de0ce68808d2ee07ea..d650c781299771bf89c2b7e028d3a502e66b334b 100644 |
--- a/chrome/test/webdriver/test/py_unittest_util.py |
+++ b/chrome/test/webdriver/test/py_unittest_util.py |
@@ -23,6 +23,7 @@ class _GTestTextTestResult(unittest._TextTestResult): |
""" |
def __init__(self, stream, descriptions, verbosity): |
unittest._TextTestResult.__init__(self, stream, descriptions, verbosity) |
+ self._fails = set() |
def _GetTestURI(self, test): |
if sys.version_info[:2] <= (2, 4): |
@@ -44,10 +45,15 @@ class _GTestTextTestResult(unittest._TextTestResult): |
def addError(self, test, err): |
unittest.TestResult.addError(self, test, err) |
self.stream.writeln('[ ERROR ] %s' % self._GetTestURI(test)) |
+ self._fails.add(self._GetTestURI(test)) |
def addFailure(self, test, err): |
unittest.TestResult.addFailure(self, test, err) |
self.stream.writeln('[ FAILED ] %s' % self._GetTestURI(test)) |
+ self._fails.add(self._GetTestURI(test)) |
+ |
+ def getRetestFilter(self): |
+ return ':'.join(self._fails) |
class GTestTextTestRunner(unittest.TextTestRunner): |