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
|
#!/usr/bin/env python3
# Copyright 2021 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import time
import unittest
import PRESUBMIT
class PresubmitError:
def __init__(self, message):
self.message = message
def __eq__(self, other):
return isinstance(other, PresubmitError) and self.message == other.message
def __repr__(self):
return 'PresubmitError({!r})'.format(self.message)
class TestCheckFreeze(unittest.TestCase):
def get_input_api(self, current_time, footers=None):
"""Get an input API to use for tests.
Args:
current_time - Current time expressed as seconds since the epoch.
"""
class FakeTime:
localtime = time.localtime
strftime = time.strftime
def time(self):
return float(current_time)
class FakeChange:
def GitFootersFromDescription(self):
return footers or []
class FakeInputApi:
time = FakeTime()
change = FakeChange()
no_diffs = False
return FakeInputApi()
def get_output_api(self):
class FakeOutputApi:
PresubmitError = PresubmitError
return FakeOutputApi
def test_before_freeze(self):
input_api = self.get_input_api(PRESUBMIT._FREEZE_START - 1)
output_api = self.get_output_api()
errors = PRESUBMIT.CheckFreeze(input_api, output_api)
self.assertEqual(errors, [])
def test_start_of_freeze(self):
input_api = self.get_input_api(PRESUBMIT._FREEZE_START + 1)
output_api = self.get_output_api()
errors = PRESUBMIT.CheckFreeze(input_api, output_api)
self.assertEqual(len(errors), 1)
self.assertTrue(
errors[0].message.startswith('There is a prod freeze in effect'))
def test_end_of_freeze(self):
input_api = self.get_input_api(PRESUBMIT._FREEZE_END - 1)
output_api = self.get_output_api()
errors = PRESUBMIT.CheckFreeze(input_api, output_api)
self.assertEqual(len(errors), 1)
self.assertTrue(
errors[0].message.startswith('There is a prod freeze in effect'))
def test_after_freeze(self):
input_api = self.get_input_api(PRESUBMIT._FREEZE_END + 1)
output_api = self.get_output_api()
errors = PRESUBMIT.CheckFreeze(input_api, output_api)
self.assertEqual(errors, [])
def test_ignore_freeze(self):
input_api = self.get_input_api(PRESUBMIT._FREEZE_START + 1,
footers={'Ignore-Freeze': 'testing'})
output_api = self.get_output_api()
errors = PRESUBMIT.CheckFreeze(input_api, output_api)
self.assertEqual(errors, [])
if __name__ == '__main__':
unittest.main()
|
