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)) } }