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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
import unittest
from unittest import mock
from mozunit import main
from gecko_taskgraph.util import partials
release_blob = {
"fileUrls": {
"release-localtest": {
"completes": {
"*": "%OS_FTP%/%LOCALE%/firefox-92.0.1.complete.mar",
}
}
},
"platforms": {
"WINNT_x86_64-msvc": {
"locales": {
"en-US": {
"buildID": "20210922161155",
}
}
}
},
}
def nightly_blob(release):
return {
"platforms": {
"WINNT_x86_64-msvc": {
"locales": {
"en-US": {
"buildID": release[-14:],
"completes": [{"fileUrl": release}],
}
}
}
}
}
class TestReleaseHistory(unittest.TestCase):
@mock.patch("gecko_taskgraph.util.partials.get_release_builds")
@mock.patch("gecko_taskgraph.util.partials.get_sorted_releases")
def test_populate_release_history(self, get_sorted_releases, get_release_builds):
self.assertEqual(
partials.populate_release_history(
"Firefox", "mozilla-release", partial_updates={}
),
{},
)
get_sorted_releases.assert_not_called()
get_release_builds.assert_not_called()
def patched_get_sorted_releases(product, branch):
assert branch == "mozilla-central"
return [
"Firefox-mozilla-central-nightly-20211003201113",
"Firefox-mozilla-central-nightly-20211003100640",
"Firefox-mozilla-central-nightly-20211002213629",
"Firefox-mozilla-central-nightly-20211002095048",
"Firefox-mozilla-central-nightly-20211001214601",
"Firefox-mozilla-central-nightly-20211001093323",
]
def patched_get_release_builds(release, branch):
if branch == "mozilla-central":
return nightly_blob(release)
elif branch == "mozilla-release":
return release_blob
get_sorted_releases.side_effect = patched_get_sorted_releases
get_release_builds.side_effect = patched_get_release_builds
self.assertEqual(
partials.populate_release_history(
"Firefox",
"mozilla-release",
partial_updates={"92.0.1": {"buildNumber": 1}},
),
{
"WINNT_x86_64-msvc": {
"en-US": {
"target-92.0.1.partial.mar": {
"buildid": "20210922161155",
"mar_url": "win64/en-US/firefox-92.0.1.complete.mar",
"previousVersion": "92.0.1",
"previousBuildNumber": 1,
"product": "Firefox",
}
}
}
},
)
self.assertEqual(
partials.populate_release_history("Firefox", "mozilla-central"),
{
"WINNT_x86_64-msvc": {
"en-US": {
"target.partial-1.mar": {
"buildid": "20211003201113",
"mar_url": "Firefox-mozilla-central-nightly-20211003201113",
},
"target.partial-2.mar": {
"buildid": "20211003100640",
"mar_url": "Firefox-mozilla-central-nightly-20211003100640",
},
"target.partial-3.mar": {
"buildid": "20211002213629",
"mar_url": "Firefox-mozilla-central-nightly-20211002213629",
},
"target.partial-4.mar": {
"buildid": "20211002095048",
"mar_url": "Firefox-mozilla-central-nightly-20211002095048",
},
}
}
},
)
if __name__ == "__main__":
main()
|
