1
0
mirror of https://gitlab.alpinelinux.org/alpine/cloud/tiny-cloud.git synced 2025-12-16 03:42:44 +03:00
tiny-cloud/tests/init-main.test
Natanael Copa ab46d7223f tests: add test for save_userdata
Test that compressed user-data gets properly detected
2023-04-06 17:19:40 +02:00

107 lines
2.3 KiB
Plaintext
Executable File

#!/usr/bin/env atf-sh
. $(atf_get_srcdir)/test_env.sh
export PREFIX="$srcdir"
export MOCK=echo
lib="$srcdir"/lib/tiny-cloud/init-main
init_tests \
set_hostname \
set_ssh_keys \
save_userdata_plain \
save_userdata_compressed
set_nocloud_meta() {
local metadata="$(cat)"
fake_bin mount <<-EOF
#!/bin/sh
# find last arg which is the mount dir
while ! [ -d "\$1" ]; do
shift
done
printf "%s" "$metadata" \
> "\$1"/meta-data
EOF
fake_bin umount <<-EOF
#!/bin/sh
while ! [ -d "\$1" ]; do
shift
done
rm -f "\$1"/meta-data
EOF
mkdir -p mnt
}
set_nocloud_userdata_from_file() {
local file="$1"
fake_bin mount <<-EOF
#!/bin/sh
# find last arg which is the mount dir
while ! [ -d "\$1" ]; do
shift
done
cp "$file" "\$1"/user-data
EOF
mkdir -p mnt
}
set_hostname_body() {
set_nocloud_meta <<-EOF
hostname: myhostname
EOF
CLOUD="nocloud" atf_check \
-o match:"hostname.*-F $PWD/etc/hostname" \
sh -c ". \"$lib\"; set_hostname"
atf_check -o match:"^myhostname$" cat etc/hostname
rm etc/hostname
CLOUD="nocloud" atf_check \
-o not-match:"hostname.*-F $PWD/etc/hostname" \
sh -c "SKIP_INIT_ACTIONS=set_hostname ; . \"$lib\"; set_hostname"
if test -f etc/hostname; then
atf_fail "etc/hostname should not exist"
fi
}
set_ssh_keys_body() {
fake_bin getent <<-EOF
#!/bin/sh
echo "alpine:x:1000:1000:Linux User,,,:/home/alpine:/bin/sh"
EOF
set_nocloud_meta user-data <<-EOF
public-keys:
- openssh-key: ssh-ed25519 keydata
- openssh-key: ssh-rsa foobar
EOF
CLOUD="nocloud" atf_check \
-o match:"chown.*/\.ssh" \
sh -c ". \"$lib\"; set_ssh_keys"
atf_check -o match:"^ssh-ed25519 keydata" \
-o match:"^ssh-rsa foobar" \
cat home/alpine/.ssh/authorized_keys
}
save_userdata_plain_body() {
printf "%s\n" "#cloud-config" > tmpfile
set_nocloud_userdata_from_file tmpfile
CLOUD="nocloud" atf_check \
sh -c ". \"$lib\"; save_userdata"
atf_check -o match:"^#cloud-config" cat var/lib/cloud/user-data
}
save_userdata_compressed_body() {
for comp in gzip bzip2 xz lzma lzop lz4 zstd; do
printf "%s\n" "#cloud-config" | $comp -c > tmpfile
set_nocloud_userdata_from_file tmpfile
CLOUD="nocloud" atf_check \
sh -c ". \"$lib\"; save_userdata"
grep "^#cloud-config" var/lib/cloud/user-data \
|| atf_fail "$comp failed"
done
}