def test_return_empty_list_if_package_not_available(self):
g = None
a = Apt(_mock_run(
['apt-cache', 'madison', 'cloud-init'],
CompletedProcess(
'N: Unable to locate package cloud-init', '', 1, 'cmd')))
assert a.list_available_versions(g, 'cloud-init') == set()
def test_skip_lines_from_madison_with_unexpected_syntax(self):
g = None
stdout = '\n'.join([
'cloud-init | 21.3-1-g6803368d-0ubuntu1~20.04.3 | repo',
'cloud-init | extra bar | repo | not-recognized',
'cloud-init | missing bar',
'no bar',
])
a = Apt(_mock_run(['apt-cache', 'madison', 'cloud-init'],
CompletedProcess(stdout, '', 0, 'cmd')))
assert a.list_available_versions(g, 'cloud-init') == {
'21.3-1-g6803368d-0ubuntu1~20.04.3'}
def test_return_versions_if_available(self):
g = None
stdout = '\n'.join([
'cloud-init | 21.3-1-g6803368d-0ubuntu1~20.04.3 | repo',
'cloud-init | 20.1-10-g71af48df-0ubuntu5 | repo',
'cloud-init | 0.7.5-0ubuntu1.23 | repo',
])
a = Apt(_mock_run(['apt-cache', 'madison', 'cloud-init'],
CompletedProcess(stdout, '', 0, 'cmd')))
assert a.list_available_versions(g, 'cloud-init') == {
'21.3-1-g6803368d-0ubuntu1~20.04.3',
'20.1-10-g71af48df-0ubuntu5',
'0.7.5-0ubuntu1.23'}
def setup_cloud_init(g: guestfs.GuestFS):
""" Install cloud-init if not present, and configure to the cloud provider.
Args:
g: A mounted GuestFS instance.
"""
a = Apt(run)
curr_version = a.get_package_version(g, 'cloud-init')
available_versions = a.list_available_versions(g, 'cloud-init')
# Try to avoid installing 21.3-1, which conflicts which the guest agent.
# On first boot, systemd reaches a deadlock deadlock and doest start
# its unit. If a version other than 21.3-1 isn't explicitly found, *and*
# cloud-init isn't currently installed, then this allows apt to pick the
# version to install.
version_to_install = Apt.determine_version_to_install(
curr_version, available_versions, {'21.3-1'})
pkg_to_install = ''
if version_to_install:
pkg_to_install = 'cloud-init=' + version_to_install
elif curr_version == '':
pkg_to_install = 'cloud-init'
# If this block doesn't execute, it means that cloud-init was found
# on the system, but there wasn't an upgrade candidate. Therefore
# leave the version that's currently installed.
if pkg_to_install:
logging.info(pkg_to_install)
utils.install_apt_packages(g, pkg_to_install)
# Ubuntu 14.04's version of cloud-init doesn't have `clean`.
if g.gcp_image_major > '14':
run(g, 'cloud-init clean')
# Remove cloud-init configs that may conflict with GCE's.
#
# - subiquity disables automatic network configuration
# https://bugs.launchpad.net/ubuntu/+source/cloud-init/+bug/1871975
for cfg in [
'azure', 'curtin', 'waagent', 'walinuxagent', 'aws', 'amazon',
'subiquity'
]:
run(g, 'rm -f /etc/cloud/cloud.cfg.d/*%s*' % cfg)
g.write('/etc/cloud/cloud.cfg.d/91-gce-system.cfg', cloud_init_config)