Test that after-reset is run when in UKI mode (#2331)

* Test that after-reset is run when in UKI mode Depends on this: kairos-io/kairos-agent#245 Signed-off-by: Dimitris Karakasilis <[email protected]> * Select recovery boot entry correctly and add "debug" and "strict" to the uki-install config yaml Signed-off-by: Dimitris Karakasilis <[email protected]> * Rename db.* uki keys so that latest enki finds them Signed-off-by: Dimitris Karakasilis <[email protected]> * Print debug output (because test fails only in CI) Signed-off-by: Dimitris Karakasilis <[email protected]> * Use the correct config for uki tests Signed-off-by: Dimitris Karakasilis <[email protected]> * Remove debug output (found the issue) Signed-off-by: Dimitris Karakasilis <[email protected]> --------- Signed-off-by: Dimitris Karakasilis <[email protected]>
kairos-io · Mar 27, 2024 · 19919bc · 19919bc
1 parent 3cc8bbb
commit 19919bc
Show file tree

Hide file tree

Showing 9 changed files with 56 additions and 3 deletions.
diff --git a/.github/workflows/uki.yaml b/.github/workflows/uki.yaml
@@ -162,7 +162,7 @@ jobs:
           earthly +uki-iso --BASE_IMAGE=ttl.sh/ubuntu-2310-${{ github.head_ref || github.ref }}:24h
       - name: Create datasource iso 🔧
         run: |
-          earthly +datasource-iso --CLOUD_CONFIG=tests/assets/autoinstall.yaml
+          earthly +datasource-iso --CLOUD_CONFIG=tests/assets/uki-install.yaml
       - name: Run tests 🔧
         env:
           USE_QEMU: true
@@ -234,7 +234,7 @@ jobs:
             --BASE_IMAGE=ttl.sh/fedora-38-${{ github.head_ref || github.ref }}:24h
       - name: Create datasource iso 🔧
         run: |
-          earthly +datasource-iso --CLOUD_CONFIG=tests/assets/autoinstall.yaml
+          earthly +datasource-iso --CLOUD_CONFIG=tests/assets/uki-install.yaml
       - name: Run tests 🔧
         env:
           USE_QEMU: true

diff --git a/.github/workflows/uki_branches.yaml b/.github/workflows/uki_branches.yaml
@@ -72,7 +72,7 @@ jobs:
           earthly +uki-iso --BASE_IMAGE=ttl.sh/ubuntu-2310-${{ github.sha }}:24h
       - name: Create datasource iso 🔧
         run: |
-          earthly +datasource-iso --CLOUD_CONFIG=tests/assets/autoinstall.yaml
+          earthly +datasource-iso --CLOUD_CONFIG=tests/assets/uki-install.yaml
       - name: Run tests 🔧
         env:
           USE_QEMU: true

diff --git a/tests/assets/uki-install.yaml b/tests/assets/uki-install.yaml
@@ -0,0 +1,36 @@
+#cloud-config
+
+install:
+  auto: true
+  reboot: true
+  device: /dev/vda
+  grub_options:
+    extra_cmdline: "rd.immucore.debug foobarzz"
+
+debug: true
+strict: true
+
+stages:
+  initramfs:
+    - name: "Set user and password"
+      users:
+        kairos:
+          passwd: "kairos"
+      hostname: kairos-{{ trunc 4 .Random }}
+  after-reset:
+      - commands:
+        - |
+          /bin/bash <<'EOF'
+          #!/bin/bash
+
+          set -e
+          echo "Creating a file in COS_PERSISTENT"
+          umount /oem || true
+          # Close all encrypted partitions
+          for p in $(ls /dev/mapper/vda*); do cryptsetup close $p; done
+          /usr/lib/systemd/systemd-cryptsetup attach persistent $(findfs PARTLABEL=persistent) - tpm2-device=auto
+          mount /dev/mapper/persistent /usr/local
+          touch /usr/local/after-reset-file
+          umount /dev/mapper/persistent
+          cryptsetup close /dev/mapper/persistent
+          EOF
diff --git a/tests/keys/DB.auth → tests/keys/db.auth b/tests/keys/DB.auth → tests/keys/db.auth
diff --git a/tests/keys/DB.der → tests/keys/db.der b/tests/keys/DB.der → tests/keys/db.der
diff --git a/tests/keys/DB.esl → tests/keys/db.esl b/tests/keys/DB.esl → tests/keys/db.esl
diff --git a/tests/keys/DB.key → tests/keys/db.key b/tests/keys/DB.key → tests/keys/db.key
diff --git a/tests/keys/DB.pem → tests/keys/db.pem b/tests/keys/DB.pem → tests/keys/db.pem
diff --git a/tests/uki_test.go b/tests/uki_test.go
@@ -153,5 +153,22 @@ var _ = Describe("kairos UKI test", Label("uki"), Ordered, func() {
 			stateContains(vm, "system.os.name", "alpine", "opensuse", "ubuntu", "debian", "fedora")
 			stateContains(vm, "kairos.flavor", "alpine", "opensuse", "ubuntu", "debian", "fedora")
 		})
+
+		By("rebooting to recovery")
+		out, err := vm.Sudo("kairos-agent bootentry --select recovery")
+		Expect(err).ToNot(HaveOccurred(), out)
+		vm.Reboot()
+		vm.EventuallyConnects(1200)
+
+		By("resetting")
+		out, err = vm.Sudo("kairos-agent --debug reset --unattended")
+		Expect(err).ToNot(HaveOccurred(), out)
+		vm.Reboot()
+		vm.EventuallyConnects(1200)
+
+		By("checking if after-reset was run")
+		out, err = vm.Sudo("ls /usr/local/after-reset-file")
+		Expect(err).ToNot(HaveOccurred(), out)
+		Expect(out).ToNot(MatchRegexp("No such file or directory"))
 	})
 })