blob: 0c73ff3c1e2bd9ee482bf2735d8815d1a04c3f57 [file] [log] [blame]
Mike Frysingere58c0e22017-10-04 15:43:30 -04001# -*- coding: utf-8 -*-
David Pursell9476bf42015-03-30 13:34:27 -07002# Copyright 2015 The Chromium OS Authors. All rights reserved.
3# Use of this source code is governed by a BSD-style license that can be
4# found in the LICENSE file.
5
6"""Unit tests for the deploy module."""
7
8from __future__ import print_function
9
10import json
Ralph Nathane01ccf12015-04-16 10:40:32 -070011import multiprocessing
David Pursell9476bf42015-03-30 13:34:27 -070012import os
Mike Frysinger050fa5a2019-12-01 16:16:29 -050013import sys
David Pursell9476bf42015-03-30 13:34:27 -070014
Ralph Nathane01ccf12015-04-16 10:40:32 -070015from chromite.cli import command
David Pursell9476bf42015-03-30 13:34:27 -070016from chromite.cli import deploy
17from chromite.lib import cros_build_lib
18from chromite.lib import cros_test_lib
Gilad Arnold0e1b1da2015-06-10 06:41:05 -070019from chromite.lib import portage_util
Ralph Nathane01ccf12015-04-16 10:40:32 -070020from chromite.lib import remote_access
Greg Edelstona4c9b3b2020-01-07 17:51:13 -070021
Chris McDonald186b9ee2020-04-16 05:56:49 -060022pytestmark = [cros_test_lib.pytestmark_inside_only]
Greg Edelstona4c9b3b2020-01-07 17:51:13 -070023
24
Mike Frysinger3f087aa2020-03-20 06:03:16 -040025assert sys.version_info >= (3, 6), 'This module requires Python 3.6+'
26
27
Greg Edelstona4c9b3b2020-01-07 17:51:13 -070028if cros_build_lib.IsInsideChroot():
29 import portage # pylint: disable=import-error
David Pursell9476bf42015-03-30 13:34:27 -070030
31
32# pylint: disable=protected-access
33
34
Ralph Nathane01ccf12015-04-16 10:40:32 -070035class ChromiumOSDeviceFake(object):
36 """Fake for device."""
37
38 def __init__(self):
39 self.board = 'board'
40 self.hostname = None
41 self.username = None
42 self.port = None
43 self.lsb_release = None
Qijiang Fan8a945032019-04-25 20:53:29 +090044 self.cmds = []
45 self.work_dir = '/testdir/'
Ben Pastene5f03b052019-08-12 18:03:24 -070046 self.selinux_available = False
Ralph Nathane01ccf12015-04-16 10:40:32 -070047
Achuith Bhandarkar0487c312019-04-22 12:19:25 -070048 def MountRootfsReadWrite(self):
Ralph Nathane01ccf12015-04-16 10:40:32 -070049 return True
50
Ben Pastene5f03b052019-08-12 18:03:24 -070051 def IsSELinuxAvailable(self):
52 return self.selinux_available
53
54 def IsSELinuxEnforced(self):
55 return True
56
Andrewc7e1c6b2020-02-27 16:03:53 -080057 def run(self, cmd, **_kwargs):
58 self.cmds.append(cmd)
59
60 def CopyToDevice(self, _src, _dest, _mode='rsync', **_kwargs):
61 return True
62
Ralph Nathane01ccf12015-04-16 10:40:32 -070063
David Pursell9476bf42015-03-30 13:34:27 -070064class ChromiumOSDeviceHandlerFake(object):
65 """Fake for chromite.lib.remote_access.ChomiumOSDeviceHandler."""
66
67 class RemoteAccessFake(object):
68 """Fake for chromite.lib.remote_access.RemoteAccess."""
69
70 def __init__(self):
71 self.remote_sh_output = None
72
73 def RemoteSh(self, *_args, **_kwargs):
74 return cros_build_lib.CommandResult(output=self.remote_sh_output)
75
Ralph Nathane01ccf12015-04-16 10:40:32 -070076 def __init__(self, *_args, **_kwargs):
David Pursell67a82762015-04-30 17:26:59 -070077 self._agent = self.RemoteAccessFake()
Mike Frysinger539db512015-05-21 18:14:01 -040078 self.device = ChromiumOSDeviceFake()
David Pursell67a82762015-04-30 17:26:59 -070079
Ralph Nathane01ccf12015-04-16 10:40:32 -070080 # TODO(dpursell): Mock remote access object in cros_test_lib (brbug.com/986).
David Pursell67a82762015-04-30 17:26:59 -070081 def GetAgent(self):
82 return self._agent
David Pursell9476bf42015-03-30 13:34:27 -070083
Ralph Nathane01ccf12015-04-16 10:40:32 -070084 def __exit__(self, _type, _value, _traceback):
85 pass
86
87 def __enter__(self):
Qijiang Fan8a945032019-04-25 20:53:29 +090088 return self.device
Ralph Nathane01ccf12015-04-16 10:40:32 -070089
90
91class BrilloDeployOperationFake(deploy.BrilloDeployOperation):
92 """Fake for deploy.BrilloDeployOperation."""
93 def __init__(self, pkg_count, emerge, queue):
94 super(BrilloDeployOperationFake, self).__init__(pkg_count, emerge)
95 self._queue = queue
96
Ralph Nathandc14ed92015-04-22 11:17:40 -070097 def ParseOutput(self, output=None):
98 super(BrilloDeployOperationFake, self).ParseOutput(output)
Ralph Nathane01ccf12015-04-16 10:40:32 -070099 self._queue.put('advance')
100
David Pursell9476bf42015-03-30 13:34:27 -0700101
102class DbApiFake(object):
103 """Fake for Portage dbapi."""
104
105 def __init__(self, pkgs):
106 self.pkg_db = {}
107 for cpv, slot, rdeps_raw, build_time in pkgs:
108 self.pkg_db[cpv] = {
109 'SLOT': slot, 'RDEPEND': rdeps_raw, 'BUILD_TIME': build_time}
110
111 def cpv_all(self):
Mike Frysinger1f4478c2019-10-20 18:33:17 -0400112 return list(self.pkg_db)
David Pursell9476bf42015-03-30 13:34:27 -0700113
114 def aux_get(self, cpv, keys):
115 pkg_info = self.pkg_db[cpv]
116 return [pkg_info[key] for key in keys]
117
118
Ralph Nathane01ccf12015-04-16 10:40:32 -0700119class PackageScannerFake(object):
120 """Fake for PackageScanner."""
121
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700122 def __init__(self, packages, pkgs_attrs, packages_cpvs=None):
Ralph Nathane01ccf12015-04-16 10:40:32 -0700123 self.pkgs = packages
Ned Nguyend0db4072019-02-22 14:19:21 -0700124 self.cpvs = packages_cpvs or packages
Ralph Nathane01ccf12015-04-16 10:40:32 -0700125 self.listed = []
Mike Frysingercce4f832019-11-14 22:44:38 -0500126 self.num_updates = 0
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700127 self.pkgs_attrs = pkgs_attrs
Ralph Nathane01ccf12015-04-16 10:40:32 -0700128
129 def Run(self, _device, _root, _packages, _update, _deep, _deep_rev):
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700130 return self.cpvs, self.listed, self.num_updates, self.pkgs_attrs
Ralph Nathane01ccf12015-04-16 10:40:32 -0700131
132
David Pursell9476bf42015-03-30 13:34:27 -0700133class PortageTreeFake(object):
134 """Fake for Portage tree."""
135
136 def __init__(self, dbapi):
137 self.dbapi = dbapi
138
139
Ralph Nathane01ccf12015-04-16 10:40:32 -0700140class TestInstallPackageScanner(cros_test_lib.MockOutputTestCase):
David Pursell9476bf42015-03-30 13:34:27 -0700141 """Test the update package scanner."""
142 _BOARD = 'foo_board'
143 _BUILD_ROOT = '/build/%s' % _BOARD
144 _VARTREE = [
145 ('foo/app1-1.2.3-r4', '0', 'foo/app2 !foo/app3', '1413309336'),
146 ('foo/app2-4.5.6-r7', '0', '', '1413309336'),
147 ('foo/app4-2.0.0-r1', '0', 'foo/app1 foo/app5', '1413309336'),
148 ('foo/app5-3.0.7-r3', '0', '', '1413309336'),
149 ]
150
151 def setUp(self):
152 """Patch imported modules."""
153 self.PatchObject(cros_build_lib, 'GetChoice', return_value=0)
154 self.device = ChromiumOSDeviceHandlerFake()
155 self.scanner = deploy._InstallPackageScanner(self._BUILD_ROOT)
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700156 self.PatchObject(deploy, '_GetDLCInfo', return_value=(None, None))
David Pursell9476bf42015-03-30 13:34:27 -0700157
158 def SetupVartree(self, vartree_pkgs):
David Pursell67a82762015-04-30 17:26:59 -0700159 self.device.GetAgent().remote_sh_output = json.dumps(vartree_pkgs)
David Pursell9476bf42015-03-30 13:34:27 -0700160
161 def SetupBintree(self, bintree_pkgs):
162 bintree = PortageTreeFake(DbApiFake(bintree_pkgs))
163 build_root = os.path.join(self._BUILD_ROOT, '')
164 portage_db = {build_root: {'bintree': bintree}}
165 self.PatchObject(portage, 'create_trees', return_value=portage_db)
166
167 def ValidatePkgs(self, actual, expected, constraints=None):
168 # Containing exactly the same packages.
Mike Frysinger2d589a12019-08-25 14:15:12 -0400169 self.assertEqual(sorted(expected), sorted(actual))
David Pursell9476bf42015-03-30 13:34:27 -0700170 # Packages appear in the right order.
171 if constraints is not None:
172 for needs, needed in constraints:
173 self.assertGreater(actual.index(needs), actual.index(needed))
174
175 def testRunUpdatedVersion(self):
176 self.SetupVartree(self._VARTREE)
177 app1 = 'foo/app1-1.2.5-r4'
178 self.SetupBintree([
179 (app1, '0', 'foo/app2 !foo/app3', '1413309336'),
180 ('foo/app2-4.5.6-r7', '0', '', '1413309336'),
181 ])
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700182 installs, listed, num_updates, _ = self.scanner.Run(
David Pursell9476bf42015-03-30 13:34:27 -0700183 self.device, '/', ['app1'], True, True, True)
184 self.ValidatePkgs(installs, [app1])
185 self.ValidatePkgs(listed, [app1])
Mike Frysinger2d589a12019-08-25 14:15:12 -0400186 self.assertEqual(num_updates, 1)
David Pursell9476bf42015-03-30 13:34:27 -0700187
188 def testRunUpdatedBuildTime(self):
189 self.SetupVartree(self._VARTREE)
190 app1 = 'foo/app1-1.2.3-r4'
191 self.SetupBintree([
192 (app1, '0', 'foo/app2 !foo/app3', '1413309350'),
193 ('foo/app2-4.5.6-r7', '0', '', '1413309336'),
194 ])
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700195 installs, listed, num_updates, _ = self.scanner.Run(
David Pursell9476bf42015-03-30 13:34:27 -0700196 self.device, '/', ['app1'], True, True, True)
197 self.ValidatePkgs(installs, [app1])
198 self.ValidatePkgs(listed, [app1])
Mike Frysinger2d589a12019-08-25 14:15:12 -0400199 self.assertEqual(num_updates, 1)
David Pursell9476bf42015-03-30 13:34:27 -0700200
201 def testRunExistingDepUpdated(self):
202 self.SetupVartree(self._VARTREE)
203 app1 = 'foo/app1-1.2.5-r2'
204 app2 = 'foo/app2-4.5.8-r3'
205 self.SetupBintree([
206 (app1, '0', 'foo/app2 !foo/app3', '1413309350'),
207 (app2, '0', '', '1413309350'),
208 ])
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700209 installs, listed, num_updates, _ = self.scanner.Run(
David Pursell9476bf42015-03-30 13:34:27 -0700210 self.device, '/', ['app1'], True, True, True)
211 self.ValidatePkgs(installs, [app1, app2], constraints=[(app1, app2)])
212 self.ValidatePkgs(listed, [app1])
Mike Frysinger2d589a12019-08-25 14:15:12 -0400213 self.assertEqual(num_updates, 2)
David Pursell9476bf42015-03-30 13:34:27 -0700214
215 def testRunMissingDepUpdated(self):
216 self.SetupVartree(self._VARTREE)
217 app1 = 'foo/app1-1.2.5-r2'
218 app6 = 'foo/app6-1.0.0-r1'
219 self.SetupBintree([
220 (app1, '0', 'foo/app2 !foo/app3 foo/app6', '1413309350'),
221 ('foo/app2-4.5.6-r7', '0', '', '1413309336'),
222 (app6, '0', '', '1413309350'),
223 ])
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700224 installs, listed, num_updates, _ = self.scanner.Run(
David Pursell9476bf42015-03-30 13:34:27 -0700225 self.device, '/', ['app1'], True, True, True)
226 self.ValidatePkgs(installs, [app1, app6], constraints=[(app1, app6)])
227 self.ValidatePkgs(listed, [app1])
Mike Frysinger2d589a12019-08-25 14:15:12 -0400228 self.assertEqual(num_updates, 1)
David Pursell9476bf42015-03-30 13:34:27 -0700229
230 def testRunExistingRevDepUpdated(self):
231 self.SetupVartree(self._VARTREE)
232 app1 = 'foo/app1-1.2.5-r2'
233 app4 = 'foo/app4-2.0.1-r3'
234 self.SetupBintree([
235 (app1, '0', 'foo/app2 !foo/app3', '1413309350'),
236 (app4, '0', 'foo/app1 foo/app5', '1413309350'),
237 ('foo/app5-3.0.7-r3', '0', '', '1413309336'),
238 ])
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700239 installs, listed, num_updates, _ = self.scanner.Run(
David Pursell9476bf42015-03-30 13:34:27 -0700240 self.device, '/', ['app1'], True, True, True)
241 self.ValidatePkgs(installs, [app1, app4], constraints=[(app4, app1)])
242 self.ValidatePkgs(listed, [app1])
Mike Frysinger2d589a12019-08-25 14:15:12 -0400243 self.assertEqual(num_updates, 2)
David Pursell9476bf42015-03-30 13:34:27 -0700244
245 def testRunMissingRevDepNotUpdated(self):
246 self.SetupVartree(self._VARTREE)
247 app1 = 'foo/app1-1.2.5-r2'
248 app6 = 'foo/app6-1.0.0-r1'
249 self.SetupBintree([
250 (app1, '0', 'foo/app2 !foo/app3', '1413309350'),
251 (app6, '0', 'foo/app1', '1413309350'),
252 ])
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700253 installs, listed, num_updates, _ = self.scanner.Run(
David Pursell9476bf42015-03-30 13:34:27 -0700254 self.device, '/', ['app1'], True, True, True)
255 self.ValidatePkgs(installs, [app1])
256 self.ValidatePkgs(listed, [app1])
Mike Frysinger2d589a12019-08-25 14:15:12 -0400257 self.assertEqual(num_updates, 1)
David Pursell9476bf42015-03-30 13:34:27 -0700258
259 def testRunTransitiveDepsUpdated(self):
260 self.SetupVartree(self._VARTREE)
261 app1 = 'foo/app1-1.2.5-r2'
262 app2 = 'foo/app2-4.5.8-r3'
263 app4 = 'foo/app4-2.0.0-r1'
264 app5 = 'foo/app5-3.0.8-r2'
265 self.SetupBintree([
266 (app1, '0', 'foo/app2 !foo/app3', '1413309350'),
267 (app2, '0', '', '1413309350'),
268 (app4, '0', 'foo/app1 foo/app5', '1413309350'),
269 (app5, '0', '', '1413309350'),
270 ])
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700271 installs, listed, num_updates, _ = self.scanner.Run(
David Pursell9476bf42015-03-30 13:34:27 -0700272 self.device, '/', ['app1'], True, True, True)
273 self.ValidatePkgs(installs, [app1, app2, app4, app5],
274 constraints=[(app1, app2), (app4, app1), (app4, app5)])
275 self.ValidatePkgs(listed, [app1])
Mike Frysinger2d589a12019-08-25 14:15:12 -0400276 self.assertEqual(num_updates, 4)
David Pursell9476bf42015-03-30 13:34:27 -0700277
278 def testRunDisjunctiveDepsExistingUpdated(self):
279 self.SetupVartree(self._VARTREE)
280 app1 = 'foo/app1-1.2.5-r2'
281 self.SetupBintree([
282 (app1, '0', '|| ( foo/app6 foo/app2 ) !foo/app3', '1413309350'),
283 ('foo/app2-4.5.6-r7', '0', '', '1413309336'),
284 ])
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700285 installs, listed, num_updates, _ = self.scanner.Run(
David Pursell9476bf42015-03-30 13:34:27 -0700286 self.device, '/', ['app1'], True, True, True)
287 self.ValidatePkgs(installs, [app1])
288 self.ValidatePkgs(listed, [app1])
Mike Frysinger2d589a12019-08-25 14:15:12 -0400289 self.assertEqual(num_updates, 1)
David Pursell9476bf42015-03-30 13:34:27 -0700290
291 def testRunDisjunctiveDepsDefaultUpdated(self):
292 self.SetupVartree(self._VARTREE)
293 app1 = 'foo/app1-1.2.5-r2'
294 app7 = 'foo/app7-1.0.0-r1'
295 self.SetupBintree([
296 (app1, '0', '|| ( foo/app6 foo/app7 ) !foo/app3', '1413309350'),
297 (app7, '0', '', '1413309350'),
298 ])
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700299 installs, listed, num_updates, _ = self.scanner.Run(
David Pursell9476bf42015-03-30 13:34:27 -0700300 self.device, '/', ['app1'], True, True, True)
301 self.ValidatePkgs(installs, [app1, app7], constraints=[(app1, app7)])
302 self.ValidatePkgs(listed, [app1])
Mike Frysinger2d589a12019-08-25 14:15:12 -0400303 self.assertEqual(num_updates, 1)
Ralph Nathane01ccf12015-04-16 10:40:32 -0700304
305
306class TestDeploy(cros_test_lib.ProgressBarTestCase):
307 """Test deploy.Deploy."""
308
Gilad Arnold0e1b1da2015-06-10 06:41:05 -0700309 @staticmethod
310 def FakeGetPackagesByCPV(cpvs, _strip, _sysroot):
311 return ['/path/to/%s.tbz2' % cpv.pv for cpv in cpvs]
312
Ralph Nathane01ccf12015-04-16 10:40:32 -0700313 def setUp(self):
Qijiang Fan8a945032019-04-25 20:53:29 +0900314 self.device = ChromiumOSDeviceHandlerFake()
315 self.PatchObject(
316 remote_access, 'ChromiumOSDeviceHandler', return_value=self.device)
Ralph Nathane01ccf12015-04-16 10:40:32 -0700317 self.PatchObject(cros_build_lib, 'GetBoard', return_value=None)
318 self.PatchObject(cros_build_lib, 'GetSysroot', return_value='sysroot')
319 self.package_scanner = self.PatchObject(deploy, '_InstallPackageScanner')
Gilad Arnold0e1b1da2015-06-10 06:41:05 -0700320 self.get_packages_paths = self.PatchObject(
321 deploy, '_GetPackagesByCPV', side_effect=self.FakeGetPackagesByCPV)
Ralph Nathane01ccf12015-04-16 10:40:32 -0700322 self.emerge = self.PatchObject(deploy, '_Emerge', return_value=None)
323 self.unmerge = self.PatchObject(deploy, '_Unmerge', return_value=None)
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700324 self.PatchObject(deploy, '_GetDLCInfo', return_value=(None, None))
Ralph Nathane01ccf12015-04-16 10:40:32 -0700325
326 def testDeployEmerge(self):
327 """Test that deploy._Emerge is called for each package."""
Gilad Arnold0e1b1da2015-06-10 06:41:05 -0700328
329 _BINPKG = '/path/to/bar-1.2.5.tbz2'
330 def FakeIsFile(fname):
331 return fname == _BINPKG
332
333 packages = ['some/foo-1.2.3', _BINPKG, 'some/foobar-2.0']
Ned Nguyend0db4072019-02-22 14:19:21 -0700334 cpvs = ['some/foo-1.2.3', 'to/bar-1.2.5', 'some/foobar-2.0']
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700335 self.package_scanner.return_value = PackageScannerFake(
336 packages,
337 {'some/foo-1.2.3': {}, _BINPKG: {}, 'some/foobar-2.0': {}},
338 cpvs)
Gilad Arnold0e1b1da2015-06-10 06:41:05 -0700339 self.PatchObject(os.path, 'isfile', side_effect=FakeIsFile)
Ralph Nathane01ccf12015-04-16 10:40:32 -0700340
Gilad Arnold0e1b1da2015-06-10 06:41:05 -0700341 deploy.Deploy(None, ['package'], force=True, clean_binpkg=False)
Ralph Nathane01ccf12015-04-16 10:40:32 -0700342
Gilad Arnold0e1b1da2015-06-10 06:41:05 -0700343 # Check that package names were correctly resolved into binary packages.
344 self.get_packages_paths.assert_called_once_with(
Ned Nguyend0db4072019-02-22 14:19:21 -0700345 [portage_util.SplitCPV(p) for p in cpvs], True, 'sysroot')
Ralph Nathane01ccf12015-04-16 10:40:32 -0700346 # Check that deploy._Emerge is called the right number of times.
347 self.assertEqual(self.emerge.call_count, len(packages))
348 self.assertEqual(self.unmerge.call_count, 0)
Qijiang Fan8a945032019-04-25 20:53:29 +0900349
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700350 def testDeployEmergeDLC(self):
351 """Test that deploy._Emerge installs images for DLC packages."""
352 packages = ['some/foodlc-1.0', 'some/bardlc-2.0']
353 cpvs = ['some/foodlc-1.0', 'some/bardlc-2.0']
354 self.package_scanner.return_value = PackageScannerFake(
355 packages, {'some/foodlc-1.0': {}, 'some/bardlc-2.0': {}}, cpvs)
356 self.PatchObject(deploy, '_GetDLCInfo',
357 return_value=('foo_id', 'foo_package'))
358
359 deploy.Deploy(None, ['package'], force=True, clean_binpkg=False)
360 # Check that dlcservice is restarted (DLC modules are deployed).
361 self.assertTrue(['restart', 'dlcservice'] in self.device.device.cmds)
362
Qijiang Fan8a945032019-04-25 20:53:29 +0900363 def testDeployEmergeSELinux(self):
364 """Test deploy progress when the device has SELinux"""
365
366 _BINPKG = '/path/to/bar-1.2.5.tbz2'
367 def FakeIsFile(fname):
368 return fname == _BINPKG
369
370 def GetRestoreconCommand(pkgfile):
371 remote_path = os.path.join('/testdir/packages/to/', pkgfile)
372 return [['setenforce', '0'],
373 ['cd', '/', '&&',
374 'tar', 'tf', remote_path, '|',
375 'restorecon', '-i', '-f', '-'],
376 ['setenforce', '1']]
377
Ben Pastene5f03b052019-08-12 18:03:24 -0700378 self.device.device.selinux_available = True
Qijiang Fan8a945032019-04-25 20:53:29 +0900379 packages = ['some/foo-1.2.3', _BINPKG, 'some/foobar-2.0']
380 cpvs = ['some/foo-1.2.3', 'to/bar-1.2.5', 'some/foobar-2.0']
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700381 self.package_scanner.return_value = PackageScannerFake(
382 packages,
383 {'some/foo-1.2.3': {}, _BINPKG: {}, 'some/foobar-2.0': {}},
384 cpvs)
Qijiang Fan8a945032019-04-25 20:53:29 +0900385 self.PatchObject(os.path, 'isfile', side_effect=FakeIsFile)
386
387 deploy.Deploy(None, ['package'], force=True, clean_binpkg=False)
388
389 # Check that package names were correctly resolved into binary packages.
390 self.get_packages_paths.assert_called_once_with(
391 [portage_util.SplitCPV(p) for p in cpvs], True, 'sysroot')
392 # Check that deploy._Emerge is called the right number of times.
393 self.assertEqual(self.emerge.call_count, len(packages))
394 self.assertEqual(self.unmerge.call_count, 0)
395
396 self.assertEqual(self.device.device.cmds,
397 GetRestoreconCommand('foo-1.2.3.tbz2') +
398 GetRestoreconCommand('bar-1.2.5.tbz2') +
399 GetRestoreconCommand('foobar-2.0.tbz2'))
Ralph Nathane01ccf12015-04-16 10:40:32 -0700400
401 def testDeployUnmerge(self):
402 """Test that deploy._Unmerge is called for each package."""
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700403 packages = ['foo', 'bar', 'foobar', 'foodlc']
404 self.package_scanner.return_value = PackageScannerFake(
405 packages, {'foo': {}, 'bar': {}, 'foobar': {},
406 'foodlc': {deploy._DLC_ID: 'foodlc',
407 deploy._DLC_PACKAGE: 'foopackage'}})
Ralph Nathane01ccf12015-04-16 10:40:32 -0700408
Gilad Arnold0e1b1da2015-06-10 06:41:05 -0700409 deploy.Deploy(None, ['package'], force=True, clean_binpkg=False,
Ralph Nathane01ccf12015-04-16 10:40:32 -0700410 emerge=False)
411
412 # Check that deploy._Unmerge is called the right number of times.
413 self.assertEqual(self.emerge.call_count, 0)
414 self.assertEqual(self.unmerge.call_count, len(packages))
415
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700416 self.assertEqual(
417 self.device.device.cmds,
Jae Hoon Kim964ed7e2020-05-15 13:59:23 -0700418 [['dlcservice_util', '--uninstall', '--id=foodlc'],
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700419 ['restart', 'dlcservice']])
420
Ralph Nathane01ccf12015-04-16 10:40:32 -0700421 def testDeployMergeWithProgressBar(self):
422 """Test that BrilloDeployOperation.Run() is called for merge."""
423 packages = ['foo', 'bar', 'foobar']
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700424 self.package_scanner.return_value = PackageScannerFake(
425 packages, {'foo': {}, 'bar': {}, 'foobar': {}})
Ralph Nathane01ccf12015-04-16 10:40:32 -0700426
427 run = self.PatchObject(deploy.BrilloDeployOperation, 'Run',
428 return_value=None)
429
430 self.PatchObject(command, 'UseProgressBar', return_value=True)
Gilad Arnold0e1b1da2015-06-10 06:41:05 -0700431 deploy.Deploy(None, ['package'], force=True, clean_binpkg=False)
Ralph Nathane01ccf12015-04-16 10:40:32 -0700432
433 # Check that BrilloDeployOperation.Run was called.
434 self.assertTrue(run.called)
435
436 def testDeployUnmergeWithProgressBar(self):
437 """Test that BrilloDeployOperation.Run() is called for unmerge."""
438 packages = ['foo', 'bar', 'foobar']
Xiaochu Liu2726e7c2019-07-18 10:28:10 -0700439 self.package_scanner.return_value = PackageScannerFake(
440 packages, {'foo': {}, 'bar': {}, 'foobar': {}})
Ralph Nathane01ccf12015-04-16 10:40:32 -0700441
442 run = self.PatchObject(deploy.BrilloDeployOperation, 'Run',
443 return_value=None)
444
445 self.PatchObject(command, 'UseProgressBar', return_value=True)
Gilad Arnold0e1b1da2015-06-10 06:41:05 -0700446 deploy.Deploy(None, ['package'], force=True, clean_binpkg=False,
Ralph Nathane01ccf12015-04-16 10:40:32 -0700447 emerge=False)
448
449 # Check that BrilloDeployOperation.Run was called.
450 self.assertTrue(run.called)
451
452 def testBrilloDeployMergeOperation(self):
453 """Test that BrilloDeployOperation works for merge."""
454 def func(queue):
Ralph Nathan90475a12015-05-20 13:19:01 -0700455 for event in op.MERGE_EVENTS:
Ralph Nathane01ccf12015-04-16 10:40:32 -0700456 queue.get()
457 print(event)
Mike Frysinger050fa5a2019-12-01 16:16:29 -0500458 sys.stdout.flush()
Ralph Nathane01ccf12015-04-16 10:40:32 -0700459
460 queue = multiprocessing.Queue()
461 # Emerge one package.
462 op = BrilloDeployOperationFake(1, True, queue)
463
464 with self.OutputCapturer():
465 op.Run(func, queue)
466
467 # Check that the progress bar prints correctly.
Ralph Nathan90475a12015-05-20 13:19:01 -0700468 self.AssertProgressBarAllEvents(len(op.MERGE_EVENTS))
Ralph Nathane01ccf12015-04-16 10:40:32 -0700469
470 def testBrilloDeployUnmergeOperation(self):
471 """Test that BrilloDeployOperation works for unmerge."""
472 def func(queue):
Ralph Nathan90475a12015-05-20 13:19:01 -0700473 for event in op.UNMERGE_EVENTS:
Ralph Nathane01ccf12015-04-16 10:40:32 -0700474 queue.get()
475 print(event)
Mike Frysinger050fa5a2019-12-01 16:16:29 -0500476 sys.stdout.flush()
Ralph Nathane01ccf12015-04-16 10:40:32 -0700477
478 queue = multiprocessing.Queue()
479 # Unmerge one package.
480 op = BrilloDeployOperationFake(1, False, queue)
481
482 with self.OutputCapturer():
483 op.Run(func, queue)
484
485 # Check that the progress bar prints correctly.
Ralph Nathan90475a12015-05-20 13:19:01 -0700486 self.AssertProgressBarAllEvents(len(op.UNMERGE_EVENTS))