integration_tests: add UPGRADE CloudInitSource

tests/integration_tests/conftest.py

@@ -124,6 +124,8 @@ def get_validated_source(
         return CloudInitSource.PPA
     elif os.path.isfile(str(source)):
         return CloudInitSource.DEB_PACKAGE
+    elif source == "UPGRADE":
+        return CloudInitSource.UPGRADE
     raise ValueError(
         'Invalid value for CLOUD_INIT_SOURCE setting: {}'.format(source))
 

tests/integration_tests/instances.py

@@ -39,6 +39,7 @@ class CloudInitSource(Enum):
     PROPOSED = 3
     PPA = 4
     DEB_PACKAGE = 5
+    UPGRADE = 6
 
     def installs_new_version(self):
         if self.name in [self.NONE.name, self.IN_PLACE.name]:
@@ -123,6 +124,8 @@ def install_new_cloud_init(
             self.install_ppa()
         elif source == CloudInitSource.PROPOSED:
             self.install_proposed_image()
+        elif source == CloudInitSource.UPGRADE:
+            self.upgrade_cloud_init()
         else:
             raise Exception(
                 "Specified to install {} which isn't supported here".format(
@@ -166,6 +169,11 @@ def install_deb(self):
         remote_script = 'dpkg -i {path}'.format(path=remote_path)
         self.execute(remote_script)
 
+    def upgrade_cloud_init(self):
+        log.info('Upgrading cloud-init to latest version in archive')
+        self.execute("apt-get update -q")
+        self.execute("apt-get install -qy cloud-init")
+
     def __enter__(self):
         return self
 

tests/integration_tests/integration_settings.py

@@ -59,6 +59,8 @@
 #   code.
 # PROPOSED
 #   Install from the Ubuntu proposed repo
+# UPGRADE
+#   Upgrade cloud-init to the version in the Ubuntu archive
 # <ppa repo>, e.g., ppa:cloud-init-dev/proposed
 #   Install from a PPA. It MUST start with 'ppa:'
 # <file path>