blob: d085c55553f6581f3f9713948c0d01c4900a7371 [file] [log] [blame]
# Copyright 2018 the V8 project 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 collections
from . import base
from .result import RerunResult
class RerunProc(base.TestProcProducer):
def __init__(self, rerun_max, rerun_max_total=None):
super(RerunProc, self).__init__('Rerun')
self._requirement = base.DROP_OUTPUT
self._rerun = {}
self._results = collections.defaultdict(list)
self._rerun_max = rerun_max
self._rerun_total_left = rerun_max_total
def _next_test(self, test):
return self._send_next_subtest(test)
def _result_for(self, test, subtest, result):
# First result
if subtest.procid[-2:] == '-1':
# Passed, no reruns
if not result.has_unexpected_output:
self._send_result(test, result)
return
self._rerun[test.procid] = 0
results = self._results[test.procid]
results.append(result)
if not self.is_stopped and self._needs_rerun(test, result):
self._rerun[test.procid] += 1
if self._rerun_total_left is not None:
self._rerun_total_left -= 1
self._send_next_subtest(test, self._rerun[test.procid])
else:
result = RerunResult.create(results)
self._finalize_test(test)
self._send_result(test, result)
def _needs_rerun(self, test, result):
# TODO(majeski): Limit reruns count for slow tests.
return ((self._rerun_total_left is None or self._rerun_total_left > 0) and
self._rerun[test.procid] < self._rerun_max and
result.has_unexpected_output)
def _send_next_subtest(self, test, run=0):
subtest = self._create_subtest(test, str(run + 1), keep_output=(run != 0))
return self._send_test(subtest)
def _finalize_test(self, test):
del self._rerun[test.procid]
del self._results[test.procid]