tests/vm/ubuntuvm.py

   1 #!/usr/bin/env python3
   2 #
   3 # Ubuntu VM testing library
   4 #
   5 # Copyright 2017 Red Hat Inc.
   6 # Copyright 2020 Linaro
   7 #
   8 # Authors:
   9 #  Robert Foley <robert.foley@linaro.org>
  10 #  Originally based on ubuntu.i386 Fam Zheng <famz@redhat.com>
  11 #
  12 # This code is licensed under the GPL version 2 or later.  See
  13 # the COPYING file in the top-level directory.
  14
  15 import os
  16 import subprocess
  17 import basevm
  18
  19 class UbuntuVM(basevm.BaseVM):
  20
  21     def __init__(self, args, config=None):
  22         self.login_prompt = "ubuntu-{}-guest login:".format(self.arch)
  23         basevm.BaseVM.__init__(self, args, config)
  24
  25     def build_image(self, img):
  26         """Build an Ubuntu VM image.  The child class will
  27            define the install_cmds to init the VM."""
  28         os_img = self._download_with_cache(self.image_link,
  29                                            sha256sum=self.image_sha256)
  30         img_tmp = img + ".tmp"
  31         subprocess.check_call(["cp", "-f", os_img, img_tmp])
  32         self.exec_qemu_img("resize", img_tmp, "+50G")
  33         ci_img = self.gen_cloud_init_iso()
  34
  35         self.boot(img_tmp, extra_args = [ "-device", "VGA", "-cdrom", ci_img, ])
  36
  37         # First command we issue is fix for slow ssh login.
  38         self.wait_ssh(wait_root=True,
  39                       cmd="chmod -x /etc/update-motd.d/*")
  40         # Wait for cloud init to finish
  41         self.wait_ssh(wait_root=True,
  42                       cmd="ls /var/lib/cloud/instance/boot-finished")
  43         self.ssh_root("touch /etc/cloud/cloud-init.disabled")
  44         # Disable auto upgrades.
  45         # We want to keep the VM system state stable.
  46         self.ssh_root('sed -ie \'s/"1"/"0"/g\' '\
  47                       '/etc/apt/apt.conf.d/20auto-upgrades')
  48         self.ssh_root("sed -ie s/^#\ deb-src/deb-src/g /etc/apt/sources.list")
  49
  50         # If the user chooses not to do the install phase,
  51         # then we will jump right to the graceful shutdown
  52         if self._config['install_cmds'] != "":
  53             # Issue the install commands.
  54             # This can be overridden by the user in the config .yml.
  55             install_cmds = self._config['install_cmds'].split(',')
  56             for cmd in install_cmds:
  57                 self.ssh_root(cmd)
  58         self.graceful_shutdown()
  59         os.rename(img_tmp, img)
  60         return 0