summaryrefslogtreecommitdiff
path: root/tests/integration_tests/datasources/test_lxd_discovery.py
blob: f7d8bfc3f8993e58fff2a78e0ae683ad878f2ec6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
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"]
        )