Add upgrade tests

scripts/integration-test

@@ -14,4 +14,4 @@ if [ ! -e ../dist/artifacts/initrd ]; then
     ../scripts/dev
 fi
 
-go test -timeout 20m
+go test -timeout 30m

tests/common_test.go

@@ -56,7 +56,23 @@ func (s *QemuSuite) RunQemu(additionalArgs ...string) error {
 	}
 	runArgs = append(runArgs, additionalArgs...)
 
-	s.qemuCmd = exec.Command(s.runCommand, runArgs...)
+	return s.runQemu(runArgs...)
+}
+
+func (s *QemuSuite) RunQemuInstalled(additionalArgs ...string) error {
+	runArgs := []string{
+		"--qemu",
+		"--no-rebuild",
+		"--no-rm-usr",
+		"--installed",
+	}
+	runArgs = append(runArgs, additionalArgs...)
+
+	return s.runQemu(runArgs...)
+}
+
+func (s *QemuSuite) runQemu(args ...string) error {
+	s.qemuCmd = exec.Command(s.runCommand, args...)
 	s.qemuCmd.Stdout = os.Stdout
 	s.qemuCmd.Stderr = os.Stderr
 	if err := s.qemuCmd.Start(); err != nil {

tests/install_test.go

@@ -1,18 +0,0 @@
-package integration
-
-import (
-	"fmt"
-
-	. "gopkg.in/check.v1"
-)
-
-func (s *QemuSuite) TestInstall(c *C) {
-	err := s.RunQemu("--no-format")
-	c.Assert(err, IsNil)
-
-	s.LoadInstallerImage(c)
-
-	s.CheckCall(c, fmt.Sprintf(`
-sudo mkfs.ext4 /dev/vda
-sudo ros install -f --no-reboot -d /dev/vda -i rancher/os:%s%s`, Version, Suffix))
-}

tests/upgrade_test.go

@@ -0,0 +1,26 @@
+package integration
+
+import (
+	"fmt"
+
+	. "gopkg.in/check.v1"
+)
+
+func (s *QemuSuite) TestUpgrade(c *C) {
+	err := s.RunQemuInstalled()
+	c.Assert(err, IsNil)
+
+	s.CheckCall(c, `
+set -ex
+sudo ros os upgrade -i rancher/os:v0.5.0 --force --no-reboot`)
+
+	s.Reboot(c)
+
+	s.CheckCall(c, "sudo ros -v | grep v0.5.0")
+	s.LoadInstallerImage(c)
+	s.CheckCall(c, fmt.Sprintf("sudo ros os upgrade -i rancher/os:%s%s --force --no-reboot", Version, Suffix))
+
+	s.Reboot(c)
+
+	s.CheckCall(c, fmt.Sprintf("sudo ros -v | grep %s", Version))
+}