Mike Frysinger | f1ba7ad | 2022-09-12 05:42:57 -0400 | [diff] [blame] | 1 | # Copyright 2013 The ChromiumOS Authors |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 2 | # Use of this source code is governed by a BSD-style license that can be |
| 3 | # found in the LICENSE file. |
| 4 | |
| 5 | """Unittests for pushimage.py""" |
| 6 | |
Mike Frysinger | d7c9309 | 2019-10-14 00:12:50 -0400 | [diff] [blame] | 7 | import collections |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 8 | import os |
Mike Frysinger | 166fea0 | 2021-02-12 05:30:33 -0500 | [diff] [blame] | 9 | from unittest import mock |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 10 | |
Mike Frysinger | 6791b1d | 2014-03-04 20:52:22 -0500 | [diff] [blame] | 11 | from chromite.lib import cros_build_lib |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 12 | from chromite.lib import cros_test_lib |
Don Garrett | 9fd20a8 | 2014-09-04 11:37:22 -0700 | [diff] [blame] | 13 | from chromite.lib import gs |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 14 | from chromite.lib import gs_unittest |
| 15 | from chromite.lib import osutils |
Mike Frysinger | 4495b03 | 2014-03-05 17:24:03 -0500 | [diff] [blame] | 16 | from chromite.lib import partial_mock |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 17 | from chromite.lib import signing |
| 18 | from chromite.scripts import pushimage |
| 19 | |
| 20 | |
Greg Edelston | a4c9b3b | 2020-01-07 17:51:13 -0700 | [diff] [blame] | 21 | # Use our local copy of insns for testing as the main one is not available in |
| 22 | # the public manifest. Even though _REL is a relative path, this works because |
| 23 | # os.join leaves absolute paths on the right hand side alone. |
| 24 | signing.INPUT_INSN_DIR_REL = signing.TEST_INPUT_INSN_DIR |
| 25 | |
| 26 | |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 27 | class InputInsnsTest(cros_test_lib.MockTestCase): |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 28 | """Tests for InputInsns""" |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 29 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 30 | def setUp(self): |
| 31 | self.StartPatcher(gs_unittest.GSContextMock()) |
Don Garrett | ac52977 | 2015-04-29 15:09:26 -0700 | [diff] [blame] | 32 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 33 | def testBasic(self): |
| 34 | """Simple smoke test""" |
| 35 | insns = pushimage.InputInsns("test.board") |
| 36 | insns.GetInsnFile("recovery") |
| 37 | self.assertEqual(insns.GetChannels(), ["dev", "canary"]) |
| 38 | self.assertEqual(insns.GetKeysets(), ["stumpy-mp-v3"]) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 39 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 40 | def testGetInsnFile(self): |
| 41 | """Verify various inputs result in right insns path""" |
| 42 | testdata = ( |
| 43 | ("UPPER_CAPS", "UPPER_CAPS"), |
| 44 | ("recovery", "test.board"), |
| 45 | ("firmware", "test.board.firmware"), |
| 46 | ("factory", "test.board.factory"), |
| 47 | ) |
| 48 | insns = pushimage.InputInsns("test.board") |
| 49 | for image_type, filename in testdata: |
| 50 | ret = insns.GetInsnFile(image_type) |
| 51 | self.assertEqual( |
| 52 | os.path.basename(ret), "%s.instructions" % (filename) |
| 53 | ) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 54 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 55 | def testSplitCfgField(self): |
| 56 | """Verify splitting behavior behaves""" |
| 57 | testdata = ( |
| 58 | ("", []), |
| 59 | ("a b c", ["a", "b", "c"]), |
| 60 | ("a, b", ["a", "b"]), |
| 61 | ("a,b", ["a", "b"]), |
| 62 | ("a,\tb", ["a", "b"]), |
| 63 | ("a\tb", ["a", "b"]), |
| 64 | ) |
| 65 | for val, exp in testdata: |
| 66 | ret = pushimage.InputInsns.SplitCfgField(val) |
| 67 | self.assertEqual(ret, exp) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 68 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 69 | def testOutputInsnsBasic(self): |
| 70 | """Verify output instructions are correct""" |
| 71 | exp_content = """[insns] |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 72 | channel = dev canary |
Mike Frysinger | d84d91e | 2015-11-05 18:02:24 -0500 | [diff] [blame] | 73 | keyset = stumpy-mp-v3 |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 74 | chromeos_shell = false |
| 75 | ensure_no_password = true |
| 76 | firmware_update = true |
| 77 | security_checks = true |
| 78 | create_nplusone = true |
| 79 | |
| 80 | [general] |
| 81 | """ |
| 82 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 83 | insns = pushimage.InputInsns("test.board") |
| 84 | self.assertEqual(insns.GetAltInsnSets(), [None]) |
| 85 | m = self.PatchObject(osutils, "WriteFile") |
| 86 | insns.OutputInsns("/bogus", {}, {}) |
| 87 | self.assertTrue(m.called) |
| 88 | content = m.call_args_list[0][0][1] |
| 89 | self.assertEqual(content.rstrip(), exp_content.rstrip()) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 90 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 91 | def testOutputInsnsReplacements(self): |
| 92 | """Verify output instructions can be updated""" |
| 93 | exp_content = """[insns] |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 94 | channel = dev |
Mike Frysinger | d84d91e | 2015-11-05 18:02:24 -0500 | [diff] [blame] | 95 | keyset = batman |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 96 | chromeos_shell = false |
| 97 | ensure_no_password = true |
| 98 | firmware_update = true |
| 99 | security_checks = true |
| 100 | create_nplusone = true |
| 101 | |
| 102 | [general] |
| 103 | board = board |
| 104 | config_board = test.board |
| 105 | """ |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 106 | sect_insns = { |
| 107 | "channel": "dev", |
| 108 | "keyset": "batman", |
| 109 | } |
| 110 | sect_general = collections.OrderedDict( |
| 111 | ( |
| 112 | ("board", "board"), |
| 113 | ("config_board", "test.board"), |
| 114 | ) |
| 115 | ) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 116 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 117 | insns = pushimage.InputInsns("test.board") |
| 118 | m = self.PatchObject(osutils, "WriteFile") |
| 119 | insns.OutputInsns("/a/file", sect_insns, sect_general) |
| 120 | self.assertTrue(m.called) |
| 121 | content = m.call_args_list[0][0][1] |
| 122 | self.assertEqual(content.rstrip(), exp_content.rstrip()) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 123 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 124 | def testOutputInsnsMergeAlts(self): |
| 125 | """Verify handling of alternative insns.xxx sections""" |
| 126 | TEMPLATE_CONTENT = """[insns] |
Mike Frysinger | 37ccc2b | 2015-11-11 17:16:51 -0500 | [diff] [blame] | 127 | channel = %(channel)s |
| 128 | chromeos_shell = false |
| 129 | ensure_no_password = true |
| 130 | firmware_update = true |
| 131 | security_checks = true |
| 132 | create_nplusone = true |
| 133 | override = sect_insns |
| 134 | keyset = %(keyset)s |
| 135 | %(extra)s |
| 136 | [general] |
| 137 | board = board |
| 138 | config_board = test.board |
| 139 | """ |
| 140 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 141 | exp_alts = ["insns.one", "insns.two", "insns.hotsoup"] |
| 142 | exp_fields = { |
| 143 | "one": { |
| 144 | "channel": "dev canary", |
| 145 | "keyset": "OneKeyset", |
| 146 | "extra": "", |
| 147 | }, |
| 148 | "two": {"channel": "best", "keyset": "TwoKeyset", "extra": ""}, |
| 149 | "hotsoup": { |
| 150 | "channel": "dev canary", |
| 151 | "keyset": "ColdKeyset", |
| 152 | "extra": "soup = cheddar\n", |
| 153 | }, |
| 154 | } |
Mike Frysinger | 37ccc2b | 2015-11-11 17:16:51 -0500 | [diff] [blame] | 155 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 156 | # Make sure this overrides the insn sections. |
| 157 | sect_insns = { |
| 158 | "override": "sect_insns", |
| 159 | } |
| 160 | sect_insns_copy = sect_insns.copy() |
| 161 | sect_general = collections.OrderedDict( |
| 162 | ( |
| 163 | ("board", "board"), |
| 164 | ("config_board", "test.board"), |
| 165 | ) |
| 166 | ) |
Mike Frysinger | 37ccc2b | 2015-11-11 17:16:51 -0500 | [diff] [blame] | 167 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 168 | insns = pushimage.InputInsns("test.multi") |
| 169 | self.assertEqual(insns.GetAltInsnSets(), exp_alts) |
| 170 | m = self.PatchObject(osutils, "WriteFile") |
Mike Frysinger | 37ccc2b | 2015-11-11 17:16:51 -0500 | [diff] [blame] | 171 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 172 | for alt in exp_alts: |
| 173 | m.reset_mock() |
| 174 | insns.OutputInsns( |
| 175 | "/a/file", sect_insns, sect_general, insns_merge=alt |
| 176 | ) |
| 177 | self.assertEqual(sect_insns, sect_insns_copy) |
| 178 | self.assertTrue(m.called) |
| 179 | content = m.call_args_list[0][0][1] |
| 180 | exp_content = TEMPLATE_CONTENT % exp_fields[alt[6:]] |
| 181 | self.assertEqual(content.rstrip(), exp_content.rstrip()) |
Mike Frysinger | 37ccc2b | 2015-11-11 17:16:51 -0500 | [diff] [blame] | 182 | |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 183 | |
Mike Frysinger | 4495b03 | 2014-03-05 17:24:03 -0500 | [diff] [blame] | 184 | class MarkImageToBeSignedTest(gs_unittest.AbstractGSContextTest): |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 185 | """Tests for MarkImageToBeSigned()""" |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 186 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 187 | def setUp(self): |
| 188 | # Minor optimization -- we call this for logging purposes in the main |
| 189 | # code, but don't really care about it for testing. It just slows us. |
| 190 | self.PatchObject( |
| 191 | cros_build_lib, "MachineDetails", return_value="1234\n" |
| 192 | ) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 193 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 194 | def testBasic(self): |
| 195 | """Simple smoke test""" |
| 196 | tbs_base = "gs://some-bucket" |
| 197 | insns_path = "chan/board/ver/file.instructions" |
| 198 | tbs_file = ( |
| 199 | "%s/tobesigned/90,chan,board,ver,file.instructions" % tbs_base |
| 200 | ) |
| 201 | ret = pushimage.MarkImageToBeSigned(self.ctx, tbs_base, insns_path, 90) |
| 202 | self.assertEqual(ret, tbs_file) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 203 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 204 | def testPriority(self): |
| 205 | """Verify diff priority values get used correctly""" |
| 206 | for prio, sprio in ((0, "00"), (9, "09"), (35, "35"), (99, "99")): |
| 207 | ret = pushimage.MarkImageToBeSigned(self.ctx, "", "", prio) |
| 208 | self.assertEqual(ret, "/tobesigned/%s," % sprio) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 209 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 210 | def testBadPriority(self): |
| 211 | """Verify we reject bad priority values""" |
| 212 | for prio in (-10, -1, 100, 91239): |
| 213 | self.assertRaises( |
| 214 | ValueError, |
| 215 | pushimage.MarkImageToBeSigned, |
| 216 | self.ctx, |
| 217 | "", |
| 218 | "", |
| 219 | prio, |
| 220 | ) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 221 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 222 | def testTbsUpload(self): |
| 223 | """Make sure we actually try to upload the file""" |
| 224 | pushimage.MarkImageToBeSigned(self.ctx, "", "", 50) |
| 225 | self.gs_mock.assertCommandContains(["cp", "--"]) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 226 | |
| 227 | |
Mike Frysinger | 4495b03 | 2014-03-05 17:24:03 -0500 | [diff] [blame] | 228 | class PushImageTests(gs_unittest.AbstractGSContextTest): |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 229 | """Tests for PushImage()""" |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 230 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 231 | def setUp(self): |
| 232 | self.mark_mock = self.PatchObject(pushimage, "MarkImageToBeSigned") |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 233 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 234 | def testBasic(self): |
| 235 | """Simple smoke test""" |
| 236 | EXPECTED = { |
| 237 | "canary": [ |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 238 | "gs://chromeos-releases/canary-channel/test.board-hi/5126.0.0/" |
| 239 | "ChromeOS-recovery-R34-5126.0.0-test.board-hi.instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 240 | ], |
| 241 | "dev": [ |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 242 | "gs://chromeos-releases/dev-channel/test.board-hi/5126.0.0/" |
| 243 | "ChromeOS-recovery-R34-5126.0.0-test.board-hi.instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 244 | ], |
| 245 | } |
| 246 | with mock.patch.object(gs.GSContext, "Exists", return_value=True): |
| 247 | urls = pushimage.PushImage( |
| 248 | "/src", "test.board", "R34-5126.0.0", profile="hi" |
| 249 | ) |
Don Garrett | 9459c2f | 2014-01-22 18:20:24 -0800 | [diff] [blame] | 250 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 251 | self.assertEqual(urls, EXPECTED) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 252 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 253 | def testBasic_SignTypesEmptyList(self): |
| 254 | """Tests PushImage behavior when |sign_types| is empty instead of None. |
Amey Deshpande | 3c48795 | 2015-09-03 12:11:13 -0700 | [diff] [blame] | 255 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 256 | As part of the buildbots, PushImage function always receives a tuple for |
| 257 | |sign_types| argument. This test checks the behavior for empty tuple. |
| 258 | """ |
| 259 | EXPECTED = { |
| 260 | "canary": [ |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 261 | "gs://chromeos-releases/canary-channel/test.board-hi/5126.0.0/" |
| 262 | "ChromeOS-recovery-R34-5126.0.0-test.board-hi.instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 263 | ], |
| 264 | "dev": [ |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 265 | "gs://chromeos-releases/dev-channel/test.board-hi/5126.0.0/" |
| 266 | "ChromeOS-recovery-R34-5126.0.0-test.board-hi.instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 267 | ], |
| 268 | } |
| 269 | with mock.patch.object(gs.GSContext, "Exists", return_value=True): |
| 270 | urls = pushimage.PushImage( |
| 271 | "/src", |
| 272 | "test.board", |
| 273 | "R34-5126.0.0", |
| 274 | profile="hi", |
| 275 | sign_types=(), |
| 276 | ) |
Amey Deshpande | 3c48795 | 2015-09-03 12:11:13 -0700 | [diff] [blame] | 277 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 278 | self.assertEqual(urls, EXPECTED) |
Amey Deshpande | 3c48795 | 2015-09-03 12:11:13 -0700 | [diff] [blame] | 279 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 280 | def testBasic_RealBoardName(self): |
| 281 | """Runs a simple smoke test using a real board name.""" |
| 282 | EXPECTED = { |
| 283 | "canary": [ |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 284 | "gs://chromeos-releases/canary-channel/x86-alex/5126.0.0/" |
| 285 | "ChromeOS-recovery-R34-5126.0.0-x86-alex.instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 286 | ], |
| 287 | "dev": [ |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 288 | "gs://chromeos-releases/dev-channel/x86-alex/5126.0.0/" |
| 289 | "ChromeOS-recovery-R34-5126.0.0-x86-alex.instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 290 | ], |
| 291 | } |
| 292 | with mock.patch.object(gs.GSContext, "Exists", return_value=True): |
| 293 | urls = pushimage.PushImage("/src", "x86-alex", "R34-5126.0.0") |
Amey Deshpande | a936c62 | 2015-08-12 17:27:54 -0700 | [diff] [blame] | 294 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 295 | self.assertEqual(urls, EXPECTED) |
Amey Deshpande | a936c62 | 2015-08-12 17:27:54 -0700 | [diff] [blame] | 296 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 297 | def testBasicMock(self): |
| 298 | """Simple smoke test in mock mode""" |
| 299 | with mock.patch.object(gs.GSContext, "Exists", return_value=True): |
| 300 | pushimage.PushImage( |
Mike Frysinger | 9d1253c | 2023-02-02 09:00:07 -0500 | [diff] [blame] | 301 | "/src", "test.board", "R34-5126.0.0", dryrun=True, mock=True |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 302 | ) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 303 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 304 | def testBadVersion(self): |
| 305 | """Make sure we barf on bad version strings""" |
| 306 | self.assertRaises(ValueError, pushimage.PushImage, "", "", "asdf") |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 307 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 308 | def testNoInsns(self): |
| 309 | """Boards w/out insn files should get skipped""" |
| 310 | urls = pushimage.PushImage("/src", "a bad bad board", "R34-5126.0.0") |
| 311 | self.assertEqual(self.gs_mock.call_count, 0) |
| 312 | self.assertEqual(urls, None) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 313 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 314 | def testSignTypesRecovery(self): |
| 315 | """Only sign the requested recovery type""" |
| 316 | EXPECTED = { |
| 317 | "canary": [ |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 318 | "gs://chromeos-releases/canary-channel/test.board/5126.0.0/" |
| 319 | "ChromeOS-recovery-R34-5126.0.0-test.board.instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 320 | ], |
| 321 | "dev": [ |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 322 | "gs://chromeos-releases/dev-channel/test.board/5126.0.0/" |
| 323 | "ChromeOS-recovery-R34-5126.0.0-test.board.instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 324 | ], |
| 325 | } |
Don Garrett | 9459c2f | 2014-01-22 18:20:24 -0800 | [diff] [blame] | 326 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 327 | with mock.patch.object(gs.GSContext, "Exists", return_value=True): |
| 328 | urls = pushimage.PushImage( |
| 329 | "/src", "test.board", "R34-5126.0.0", sign_types=["recovery"] |
| 330 | ) |
Shuhei Takahashi | 7e05dbf | 2023-05-19 10:40:47 +0900 | [diff] [blame] | 331 | self.assertEqual(self.gs_mock.call_count, 26) |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 332 | self.assertTrue(self.mark_mock.called) |
| 333 | self.assertEqual(urls, EXPECTED) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 334 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 335 | def testSignTypesBase(self): |
| 336 | """Only sign the requested recovery type""" |
| 337 | EXPECTED = { |
| 338 | "canary": [ |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 339 | "gs://chromeos-releases/canary-channel/test.board/5126.0.0/" |
| 340 | "ChromeOS-base-R34-5126.0.0-test.board.instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 341 | ], |
| 342 | "dev": [ |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 343 | "gs://chromeos-releases/dev-channel/test.board/5126.0.0/" |
| 344 | "ChromeOS-base-R34-5126.0.0-test.board.instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 345 | ], |
| 346 | } |
Amey Deshpande | a936c62 | 2015-08-12 17:27:54 -0700 | [diff] [blame] | 347 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 348 | with mock.patch.object(gs.GSContext, "Exists", return_value=True): |
| 349 | urls = pushimage.PushImage( |
| 350 | "/src", "test.board", "R34-5126.0.0", sign_types=["base"] |
| 351 | ) |
Shuhei Takahashi | 7e05dbf | 2023-05-19 10:40:47 +0900 | [diff] [blame] | 352 | self.assertEqual(self.gs_mock.call_count, 28) |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 353 | self.assertTrue(self.mark_mock.called) |
| 354 | self.assertEqual(urls, EXPECTED) |
LaMont Jones | 7d6c98f | 2019-09-27 12:37:33 -0600 | [diff] [blame] | 355 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 356 | def testSignTypesGscFirmware(self): |
| 357 | """Only sign the requested type""" |
| 358 | EXPECTED = { |
| 359 | "canary": [ |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 360 | "gs://chromeos-releases/canary-channel/board2/5126.0.0/" |
| 361 | "ChromeOS-gsc_firmware-R34-5126.0.0-board2.instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 362 | ], |
| 363 | "dev": [ |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 364 | "gs://chromeos-releases/dev-channel/board2/5126.0.0/" |
| 365 | "ChromeOS-gsc_firmware-R34-5126.0.0-board2.instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 366 | ], |
| 367 | } |
LaMont Jones | 7d6c98f | 2019-09-27 12:37:33 -0600 | [diff] [blame] | 368 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 369 | with mock.patch.object(gs.GSContext, "Exists", return_value=True): |
| 370 | urls = pushimage.PushImage( |
| 371 | "/src", "board2", "R34-5126.0.0", sign_types=["gsc_firmware"] |
| 372 | ) |
Shuhei Takahashi | 7e05dbf | 2023-05-19 10:40:47 +0900 | [diff] [blame] | 373 | self.assertEqual(self.gs_mock.call_count, 28) |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 374 | self.assertTrue(self.mark_mock.called) |
| 375 | self.assertEqual(urls, EXPECTED) |
Amey Deshpande | a936c62 | 2015-08-12 17:27:54 -0700 | [diff] [blame] | 376 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 377 | def testSignTypesNone(self): |
| 378 | """Verify nothing is signed when we request an unavailable type""" |
| 379 | urls = pushimage.PushImage( |
| 380 | "/src", "test.board", "R34-5126.0.0", sign_types=["nononononono"] |
| 381 | ) |
Shuhei Takahashi | 7e05dbf | 2023-05-19 10:40:47 +0900 | [diff] [blame] | 382 | self.assertEqual(self.gs_mock.call_count, 24) |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 383 | self.assertFalse(self.mark_mock.called) |
| 384 | self.assertEqual(urls, {}) |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 385 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 386 | def testGsError(self): |
| 387 | """Verify random GS errors don't make us blow up entirely""" |
| 388 | self.gs_mock.AddCmdResult( |
| 389 | partial_mock.In("stat"), returncode=1, stdout="gobblety gook\n" |
| 390 | ) |
| 391 | with cros_test_lib.LoggingCapturer("chromite"): |
| 392 | self.assertRaises( |
| 393 | pushimage.PushError, |
| 394 | pushimage.PushImage, |
| 395 | "/src", |
| 396 | "test.board", |
| 397 | "R34-5126.0.0", |
| 398 | ) |
Mike Frysinger | 4495b03 | 2014-03-05 17:24:03 -0500 | [diff] [blame] | 399 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 400 | def testMultipleKeysets(self): |
| 401 | """Verify behavior when processing an insn w/multiple keysets""" |
| 402 | EXPECTED = { |
| 403 | "canary": [ |
| 404 | ( |
| 405 | "gs://chromeos-releases/canary-channel/test.board/5126.0.0/" |
| 406 | "ChromeOS-recovery-R34-5126.0.0-test.board.instructions" |
| 407 | ), |
| 408 | ( |
| 409 | "gs://chromeos-releases/canary-channel/test.board/5126.0.0/" |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 410 | "ChromeOS-recovery-R34-5126.0.0-test.board-key2." |
| 411 | "instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 412 | ), |
| 413 | ( |
| 414 | "gs://chromeos-releases/canary-channel/test.board/5126.0.0/" |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 415 | "ChromeOS-recovery-R34-5126.0.0-test.board-key3." |
| 416 | "instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 417 | ), |
| 418 | ], |
| 419 | "dev": [ |
| 420 | ( |
| 421 | "gs://chromeos-releases/dev-channel/test.board/5126.0.0/" |
| 422 | "ChromeOS-recovery-R34-5126.0.0-test.board.instructions" |
| 423 | ), |
| 424 | ( |
| 425 | "gs://chromeos-releases/dev-channel/test.board/5126.0.0/" |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 426 | "ChromeOS-recovery-R34-5126.0.0-test.board-key2." |
| 427 | "instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 428 | ), |
| 429 | ( |
| 430 | "gs://chromeos-releases/dev-channel/test.board/5126.0.0/" |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 431 | "ChromeOS-recovery-R34-5126.0.0-test.board-key3." |
| 432 | "instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 433 | ), |
| 434 | ], |
| 435 | } |
| 436 | with mock.patch.object(gs.GSContext, "Exists", return_value=True): |
| 437 | urls = pushimage.PushImage( |
| 438 | "/src", |
| 439 | "test.board", |
| 440 | "R34-5126.0.0", |
| 441 | force_keysets=("key1", "key2", "key3"), |
| 442 | ) |
| 443 | self.assertEqual(urls, EXPECTED) |
Mike Frysinger | d84d91e | 2015-11-05 18:02:24 -0500 | [diff] [blame] | 444 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 445 | def testForceChannel(self): |
| 446 | """Verify behavior when user has specified custom channel""" |
| 447 | EXPECTED = { |
| 448 | "meep": [ |
| 449 | ( |
| 450 | "gs://chromeos-releases/meep-channel/test.board/5126.0.0/" |
| 451 | "ChromeOS-recovery-R34-5126.0.0-test.board.instructions" |
| 452 | ), |
| 453 | ], |
| 454 | } |
| 455 | with mock.patch.object(gs.GSContext, "Exists", return_value=True): |
| 456 | urls = pushimage.PushImage( |
| 457 | "/src", "test.board", "R34-5126.0.0", force_channels=("meep",) |
| 458 | ) |
| 459 | self.assertEqual(urls, EXPECTED) |
Mike Frysinger | 7791210 | 2017-08-30 18:35:46 -0400 | [diff] [blame] | 460 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 461 | def testMultipleAltInsns(self): |
| 462 | """Verify behavior when processing an insn w/multiple insn overlays""" |
| 463 | EXPECTED = { |
| 464 | "canary": [ |
| 465 | ( |
| 466 | "gs://chromeos-releases/canary-channel/test.multi/1.0.0/" |
| 467 | "ChromeOS-recovery-R1-1.0.0-test.multi.instructions" |
| 468 | ), |
| 469 | ( |
| 470 | "gs://chromeos-releases/canary-channel/test.multi/1.0.0/" |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 471 | "ChromeOS-recovery-R1-1.0.0-test.multi-TwoKeyset." |
| 472 | "instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 473 | ), |
| 474 | ( |
| 475 | "gs://chromeos-releases/canary-channel/test.multi/1.0.0/" |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 476 | "ChromeOS-recovery-R1-1.0.0-test.multi-ColdKeyset." |
| 477 | "instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 478 | ), |
| 479 | ], |
| 480 | "dev": [ |
| 481 | ( |
| 482 | "gs://chromeos-releases/dev-channel/test.multi/1.0.0/" |
| 483 | "ChromeOS-recovery-R1-1.0.0-test.multi.instructions" |
| 484 | ), |
| 485 | ( |
| 486 | "gs://chromeos-releases/dev-channel/test.multi/1.0.0/" |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 487 | "ChromeOS-recovery-R1-1.0.0-test.multi-TwoKeyset." |
| 488 | "instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 489 | ), |
| 490 | ( |
| 491 | "gs://chromeos-releases/dev-channel/test.multi/1.0.0/" |
Trent Apted | c4f366a | 2023-05-16 15:32:48 +1000 | [diff] [blame^] | 492 | "ChromeOS-recovery-R1-1.0.0-test.multi-ColdKeyset." |
| 493 | "instructions" |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 494 | ), |
| 495 | ], |
| 496 | } |
| 497 | with mock.patch.object(gs.GSContext, "Exists", return_value=True): |
| 498 | urls = pushimage.PushImage("/src", "test.multi", "R1-1.0.0") |
| 499 | self.assertEqual(urls, EXPECTED) |
Mike Frysinger | 37ccc2b | 2015-11-11 17:16:51 -0500 | [diff] [blame] | 500 | |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 501 | |
| 502 | class MainTests(cros_test_lib.MockTestCase): |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 503 | """Tests for main()""" |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 504 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 505 | def setUp(self): |
| 506 | self.PatchObject(pushimage, "PushImage") |
Mike Frysinger | d13faeb | 2013-09-05 16:00:46 -0400 | [diff] [blame] | 507 | |
Alex Klein | 1699fab | 2022-09-08 08:46:06 -0600 | [diff] [blame] | 508 | def testBasic(self): |
| 509 | """Simple smoke test""" |
| 510 | pushimage.main(["--board", "test.board", "/src", "--yes"]) |