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 *serviceSuite) TearDownTest(c *check.C) {
if !common.NeedsReboot() && common.CheckRebootMark("") {
common.RemoveSnap(c, data.BasicServiceSnapName)
}
// run cleanup last
s.SnappySuite.TearDownTest(c)
}
func (s *rollbackSuite) TestRollbackMustRebootToOtherVersion(c *check.C) {
if common.BeforeReboot() {
common.CallFakeUpdate(c)
common.Reboot(c)
} else if common.CheckRebootMark(c.TestName()) {
common.RemoveRebootMark(c)
currentVersion := common.GetCurrentUbuntuCoreVersion(c)
c.Assert(currentVersion > common.GetSavedVersion(c), check.Equals, true,
check.Commentf("Rebooted to the wrong version: %d", currentVersion))
cli.ExecCommand(c, "sudo", "snappy", "rollback", "ubuntu-core",
strconv.Itoa(common.GetSavedVersion(c)))
common.SetSavedVersion(c, currentVersion)
common.RebootWithMark(c, c.TestName()+"-rollback")
} else if common.CheckRebootMark(c.TestName() + "-rollback") {
common.RemoveRebootMark(c)
currentVersion := common.GetCurrentUbuntuCoreVersion(c)
c.Assert(currentVersion < common.GetSavedVersion(c), check.Equals, true,
check.Commentf("Rebooted to the wrong version: %d", currentVersion))
}
}
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))
}
}