1 files changed, 68 insertions, 0 deletions
diff --git a/tests/integration_tests/modules/test_growpart.py b/tests/integration_tests/modules/test_growpart.py
new file mode 100644
index 00000000..67251817
--- /dev/null
+++ b/tests/integration_tests/modules/test_growpart.py
@@ -0,0 +1,68 @@
+import json
+import os
+import pathlib
+from uuid import uuid4
+
+import pytest
+from pycloudlib.lxd.instance import LXDInstance
+
+from cloudinit.subp import subp
+from tests.integration_tests.instances import IntegrationInstance
+
+DISK_PATH = "/tmp/test_disk_setup_{}".format(uuid4())
+
+
+def setup_and_mount_lxd_disk(instance: LXDInstance):
+    subp(
+        "lxc config device add {} test-disk-setup-disk disk source={}".format(
+            instance.name, DISK_PATH
+        ).split()
+    )
+
+
+@pytest.fixture(scope="class", autouse=True)
+def create_disk():
+    """Create 16M sparse file"""
+    pathlib.Path(DISK_PATH).touch()
+    os.truncate(DISK_PATH, 1 << 24)
+    yield
+    os.remove(DISK_PATH)
+
+
+# Create undersized partition in bootcmd
+ALIAS_USERDATA = """\
+#cloud-config
+bootcmd:
+  - parted /dev/sdb --script                \
+          mklabel gpt                       \
+          mkpart primary 0 1MiB
+  - parted /dev/sdb --script print
+growpart:
+  devices:
+  - "/"
+  - "/dev/sdb1"
+runcmd:
+  - parted /dev/sdb --script print
+"""
+
+
+@pytest.mark.user_data(ALIAS_USERDATA)
+@pytest.mark.lxd_setup.with_args(setup_and_mount_lxd_disk)
+@pytest.mark.ubuntu
+@pytest.mark.lxd_vm
+class TestGrowPart:
+    """Test growpart"""
+
+    def test_grow_part(self, client: IntegrationInstance):
+        """Verify"""
+        log = client.read_from_file("/var/log/cloud-init.log")
+        assert (
+            "cc_growpart.py[INFO]: '/dev/sdb1' resized:"
+            " changed (/dev/sdb, 1) from" in log
+        )
+
+        lsblk = json.loads(client.execute("lsblk --json"))
+        sdb = [x for x in lsblk["blockdevices"] if x["name"] == "sdb"][0]
+        assert len(sdb["children"]) == 1
+        assert sdb["children"][0]["name"] == "sdb1"
+        assert sdb["size"] == "16M"