os/tests/upgrade_test.go

package integration

import (
	"fmt"
	"time"

	"strings"

	. "gopkg.in/check.v1"
)

func (s *QemuSuite) TestUpgrade050(c *C) {
	s.commonTestCode(c, "v0.5.0", "default")
}
func (s *QemuSuite) TestUpgrade061(c *C) {
	s.commonTestCode(c, "v0.6.1", "debian")
}
func (s *QemuSuite) TestUpgrade070(c *C) {
	s.commonTestCode(c, "v0.7.0", "debian")
}
func (s *QemuSuite) TestUpgrade071(c *C) {
	s.commonTestCode(c, "v0.7.1", "default")
}
func (s *QemuSuite) TestUpgrade071Persistent(c *C) {
	s.commonTestCode(c, "v0.7.1", "ubuntu")
}
func (s *QemuSuite) TestUpgrade080rc1(c *C) {
	s.commonTestCode(c, "v0.8.0-rc1", "debian")
}
func (s *QemuSuite) TestUpgrade080rc7(c *C) {
	// alpine console is unlikely to work before 0.8.0-rc5
	s.commonTestCode(c, "v0.8.0-rc7", "alpine")
}
func (s *QemuSuite) TestUpgrade081Persistent(c *C) {
	s.commonTestCode(c, "v0.8.1", "alpine")
}
func (s *QemuSuite) TestUpgrade081RollBack(c *C) {
	s.commonTestCode(c, "v0.7.1", "default")

	runArgs := []string{
		"--boothd",
	}
	{
		// and now rollback to 0.8.1
		thisVersion := "v0.8.1"
		s.RunQemuWith(c, runArgs...)

		s.CheckCall(c, fmt.Sprintf("sudo ros os upgrade --no-reboot -i rancher/os:%s%s --force", thisVersion, Suffix))

		s.Reboot(c)
		s.CheckOutput(c, "ros version "+thisVersion+"\n", Equals, "sudo ros -v")
		s.Stop(c)
	}
	{
		// and now re-upgrade to latest
		thisVersion := Version
		s.RunQemuWith(c, runArgs...)

		s.CheckCall(c, fmt.Sprintf("sudo ros os upgrade --no-reboot -i rancher/os:%s%s --force", thisVersion, Suffix))

		s.Reboot(c)
		s.CheckOutput(c, "ros version "+thisVersion+"\n", Equals, "sudo ros -v")
		s.Stop(c)
	}
}

// DisabledTestUpgradeInner is used to debug the above tests if they fail - the current imple of the check code limits itself to depths _one_ stacktrace
func (s *QemuSuite) DisableTestUpgradeInner(c *C) {
	startWithVersion := "v0.5.0"
	// CENTOS fails due to "sudo: sorry, you must have a tty to run sudo" :) so we won't test with it atm
	console := "debian"
	runArgs := []string{
		"--iso",
		"--fresh",
		"--cloud-config",
		"./tests/assets/test_12/cloud-config.yml",
	}
	version := ""
	{
		s.RunQemuWith(c, runArgs...)
		version = s.CheckOutput(c, version, Not(Equals), "sudo ros -v")
		version = strings.TrimSpace(strings.TrimPrefix(version, "ros version"))
		c.Assert(Version, Equals, version)

		fmt.Printf("installing %s", startWithVersion)
		s.PullAndLoadImage(c, fmt.Sprintf("rancher/os:%s", startWithVersion))

		//ADD a custom append line and make sure its kept in the upgraded version too

		s.CheckCall(c, fmt.Sprintf(`
echo "---------------------------------- generic"
set -ex
echo "ssh_authorized_keys:" > config.yml
echo "  - $(cat /home/rancher/.ssh/authorized_keys)" >> config.yml
sudo ros install --force --no-reboot --device /dev/vda -c config.yml --append rancher.password=rancher -i rancher/os:%s
sync
		`, startWithVersion))
		time.Sleep(500 * time.Millisecond)
		s.Stop(c)
	}

	// ./scripts/run --no-format --append "rancher.debug=true"
	runArgs = []string{
		"--boothd",
	}
	s.RunQemuWith(c, runArgs...)

	s.CheckOutput(c, "ros version "+startWithVersion+"\n", Equals, "sudo ros -v")

	if console != "default" {
		// Can't preload the startWithVersion console image, as some don't exist by that name - not sure how to approach that
		//s.PullAndLoadImage(c, fmt.Sprintf("rancher/os-%sconsole:%s", console, startWithVersion))
		// TODO: ouch. probably need to tag the dev / master version as latest cos this won't work
		// Need to pull the image here - if we do it at boot, then the test will fail.
		s.PullAndLoadImage(c, fmt.Sprintf("rancher/os-%sconsole:%s", console, "v0.8.0-rc3"))
		s.MakeCall(fmt.Sprintf("sudo ros console switch -f %s", console))
		c.Assert(s.WaitForSSH(), IsNil)
	}

	consoleVer := s.CheckOutput(c, "", Not(Equals), "sudo system-docker ps --filter name=^/console$ --format {{.Image}}")

	s.LoadInstallerImage(c)
	s.CheckCall(c, fmt.Sprintf("sudo ros os upgrade --no-reboot -i rancher/os:%s%s --force", Version, Suffix))

	s.Reboot(c)
	s.CheckOutput(c, "ros version "+Version+"\n", Equals, "sudo ros -v")
	s.CheckOutput(c, consoleVer, Equals, "sudo system-docker ps --filter name=^/console$ --format {{.Image}}")

	s.Stop(c)
}

func (s *QemuSuite) commonTestCode(c *C, startWithVersion, console string) {
	runArgs := []string{
		"--iso",
		"--fresh",
		"--cloud-config",
		"./tests/assets/test_12/cloud-config.yml",
	}
	version := ""
	{
		s.RunQemuWith(c, runArgs...)
		version = s.CheckOutput(c, version, Not(Equals), "sudo ros -v")
		version = strings.TrimSpace(strings.TrimPrefix(version, "ros version"))
		c.Assert(Version, Equals, version)

		fmt.Printf("installing %s", startWithVersion)
		s.PullAndLoadImage(c, fmt.Sprintf("rancher/os:%s", startWithVersion))

		//ADD a custom append line and make sure its kept in the upgraded version too

		s.CheckCall(c, fmt.Sprintf(`
echo "---------------------------------- generic"
set -ex
echo "ssh_authorized_keys:" > config.yml
echo "  - $(cat /home/rancher/.ssh/authorized_keys)" >> config.yml
sudo ros install --force --no-reboot --device /dev/vda -c config.yml --append rancher.password=rancher -i rancher/os:%s
sync
		`, startWithVersion))
		time.Sleep(500 * time.Millisecond)
		s.Stop(c)
	}

	// ./scripts/run --no-format --append "rancher.debug=true"
	runArgs = []string{
		"--boothd",
	}
	s.RunQemuWith(c, runArgs...)

	s.CheckOutput(c, "ros version "+startWithVersion+"\n", Equals, "sudo ros -v")

	if console != "default" {
		// Can't preload the startWithVersion console image, as some don't exist by that name - not sure how to approach that
		//s.PullAndLoadImage(c, fmt.Sprintf("rancher/os-%sconsole:%s", console, startWithVersion))
		// TODO: ouch. probably need to tag the dev / master version as latest cos this won't work
		// Need to pull the image here - if we do it at boot, then the test will fail.
		if console == "alpine" {
			s.PullAndLoadImage(c, fmt.Sprintf("rancher/os-%sconsole:%s", console, "v0.8.0-rc5"))
		} else {
			s.PullAndLoadImage(c, fmt.Sprintf("rancher/os-%sconsole:%s", console, "v0.8.0-rc3"))
		}
		s.MakeCall(fmt.Sprintf("sudo ros console switch -f %s", console))
		c.Assert(s.WaitForSSH(), IsNil)
	}

	consoleVer := s.CheckOutput(c, "", Not(Equals), "sudo system-docker ps --filter name=^/console$ --format {{.Image}}")

	s.LoadInstallerImage(c)
	s.CheckCall(c, fmt.Sprintf("sudo ros os upgrade --no-reboot -i rancher/os:%s%s --force", Version, Suffix))

	s.Reboot(c)
	s.CheckOutput(c, "ros version "+Version+"\n", Equals, "sudo ros -v")
	if console != "default" {
		s.CheckOutput(c, consoleVer, Equals, "sudo system-docker ps --filter name=^/console$ --format {{.Image}}")
	} else {
		s.CheckOutput(c, consoleVer, Not(Equals), "sudo system-docker ps --filter name=^/console$ --format {{.Image}}")
	}

	s.Stop(c)
}