def testUpdateChroot(self):
"""Test the update_chroot related handling."""
# Prevent it from doing anything else for this test.
self.PatchObject(sysroot, '_CreateSysrootSkeleton')
self.PatchObject(sysroot, '_InstallConfigs')
self.PatchObject(sysroot, '_InstallPortageConfigs')
# Make sure we have a board we haven't setup to avoid triggering the
# existing sysroot logic. That is entirely unrelated to the chroot update.
target = self.unbuilt_target
# Test no update case.
config = sysroot.SetupBoardRunConfig(upgrade_chroot=False)
get_args_patch = self.PatchObject(config, 'GetUpdateChrootArgs')
sysroot.Create(target, config, None)
# The update chroot args not being fetched is a
# strong enough signal that the update wasn't run.
get_args_patch.assert_not_called()
# Test update case.
script_loc = os.path.join(constants.CROSUTILS_DIR, 'update_chroot')
config = sysroot.SetupBoardRunConfig(upgrade_chroot=True)
sysroot.Create(target, config, None)
self.assertCommandContains([script_loc])
def testForce(self):
"""Test the force flag."""
# Prevent it from doing anything else for this test.
self.PatchObject(sysroot, '_CreateSysrootSkeleton')
self.PatchObject(sysroot, '_InstallConfigs')
self.PatchObject(sysroot, '_InstallPortageConfigs')
delete_patch = self.PatchObject(sysroot_lib.Sysroot, 'Delete')
config = sysroot.SetupBoardRunConfig(force=False)
sysroot.Create(self.build_target, config, None)
delete_patch.assert_not_called()
config = sysroot.SetupBoardRunConfig(force=True)
sysroot.Create(self.build_target, config, None)
delete_patch.assert_called_once()
def Create(input_proto, output_proto, _config):
"""Create or replace a sysroot."""
update_chroot = not input_proto.flags.chroot_current
replace_sysroot = input_proto.flags.replace
build_target = controller_util.ParseBuildTarget(input_proto.build_target,
input_proto.profile)
package_indexes = [
binpkg.PackageIndexInfo.from_protobuf(x)
for x in input_proto.package_indexes
]
run_configs = sysroot.SetupBoardRunConfig(force=replace_sysroot,
upgrade_chroot=update_chroot,
package_indexes=package_indexes)
try:
created = sysroot.Create(build_target,
run_configs,
accept_licenses=_ACCEPTED_LICENSES)
except sysroot.Error as e:
cros_build_lib.Die(e)
output_proto.sysroot.path = created.path
output_proto.sysroot.build_target.name = build_target.name
return controller.RETURN_CODE_SUCCESS
def Create(input_proto, output_proto, _config):
"""Create or replace a sysroot."""
update_chroot = not input_proto.flags.chroot_current
replace_sysroot = input_proto.flags.replace
build_target_name = input_proto.build_target.name
profile = input_proto.profile.name or None
build_target = build_target_util.BuildTarget(name=build_target_name,
profile=profile)
run_configs = sysroot.SetupBoardRunConfig(force=replace_sysroot,
upgrade_chroot=update_chroot)
try:
created = sysroot.Create(build_target,
run_configs,
accept_licenses=_ACCEPTED_LICENSES)
except sysroot.Error as e:
cros_build_lib.Die(e)
output_proto.sysroot.path = created.path
output_proto.sysroot.build_target.name = build_target_name
return controller.RETURN_CODE_SUCCESS