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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
"""Unit tests for timeout.py."""
import unittest
from buildscripts.timeouts import timeout as under_test
# pylint: disable=invalid-name
class CalculateTimeoutTest(unittest.TestCase):
def test_min_timeout(self):
self.assertEqual(under_test.MIN_TIMEOUT_SECONDS, under_test.calculate_timeout(15, 1))
def test_over_timeout_by_one_minute(self):
self.assertEqual(360, under_test.calculate_timeout(301, 1))
def test_float_runtimes(self):
self.assertEqual(360, under_test.calculate_timeout(300.14, 1))
def test_scaling_factor(self):
avg_runtime = 30
scaling_factor = 10
self.assertEqual(avg_runtime * scaling_factor + 60,
under_test.calculate_timeout(avg_runtime, scaling_factor))
class TimeoutEstimateTest(unittest.TestCase):
def test_too_high_a_timeout_raises_errors(self):
timeout_est = under_test.TimeoutEstimate(
max_test_runtime=5, expected_task_runtime=under_test.MAX_EXPECTED_TIMEOUT)
with self.assertRaises(ValueError):
timeout_est.generate_timeout_cmd(is_patch=True, repeat_factor=1)
def test_is_specified_should_return_true_when_a_test_runtime_is_specified(self):
timeout_est = under_test.TimeoutEstimate(max_test_runtime=3.14, expected_task_runtime=None)
self.assertTrue(timeout_est.is_specified())
def test_is_specified_should_return_true_when_a_task_runtime_is_specified(self):
timeout_est = under_test.TimeoutEstimate(max_test_runtime=None, expected_task_runtime=3.14)
self.assertTrue(timeout_est.is_specified())
def test_is_specified_should_return_false_when_no_data_is_specified(self):
timeout_est = under_test.TimeoutEstimate(max_test_runtime=None, expected_task_runtime=None)
self.assertFalse(timeout_est.is_specified())
class TestGenerateTimeoutCmd(unittest.TestCase):
def test_evg_config_does_not_fails_if_test_timeout_too_high_on_mainline(self):
timeout = under_test.TimeoutEstimate(max_test_runtime=under_test.MAX_EXPECTED_TIMEOUT + 1,
expected_task_runtime=None)
time_cmd = timeout.generate_timeout_cmd(is_patch=False, repeat_factor=1)
self.assertGreater(time_cmd.timeout, under_test.MAX_EXPECTED_TIMEOUT)
def test_evg_config_does_not_fails_if_task_timeout_too_high_on_mainline(self):
timeout = under_test.TimeoutEstimate(
expected_task_runtime=under_test.MAX_EXPECTED_TIMEOUT + 1, max_test_runtime=None)
time_cmd = timeout.generate_timeout_cmd(is_patch=False, repeat_factor=1)
self.assertGreater(time_cmd.exec_timeout, under_test.MAX_EXPECTED_TIMEOUT)
class TestTimeoutInfo(unittest.TestCase):
def test_default_timeout(self):
timeout_info = under_test.TimeoutInfo.default_timeout()
self.assertIsNone(timeout_info.cmd)
def test_timeout_only_set(self):
timeout = 5
timeout_info = under_test.TimeoutInfo.overridden(timeout=timeout)
cmd = timeout_info.cmd.as_dict()
self.assertEqual("timeout.update", cmd["command"])
self.assertEqual(timeout, cmd["params"]["timeout_secs"])
self.assertNotIn("exec_timeout_secs", cmd["params"])
def test_exec_timeout_only_set(self):
exec_timeout = 5
timeout_info = under_test.TimeoutInfo.overridden(exec_timeout=exec_timeout)
cmd = timeout_info.cmd.as_dict()
self.assertEqual("timeout.update", cmd["command"])
self.assertEqual(exec_timeout, cmd["params"]["exec_timeout_secs"])
self.assertNotIn("timeout_secs", cmd["params"])
def test_both_timeouts_set(self):
timeout = 3
exec_timeout = 5
timeout_info = under_test.TimeoutInfo.overridden(exec_timeout=exec_timeout, timeout=timeout)
cmd = timeout_info.cmd.as_dict()
self.assertEqual("timeout.update", cmd["command"])
self.assertEqual(exec_timeout, cmd["params"]["exec_timeout_secs"])
self.assertEqual(timeout, cmd["params"]["timeout_secs"])
def test_override_with_no_values(self):
with self.assertRaises(ValueError):
under_test.TimeoutInfo.overridden()
|
