| #!/usr/bin/env python3 | |
| # | |
| # Ubuntu VM testing library | |
| # | |
| # Copyright 2017 Red Hat Inc. | |
| # Copyright 2020 Linaro | |
| # | |
| # Authors: | |
| # Robert Foley <robert.foley@linaro.org> | |
| # Originally based on ubuntu.i386 Fam Zheng <famz@redhat.com> | |
| # | |
| # This code is licensed under the GPL version 2 or later. See | |
| # the COPYING file in the top-level directory. | |
| import os | |
| import subprocess | |
| import basevm | |
| class UbuntuVM(basevm.BaseVM): | |
| def __init__(self, args, config=None): | |
| self.login_prompt = "ubuntu-{}-guest login:".format(self.arch) | |
| basevm.BaseVM.__init__(self, args, config) | |
| def build_image(self, img): | |
| """Build an Ubuntu VM image. The child class will | |
| define the install_cmds to init the VM.""" | |
| os_img = self._download_with_cache(self.image_link, | |
| sha256sum=self.image_sha256) | |
| img_tmp = img + ".tmp" | |
| subprocess.check_call(["cp", "-f", os_img, img_tmp]) | |
| self.exec_qemu_img("resize", img_tmp, "+50G") | |
| ci_img = self.gen_cloud_init_iso() | |
| self.boot(img_tmp, extra_args = [ "-device", "VGA", "-cdrom", ci_img, ]) | |
| # First command we issue is fix for slow ssh login. | |
| self.wait_ssh(wait_root=True, | |
| cmd="chmod -x /etc/update-motd.d/*") | |
| # Wait for cloud init to finish | |
| self.wait_ssh(wait_root=True, | |
| cmd="ls /var/lib/cloud/instance/boot-finished") | |
| self.ssh_root("touch /etc/cloud/cloud-init.disabled") | |
| # Disable auto upgrades. | |
| # We want to keep the VM system state stable. | |
| self.ssh_root('sed -ie \'s/"1"/"0"/g\' '\ | |
| '/etc/apt/apt.conf.d/20auto-upgrades') | |
| self.ssh_root("sed -ie s/^#\ deb-src/deb-src/g /etc/apt/sources.list") | |
| # If the user chooses not to do the install phase, | |
| # then we will jump right to the graceful shutdown | |
| if self._config['install_cmds'] != "": | |
| # Issue the install commands. | |
| # This can be overridden by the user in the config .yml. | |
| install_cmds = self._config['install_cmds'].split(',') | |
| for cmd in install_cmds: | |
| self.ssh_root(cmd) | |
| self.graceful_shutdown() | |
| os.rename(img_tmp, img) | |
| return 0 |