token2/snapd
0
0
Fork 0
mirror of https://github.com/token2/snapd.git synced 2026-03-13 11:15:47 -07:00
Code Issues Packages Projects Releases Wiki Activity

Added zero size systemd failover test

Browse Source
This commit is contained in:
Federico Gimenez
2015-06-22 13:50:54 +02:00
parent 0dca6ccb1c
commit ba93394dec
1 changed files with 36 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
_integration-tests/tests/failover_zero_size_file_test.go
View File

@@ -28,51 +28,64 @@ import (
)
const (
origFilenamePattern = "boot/%s%s*"
kernelFilename = "vmlinuz"
initrdFilename = "initrd"
destFilenamePrefix = "snappy-selftest-"
origBootFilenamePattern = "boot/%s%s*"
origSystemdFilenamePattern = "lib/systemd/%s%s"
kernelFilename = "vmlinuz"
initrdFilename = "initrd"
systemdFilename = "systemd"
destFilenamePrefix = "snappy-selftest-"
)
type zeroSizeKernel struct{}
type zeroSizeInitrd struct{}
type zeroSizeSystemd struct{}
func (zeroSizeKernel) set(c *C) {
commonSet(c, kernelFilename)
commonSet(c, origBootFilenamePattern, kernelFilename)
}
func (zeroSizeKernel) unset(c *C) {
commonUnset(c, kernelFilename)
commonUnset(c, origBootFilenamePattern, kernelFilename)
}
func (zeroSizeInitrd) set(c *C) {
commonSet(c, initrdFilename)
commonSet(c, origBootFilenamePattern, initrdFilename)
}
func (zeroSizeInitrd) unset(c *C) {
commonUnset(c, initrdFilename)
commonUnset(c, origBootFilenamePattern, initrdFilename)
}
func commonSet(c *C, filename string) {
filenamePattern := fmt.Sprintf(origFilenamePattern, "", filename)
func (zeroSizeSystemd) set(c *C) {
commonSet(c, origSystemdFilenamePattern, systemdFilename)
}
func (zeroSizeSystemd) unset(c *C) {
commonUnset(c, origSystemdFilenamePattern, systemdFilename)
}
func commonSet(c *C, origPattern, filename string) {
filenamePattern := fmt.Sprintf(origPattern, "", filename)
completePattern := filepath.Join(
baseOtherPath,
filenamePattern)
oldKernelFilename := getSingleFilename(c, completePattern)
newKernelFilename := fmt.Sprintf(
"%s/boot/%s%s", baseOtherPath, destFilenamePrefix, filepath.Base(oldKernelFilename))
oldFilename := getSingleFilename(c, completePattern)
filenameSuffix := fmt.Sprintf(
strings.Replace(origPattern, "*", "", 1), destFilenamePrefix, filepath.Base(oldFilename))
newFilename := fmt.Sprintf(
"%s/%s", baseOtherPath, filenameSuffix)
renameFile(c, baseOtherPath, oldKernelFilename, newKernelFilename)
renameFile(c, baseOtherPath, oldFilename, newFilename)
}
func commonUnset(c *C, filename string) {
func commonUnset(c *C, origPattern, filename string) {
completePattern := filepath.Join(
baseOtherPath,
fmt.Sprintf(origFilenamePattern, destFilenamePrefix, filename))
oldKernelFilename := getSingleFilename(c, completePattern)
newKernelFilename := strings.Replace(oldKernelFilename, destFilenamePrefix, "", 1)
fmt.Sprintf(origPattern, destFilenamePrefix, filename))
oldFilename := getSingleFilename(c, completePattern)
newFilename := strings.Replace(oldFilename, destFilenamePrefix, "", 1)
renameFile(c, baseOtherPath, oldKernelFilename, newKernelFilename)
renameFile(c, baseOtherPath, oldFilename, newFilename)
}
func renameFile(c *C, basePath, oldFilename, newFilename string) {
@@ -101,3 +114,7 @@ func (s *FailoverSuite) TestZeroSizeKernel(c *C) {
func (s *FailoverSuite) TestZeroSizeInitrd(c *C) {
commonFailoverTest(c, zeroSizeInitrd{})
}
func (s *FailoverSuite) TestZeroSizeSystemd(c *C) {
commonFailoverTest(c, zeroSizeSystemd{})
}