From fa01b4ea9a66aebcca04bf2d5205045fded1ad04 Mon Sep 17 00:00:00 2001
From: Natanael Copa <ncopa@alpinelinux.org>
Date: Thu, 6 Apr 2023 22:08:11 +0200
Subject: [PATCH] tests: add tests for init-final

---
 tests/init-final.test | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100755 tests/init-final.test

diff --git a/tests/init-final.test b/tests/init-final.test
new file mode 100755
index 0000000..b2053fd
--- /dev/null
+++ b/tests/init-final.test
@@ -0,0 +1,42 @@
+#!/usr/bin/env atf-sh
+
+. $(atf_get_srcdir)/test_env.sh
+
+export PREFIX="$srcdir"
+export MOCK=echo
+lib="$srcdir"/lib/tiny-cloud/init-final
+init_main="$srcdir"/lib/tiny-cloud/init-main
+PROVIDERS="aws azure gcp nocloud oci"
+
+init_tests \
+	is_userdata_script \
+	run_userdata
+
+
+is_userdata_script_body() {
+	mkdir -p var/lib/cloud
+	for c in $PROVIDERS; do
+		echo "#tiny-cloud-config" > var/lib/cloud/user-data
+		CLOUD="$c" atf_check -s not-exit:0 \
+			sh -c ". \"$lib\"; is_userdata_script"
+	
+		echo "#!/bin/sh" > var/lib/cloud/user-data
+		CLOUD="$c" atf_check -s exit:0 \
+			sh -c ". \"$lib\"; is_userdata_script"
+	done
+}
+
+run_userdata_body() {
+	fake_userdata_nocloud <<-EOF
+		#!/bin/sh
+		echo "hello from user-data"
+	EOF
+	CLOUD="nocloud" atf_check \
+		sh -c ". \"$init_main\"; save_userdata"
+	CLOUD="nocloud" atf_check \
+		-o match:"hello from user-data" \
+		sh -c ". \"$lib\"; run_userdata"
+	grep "hello from user-data" var/log/user-data.log || atf_fail "user-data.log failed"
+	grep -w "0" var/log/user-data.exit || atf_fail "user-data.exit failed"
+}
+