Sync upgrade tests

Signed-off-by: Itxaka <[email protected]>
kairos-io · May 25, 2023 · 905cd54 · 905cd54
1 parent ec443ae
commit 905cd54
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 24 deletions.
diff --git a/tests/upgrade_cli_test.go b/tests/upgrade_cli_test.go
@@ -11,13 +11,14 @@ import (
 	. "github.com/spectrocloud/peg/matcher"
 )
 
-// test ci
 var _ = Describe("k3s upgrade manual test", Label("upgrade-with-cli"), func() {
-
+	var vm VM
 	containerImage := os.Getenv("CONTAINER_IMAGE")
 
-	var vm VM
 	BeforeEach(func() {
+		if containerImage == "" {
+			Fail("CONTAINER_IMAGE needs to be set")
+		}
 		_, vm = startVM()
 		vm.EventuallyConnects(1200)
 	})
@@ -35,19 +36,19 @@ var _ = Describe("k3s upgrade manual test", Label("upgrade-with-cli"), func() {
 
 	Context("upgrades", func() {
 		BeforeEach(func() {
-			if containerImage == "" {
-				Fail("CONTAINER_IMAGE needs to be set")
-			}
-
 			expectDefaultService(vm)
 			By("Copying config file")
 			err := vm.Scp("assets/config.yaml", "/tmp/config.yaml", "0770")
 			Expect(err).ToNot(HaveOccurred())
 			By("Manually installing")
-			out, err := vm.Sudo("kairos-agent --debug manual-install --device auto /tmp/config.yaml")
-			Expect(err).ToNot(HaveOccurred(), string(out))
+			out, err := vm.Sudo("/bin/bash -c 'set -o pipefail && kairos-agent manual-install --debug --device auto /tmp/config.yaml 2>&1 | tee manual-install.txt'")
+			Expect(err).ToNot(HaveOccurred(), out)
+
 			Expect(out).Should(ContainSubstring("Running after-install hook"))
 			vm.Sudo("sync")
+
+			err = vm.DetachCD()
+			Expect(err).ToNot(HaveOccurred())
 			By("Rebooting")
 			vm.Reboot()
 		})

diff --git a/tests/upgrade_latest_cli_test.go b/tests/upgrade_latest_cli_test.go
@@ -3,6 +3,7 @@ package mos_test
 import (
 	"fmt"
 	"os"
+	"path/filepath"
 	"time"
 
 	. "github.com/onsi/ginkgo/v2"
@@ -15,21 +16,31 @@ var _ = Describe("k3s upgrade manual test", Label("upgrade-latest-with-cli"), fu
 	containerImage := os.Getenv("CONTAINER_IMAGE")
 
 	BeforeEach(func() {
+		if containerImage == "" {
+			Fail("CONTAINER_IMAGE needs to be set")
+		}
 		_, vm = startVM()
 		vm.EventuallyConnects(1200)
 	})
 
 	AfterEach(func() {
+		if CurrentSpecReport().Failed() {
+			gatherLogs(vm)
+			serial, _ := os.ReadFile(filepath.Join(vm.StateDir, "serial.log"))
+			_ = os.MkdirAll("logs", os.ModePerm|os.ModeDir)
+			_ = os.WriteFile(filepath.Join("logs", "serial.log"), serial, os.ModePerm)
+			fmt.Println(string(serial))
+		}
 		Expect(vm.Destroy(nil)).ToNot(HaveOccurred())
 	})
 
 	Context("upgrades", func() {
 		BeforeEach(func() {
 			expectDefaultService(vm)
-
+			By("Copying config file")
 			err := vm.Scp("assets/config.yaml", "/tmp/config.yaml", "0770")
 			Expect(err).ToNot(HaveOccurred())
-
+			By("Manually installing")
 			out, err := vm.Sudo("/bin/bash -c 'set -o pipefail && kairos-agent manual-install --device auto /tmp/config.yaml 2>&1 | tee manual-install.txt'")
 			Expect(err).ToNot(HaveOccurred(), out)
 
@@ -38,26 +49,22 @@ var _ = Describe("k3s upgrade manual test", Label("upgrade-latest-with-cli"), fu
 
 			err = vm.DetachCD()
 			Expect(err).ToNot(HaveOccurred())
+			By("Rebooting")
 			vm.Reboot()
 		})
 
 		It("can upgrade to current image", func() {
 			currentVersion, err := vm.Sudo(getVersionCmd)
 			Expect(err).ToNot(HaveOccurred())
+			By(fmt.Sprintf("Checking current version: %s", currentVersion))
 			Expect(currentVersion).To(ContainSubstring("v"))
-			_, err = vm.Sudo("kairos-agent")
-			if err == nil {
-				out, err := vm.Sudo("kairos-agent upgrade --force --image " + containerImage)
-				Expect(err).ToNot(HaveOccurred(), string(out))
-				Expect(out).To(ContainSubstring("Upgrade completed"))
-				Expect(out).To(ContainSubstring(containerImage))
-				fmt.Println(out)
-			} else {
-				out, err := vm.Sudo("kairos upgrade --force --image " + containerImage)
-				Expect(err).ToNot(HaveOccurred(), string(out))
-				Expect(out).To(ContainSubstring("Upgrade completed"))
-				Expect(out).To(ContainSubstring(containerImage))
-			}
+
+			By(fmt.Sprintf("Upgrading to: %s", containerImage))
+			out, err := vm.Sudo("kairos-agent upgrade --force --image " + containerImage)
+			Expect(err).ToNot(HaveOccurred(), string(out))
+			Expect(out).To(ContainSubstring("Upgrade completed"))
+			Expect(out).To(ContainSubstring(containerImage))
+			fmt.Println(out)
 
 			vm.Reboot()