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
|
"""Integration tests for various handlers."""
from io import StringIO
import pytest
from cloudinit.cmd.devel.make_mime import create_mime_message
from tests.integration_tests.instances import IntegrationInstance
PER_FREQ_TEMPLATE = """\
#!/bin/bash
touch /tmp/test_per_freq_{}
"""
PER_ALWAYS_FILE = StringIO(PER_FREQ_TEMPLATE.format("always"))
PER_INSTANCE_FILE = StringIO(PER_FREQ_TEMPLATE.format("instance"))
PER_ONCE_FILE = StringIO(PER_FREQ_TEMPLATE.format("once"))
FILES = [
(PER_ALWAYS_FILE, "always.sh", "x-shellscript-per-boot"),
(PER_INSTANCE_FILE, "instance.sh", "x-shellscript-per-instance"),
(PER_ONCE_FILE, "once.sh", "x-shellscript-per-once"),
]
USER_DATA = create_mime_message(FILES)[0].as_string()
@pytest.mark.ci
@pytest.mark.user_data(USER_DATA)
def test_per_freq(client: IntegrationInstance):
# Sanity test for scripts folder
cmd = "test -d /var/lib/cloud/scripts"
assert client.execute(cmd).ok
# Test per-boot
cmd = "test -f /var/lib/cloud/scripts/per-boot/always.sh"
assert client.execute(cmd).ok
cmd = "test -f /tmp/test_per_freq_always"
assert client.execute(cmd).ok
# Test per-instance
cmd = "test -f /var/lib/cloud/scripts/per-instance/instance.sh"
assert client.execute(cmd).ok
cmd = "test -f /tmp/test_per_freq_instance"
assert client.execute(cmd).ok
# Test per-once
cmd = "test -f /var/lib/cloud/scripts/per-once/once.sh"
assert client.execute(cmd).ok
cmd = "test -f /tmp/test_per_freq_once"
assert client.execute(cmd).ok
|
