func (s *rollbackSuite) TestRollbackMustRebootToOtherVersion(c *check.C) {
if common.BeforeReboot() {
// here we upgrade
updates.CallFakeOSUpdate(c)
common.Reboot(c)
} else if common.CheckRebootMark(c.TestName()) {
// after the first reboot we check that it actually booted
// a newer version than before
common.RemoveRebootMark(c)
currentVersion := common.GetCurrentUbuntuCoreVersion(c)
savedVersion := common.GetSavedVersion(c)
c.Assert(snappy.VersionCompare(currentVersion, savedVersion), check.Equals, 1,
check.Commentf("First reboot to the wrong version: %s <= %s", currentVersion, savedVersion))
// now we rollback to the previous version
cli.ExecCommand(c, "sudo", "snappy", "rollback", partition.OSSnapName(c),
common.GetSavedVersion(c))
common.RebootWithMark(c, c.TestName()+"-rollback")
} else if common.CheckRebootMark(c.TestName() + "-rollback") {
// and on the second reboot we check that the rollback
// did indeed rolled us back to the previous version
common.RemoveRebootMark(c)
currentVersion := common.GetCurrentUbuntuCoreVersion(c)
savedVersion := common.GetSavedVersion(c)
c.Assert(currentVersion, check.Equals, savedVersion,
check.Commentf("Second reboot to the wrong version: %s != %s", currentVersion, savedVersion))
}
}
func (s *rollbackSuite) TestRollbackMustRebootToOtherVersion(c *check.C) {
c.Skip("KNOWN BUG FIXME: https://bugs.launchpad.net/snappy/+bug/1534029")
if common.BeforeReboot() {
updates.CallFakeOSUpdate(c)
common.Reboot(c)
} else if common.CheckRebootMark(c.TestName()) {
common.RemoveRebootMark(c)
currentVersion := common.GetCurrentUbuntuCoreVersion(c)
c.Assert(snappy.VersionCompare(currentVersion, common.GetSavedVersion(c)), check.Equals, -1,
check.Commentf("Rebooted to the wrong version: %s", currentVersion))
cli.ExecCommand(c, "sudo", "snappy", "rollback", partition.OSSnapName(c),
common.GetSavedVersion(c))
common.RebootWithMark(c, c.TestName()+"-rollback")
} else if common.CheckRebootMark(c.TestName() + "-rollback") {
common.RemoveRebootMark(c)
currentVersion := common.GetCurrentUbuntuCoreVersion(c)
c.Assert(snappy.VersionCompare(currentVersion, common.GetSavedVersion(c)), check.Equals, 0,
check.Commentf("Rebooted to the wrong version: %s", currentVersion))
}
}
// Test that the ubuntu-core update to the same release and channel must install a newer
// version. If there is no update available, the channel version will be
// modified to fake an update. If there is a version available, the image will
// be up-to-date after running this test.
func (s *updateOSSuite) TestUpdateToSameReleaseAndChannel(c *check.C) {
if common.BeforeReboot() {
updateOutput := updates.CallFakeOSUpdate(c)
expected := "(?ms)" +
".*" +
fmt.Sprintf("^Reboot to use %s version .*\\.\n", partition.OSSnapName(c))
c.Assert(updateOutput, check.Matches, expected)
s.assertBootDirContents(c)
common.Reboot(c)
} else if common.AfterReboot(c) {
common.RemoveRebootMark(c)
currentVersion := common.GetCurrentUbuntuCoreVersion(c)
c.Assert(snappy.VersionCompare(currentVersion, common.GetSavedVersion(c)), check.Equals, 1,
check.Commentf("Rebooted to the wrong version: %d", currentVersion))
}
}
func showRebootMessage(installed []snappy.Part, o io.Writer) {
// Initialise to handle systems without a provisioned "other"
otherVersion := "0"
currentVersion := "0"
otherName := ""
needsReboot := false
for _, part := range installed {
// FIXME: extend this later to look at more than just
// core - once we do that the logic here needs
// to be modified as the current code assumes
// there are only two version instaleld and
// there is only a single part that may requires
// a reboot
if part.Type() != pkg.TypeCore {
continue
}
if part.NeedsReboot() {
needsReboot = true
}
if part.IsActive() {
currentVersion = part.Version()
} else {
otherVersion = part.Version()
otherName = part.Name()
}
}
if needsReboot {
if snappy.VersionCompare(otherVersion, currentVersion) > 0 {
// TRANSLATORS: the %s is a pkgname
fmt.Fprintln(o, fmt.Sprintf(i18n.G("Reboot to use the new %s."), otherName))
} else {
// TRANSLATORS: the first %s is a pkgname the second a version
fmt.Fprintln(o, fmt.Sprintf(i18n.G("Reboot to use %s version %s."), otherName, otherVersion))
}
}
}