blob: 052864c9180ca988ec59d136d3934453ff111ee5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
"""Queue entry interface."""
class QueueElemRepeatNum(object):
"""Base class for an element on the queue."""
def __init__(self, testcase, test_config, suite_options):
"""Initialize QueueElemRepeatNum class."""
self.testcase = testcase
self.test_config = test_config
self.repeat_num_min = suite_options.num_repeat_tests
self.repeat_time_elapsed = 0
self.repeat_num = 0
def job_completed(self, job_time):
"""Increment values when the job completes."""
self.repeat_num += 1
self.repeat_time_elapsed += job_time
def should_requeue(self):
"""Return True if the queue element should be requeued."""
return self.repeat_num < self.repeat_num_min
class QueueElemRepeatTime(QueueElemRepeatNum):
"""Queue element for repeat time."""
def __init__(self, testcase, config, suite_options):
"""Initialize QueueElemRepeatTime class."""
super(QueueElemRepeatTime, self).__init__(testcase, config, suite_options)
self.repeat_num_min = suite_options.num_repeat_tests_min
self.repeat_num_max = suite_options.num_repeat_tests_max
self.repeat_secs = suite_options.time_repeat_tests_secs
def should_requeue(self):
"""Return True if the queue element should be requeued."""
avg_time = 0 if self.repeat_num == 0 else self.repeat_time_elapsed / self.repeat_num
# Minumim number of tests has not been run.
if self.repeat_num_min and self.repeat_num < self.repeat_num_min:
return True
# Maximum number of tests has been run or elapsed time has been exceeded.
if ((self.repeat_num_max and self.repeat_num >= self.repeat_num_max)
or self.repeat_time_elapsed + avg_time > self.repeat_secs):
return False
return True
|
