1 files changed, 80 insertions, 0 deletions
diff --git a/tests/integration_tests/datasources/test_lxd_discovery.py b/tests/integration_tests/datasources/test_lxd_discovery.py
new file mode 100644
index 00000000..f7d8bfc3
--- /dev/null
+++ b/tests/integration_tests/datasources/test_lxd_discovery.py
@@ -0,0 +1,80 @@
+import json
+import pytest
+import yaml
+
+from tests.integration_tests.clouds import IntegrationCloud
+from tests.integration_tests.conftest import get_validated_source
+from tests.integration_tests.util import verify_clean_log
+
+
+def _setup_custom_image(session_cloud: IntegrationCloud):
+    """Like `setup_image` in conftest.py, but with customized content."""
+    source = get_validated_source(session_cloud)
+    if not source.installs_new_version():
+        return
+    client = session_cloud.launch()
+
+    # Insert our "detect_lxd_ds" file here
+    client.write_to_file(
+        '/etc/cloud/cloud.cfg.d/99-detect-lxd.cfg',
+        'datasource_list: [LXD]\n',
+    )
+
+    client.execute('rm -f /etc/netplan/50-cloud-init.yaml')
+
+    client.install_new_cloud_init(source)
+    # Even if we're keeping instances, we don't want to keep this
+    # one around as it was just for image creation
+    client.destroy()
+
+
+# This test should be able to work on any cloud whose datasource specifies
+# a NETWORK dependency
+@pytest.mark.lxd_container
+@pytest.mark.lxd_vm
+@pytest.mark.ubuntu  # Because netplan
+def test_lxd_datasource_discovery(session_cloud: IntegrationCloud):
+    """Test that DataSourceLXD is detected instead of NoCloud."""
+    _setup_custom_image(session_cloud)
+    nic_dev = "eth0"
+    if session_cloud.settings.PLATFORM == "lxd_vm":
+        nic_dev = "enp5s0"
+
+    with session_cloud.launch() as client:
+        result = client.execute('cloud-init status --long')
+        if not result.ok:
+            raise AssertionError('cloud-init failed:\n%s',
+                                 result.stderr)
+        if "DataSourceLXD" not in result.stdout:
+            raise AssertionError(
+                'cloud-init did not discover DataSourceLXD', result.stdout
+            )
+        netplan_yaml = client.execute('cat /etc/netplan/50-cloud-init.yaml')
+        netplan_cfg = yaml.safe_load(netplan_yaml)
+        assert {
+            'network': {'ethernets': {nic_dev: {'dhcp4': True}}, 'version': 2}
+        } == netplan_cfg
+        log = client.read_from_file('/var/log/cloud-init.log')
+        verify_clean_log(log)
+        result = client.execute('cloud-id')
+        if "lxd" != result.stdout:
+            raise AssertionError(
+                "cloud-id didn't report lxd. Result: %s", result.stdout
+            )
+        # Validate config instance data represented
+        data = json.loads(client.read_from_file(
+            '/run/cloud-init/instance-data.json')
+        )
+        v1 = data["v1"]
+        ds_cfg = data["ds"]
+        assert "lxd" == v1["platform"]
+        assert "LXD socket API v. 1.0 (/dev/lxd/sock)" == v1["subplatform"]
+        ds_cfg = json.loads(client.execute('cloud-init query ds').stdout)
+        assert ["config", "meta_data"] == sorted(list(ds_cfg["1.0"].keys()))
+        assert ["user.meta_data"] == list(ds_cfg["1.0"]["config"].keys())
+        assert {"public-keys": v1["public_ssh_keys"][0]} == (
+            yaml.safe_load(ds_cfg["1.0"]["config"]["user.meta_data"])
+        )
+        assert (
+            "#cloud-config\ninstance-id" in ds_cfg["1.0"]["meta_data"]
+        )