// SetupHostPathVolumes will create multiple PersistentVolumes with given capacity
func SetupHostPathVolumes(c kclient.PersistentVolumeInterface, prefix, capacity string, count int) (volumes []*kapi.PersistentVolume, err error) {
rootDir, err := ioutil.TempDir(TestContext.OutputDir, "persistent-volumes")
if err != nil {
return volumes, err
}
for i := 0; i < count; i++ {
dir, err := ioutil.TempDir(rootDir, fmt.Sprintf("%0.4d", i))
if err != nil {
return volumes, err
}
if _, err = exec.LookPath("chcon"); err != nil {
err := exec.Command("chcon", "-t", "svirt_sandbox_file_t", dir).Run()
if err != nil {
return volumes, err
}
}
if err = os.Chmod(dir, 0777); err != nil {
return volumes, err
}
pv, err := c.Create(CreatePersistentVolume(fmt.Sprintf("%s%s-%0.4d", pvPrefix, prefix, i), capacity, dir))
if err != nil {
return volumes, err
}
volumes = append(volumes, pv)
}
return volumes, err
}
// CleanupHostPathVolumes removes all PersistentVolumes created by
// SetupHostPathVolumes, with a given prefix
func CleanupHostPathVolumes(c kclient.PersistentVolumeInterface, prefix string) error {
pvs, err := c.List(kapi.ListOptions{})
if err != nil {
return err
}
prefix = fmt.Sprintf("%s%s-", pvPrefix, prefix)
for _, pv := range pvs.Items {
if strings.HasPrefix(pv.Name, prefix) {
c.Delete(pv.Name)
}
}
return nil
}
// CleanupHostPathVolumes removes all PersistentVolumes created by
// SetupHostPathVolumes, with a given prefix
func CleanupHostPathVolumes(c kclient.PersistentVolumeInterface, prefix string) error {
pvs, err := c.List(kapi.ListOptions{})
if err != nil {
return err
}
prefix = fmt.Sprintf("%s%s-", pvPrefix, prefix)
for _, pv := range pvs.Items {
if !strings.HasPrefix(pv.Name, prefix) {
continue
}
if err = c.Delete(pv.Name); err != nil {
fmt.Fprintf(g.GinkgoWriter, "WARNING: couldn't remove PV %s: %v\n", pv.Name, err)
continue
}
pvInfo, err := c.Get(pv.Name)
if err != nil {
fmt.Fprintf(g.GinkgoWriter, "WARNING: couldn't get meta info for PV %s: %v\n", pv.Name, err)
continue
}
volumeDir := pvInfo.Spec.HostPath.Path
if err = os.RemoveAll(volumeDir); err != nil {
fmt.Fprintf(g.GinkgoWriter, "WARNING: couldn't remove directory %q: %v\n", volumeDir, err)
continue
}
parentDir := filepath.Dir(volumeDir)
if parentDir == "." || parentDir == "/" {
continue
}
if err = os.Remove(parentDir); err != nil {
fmt.Fprintf(g.GinkgoWriter, "WARNING: couldn't remove directory %q: %v\n", parentDir, err)
continue
}
}
return nil
}